给定一个长度为 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
首先,数据范围是2e5,所以时间复杂度应该控制在nlogn之内,因为这题是让求所有区间的乘积的正负,求一段区间的乘积,可以考虑用前缀和的思想做。
1.因为这题求的是正负,所以可以在前缀和数组中只存1、-1,用来代表正负。
2.因为这题让求所有区间的乘积,所以只用前缀和的话,必须要两重for,会超时的,所以可以每次枚举一个数代表区间的右端点,然后通过该店左边的正数的个数和负数的个数来计算一当前点为右端点的区间的乘积的正数和负数数量。可以用curfu、curzh来代表当前点左边的负数、正数的数量,
3.在计算当前负数、正数的数量的时候,每次都需要计算一次当前的右端点,所以curzh的初始值应该是1,(这里不清楚)
代码如下:
#include<iostream>
using namespace std;
const int N=2e5+10;
typedef long long ll;
int main(){
ll a=0,b=0;//a代表正数
int ca=1,cb=0,sum=1;
int x,c;
cin>>x;
for(int i=0;i<x;i++){
cin>>c;
if(c<0)sum*=-1;
if(sum<0){
a+=cb;
b+=ca;
cb++;
}
else{
a+=ca;
b+=cb;
ca++;
}
}
}