题目:3574. 乘积数量
题解:像这种求连续区间的个数的题,都可以想到用前缀和。S[i]表示的是1~i项的乘积的符号(因为数据可能会超过long long ,且求解只与符号有关,所以用1,-1即可),然后看看前面s[1]~s[i-1]项的正负数的情况(f为负数的个数,z为正数的个数)。
奇数个负数为负数,所以当S[i]为负数时,看看前面有多少个负数S[j],这样j+1~i之间就有偶数个负数,这就是一个正的索引对。其他情况和这一样推理。
当S[i]==1,这时负的索引对f_ct+=f,正的索引对z_ct+=z+1;
当S[i]==-1,这时负的索引对f_ct+=z+1,正的索引对z_ct+=f;
这里+1的理由是是s[i]本身也是一个新的分组
#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
typedef pair<int ,int> PII;
const int N=2e5+10;
const int mod=100000007;
int n;
int main(){
cin>>n;
int f=0,z=0;
int x;
LL f_ct=0,z_ct=0;
LL t=1;
for(int i=1;i<=n;i++){
cin>>x;
if(x<0){
t*=-1;
}
if(t<0){
f_ct+=z+1;
z_ct+=f;
f++;
}else{
z_ct+=z+1;
f_ct+=f;
z++;
}
}
cout<<f_ct<<" "<<z_ct;
return 0;
}