乘积数量题解

2021/05/26乘积数量题解

乘积数量

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

请你计算以下两值：

1. 使得 al×al+1×…×aral×al+1×…×ar 为负的索引对 (l,r)(l≤r)(l,r)(l≤r) 的数量。
2. 使得 al×al+1×…×aral×al+1×…×ar 为正的索引对 (l,r)(l≤r)(l,r)(l≤r) 的数量。

输入格式

第一行一个整数 nn。

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

输出格式

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

数据范围

1≤n≤2×1051≤n≤2×105,
−109≤ai≤109,ai≠0−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

前缀和的变形===>前缀积

这个题要优化 ，不然卡时间，自己做的时候没有优化，就只是用了O(n的平方)卡时间，优化成n就可以AC了。

首先，这道题是关于连续区间的，那么可以转化为前缀和。

因为只统计正数、负数的个数，所以并不需要存真正的数，只需要存一下正负即可。
（这里采用小技巧，当输入的数>0，置成+1，当输入的数<0，置成-1，这样算前缀积的时候就不会爆long long了）

声明，a数组表示原数组，s数组表示前缀积数组。

定义两个数组：b数组，c数组

b[i]状态表示： s[1]~s[i]中正数的个数

b数组状态计算: if(s[i]>0) b[i]=b[i-1]+1;
else b[i]=b[i-1];

c[i]状态表示： s[1]~s[i]中负数的个数

c数组状态计算：if(s[i]<0) c[i]=c[i-1]+1;
else c[i]=c[i-1];

前缀积可以类比前缀和求出，但要初始化s[0]=1，不然全部都是0，就不正确了。（当然，也可以从第二项开始算前缀积）。

现在算正的索引对
现在扫描前缀积s，当s[i]>0,我们要找以从1到1~s[i]的正数的前缀区间个数,即b[i],则res+=b[i];
为什么这样找呢,因为s[i]/s[j]表示j+1~i这个连续区间的乘积（1=<j<=i）采用这种方式可以找遍所有以第i个数结尾的连续区间,下面同理

​ 当s[i]<0,我们要找以从1到1~s[i-1]的负数的前缀区间个数,即c[i-1],则res+=c[i-1];
​ 这里为什么不能找s[i]呢，因为s[i]是负,则s[i]~s[i]为s[i]为负，不满足要求。所以从前一项找起。

现在我们知道了正的索引对，因为连续区间个数是(n+1)*n/2 （以每个点为起点的区间个数是一个等差数列，利用公式求和可以求出连续区间总数）
那我们减一下就可以求出负的索引对了。

AC源码：

#include <bits/stdc++.h>
using namespace std;
const int cmax = 2e5+5;
typedef long long ll;
int a[cmax];
int s[cmax];
int b[cmax];
int c[cmax];
int main()
{
    int n;
    cin>>n;
    for (int i = 0; i < n; ++i) {
        scanf("%d",&a[i]);
    }
    s[0]=1;
    for (int i = 1; i <=n ; ++i) {
        if (a[i-1]>0)
            s[i]=s[i-1]*1;
        else
            s[i]=s[i-1]*-1;
        if (s[i]>0)
        {
           b[i]=b[i-1]+1;
           c[i]=c[i-1];
        }
        else
        {
            c[i]=c[i-1]+1;
            b[i]=b[i-1];
        }
    }
    ll res=0;
    for (int i = 1; i <= n; i++)
    {
        if(s[i]>0)
        res+=b[i];
        else
        {
            res+=c[i-1];
        }
    }
    cout<<(ll)n*(n+1)/2-res<<" "<<res<<endl;
    return 0;
}