题目描述

给定一个长度为 n 且不包含 0 的整数序列 a1,a2,…,an。

请你计算以下两值：

使得 al×al+1×…×ar 为负的索引对 (l,r)(l≤r) 的数量。
使得 al×al+1×…×ar 为正的索引对 (l,r)(l≤r) 的数量。
输入格式
第一行一个整数 n。

第二行包含 n 个整数 a1,…,an。

输出格式
共一行，输出单个空格隔开的两个整数，分别表示负的索引对数和正的索引对数。

数据范围
1≤n≤2×105,
−109≤ai≤109,ai≠0。

样例

输入样例1：
5
5 -3 3 -1 1
输出样例1：
8 7
输入样例2：
10
4 2 -4 3 1 2 -4 3 2 3
输出样例2：
28 27
输入样例3：
5
-1 -2 -3 -4 -5
输出样例3：
9 6

看代码

C++ 代码

#include  <iostream>
using namespace std;
//onst int p=200010;
typedef long long ll;
int main()
{
    int n,a;
    cin>>n;
   // s[0]=1;
    ll x1=0,x2=0;//x1目前有多少项乘积为正号，x2有多少项乘积为负号的
    ll y1=1,y2=0,s=1;//y1表示前面有多少s为正号，y2有多少个s为负号
    for(int i=1;i<=n;i++)
    {
        cin>>a;
        if(a<0)
        s*=-1;
        if(s>0)//如果这个位置是正的
        {
          x1+=y1;//x1需要加上前面是正号的s，即y1
          x2+=y2;//x2需要加上前面是负号的s，即y2
          y1++;//此位置为正，y1加1
        }
        else//如果这个位置是负的
        {
            x1+=y2;//x1需要加上和这个位置为负号的s才能变成正号，即y2
            x2+=y1;//x2需要加上和这个位置为正号的s才能变成负号，即y1
            y2++;//此位置为负，y2加1
        }
    }
    cout<<x2<<" "<<x1;
    return 0;
}