| MEPhI Аutumn Cup 2025 |
|---|
| Finished |
After studying linear algebra for machine learning, Leonid decided to return to discrete problems. One of them he liked so much that he decided to share it with you.
There is an array of non-zero integers $$$a_1, a_2, \ldots, a_n$$$, where a positive number $$$x$$$ in the array corresponds to $$$x$$$ opening brackets, and a negative number corresponds to $$$-x$$$ closing brackets. For example, the array [3, -2] corresponds to ((()).
A subarray $$$a_l, a_{l+1}, \ldots, a_r$$$, where $$$(l \le r)$$$ is called $$$good$$$ if its elements can be rearranged to form an array $$$b_1, b_2, \ldots, b_{r-l+1}$$$ such that the corresponding bracket sequence is balanced and there is at least one negative number among any two consecutive elements of the array $$$b$$$.
For the given array $$$a$$$, you need to count the number of its $$$good$$$ subarrays.
A bracket sequence is called balanced if it can be turned into a correct mathematical expression by adding the symbols + and 1.
The first line of input contains a single integer $$$n$$$ $$$(1 \le n \le 5 \cdot 10^5)$$$.
The second line contains $$$n$$$ numbers $$$a_1, a_2, \ldots, a_n\, (1 \le |a_i| \le 10^9)$$$.
In a single line, output one number – the answer to the problem.
53 -1 2 -2 1
2
| Name |
|---|


