F - The Number of Products （前缀积）

最新推荐文章于 2021-07-09 11:05:38 发布

九羽-

最新推荐文章于 2021-07-09 11:05:38 发布

阅读量446

点赞数 1

分类专栏：前缀和前缀积

本文链接：https://blog.csdn.net/weixin_43828245/article/details/102632344

版权

前缀和前缀积专栏收录该内容

3 篇文章 0 订阅

订阅专栏

F - The Number of Products （前缀积）

题意：给n个数 a1，a2，a3 ... an 。任意区间 [ L, R ] 使得 $a{_{l}}^{}$ * $a{_{l+1}}^{}$ *...* $a{_{r}}^{}$ 为负数，零，正数的方案分别有几种。

思路：维护一个前缀积d，分析，如果d[R] 为正数， d[L-1] 为负数，那么区间 [ L, R ] 一定是负数。相似的分析其他情况。

定义：前缀积为正数的个数为cnt1，为负数的个数为cntfu

我们发现正数的情况是：L R 的前缀积都为负， L R 的前缀积都为正， LR相等切为正。 ans = $C_{cnt1}^{2}$ + $C_{cnt2}^{2}$ + cnt1 = cnt1*(cnt1-1)/2 + cntfu*(cntfu-1)/2 + cnt1;

负数的情况是：L R 的前缀积一正一负， LR相等切为负，ans = $C_{cnt1}^{1}$ * $C_{cnt2}^{1}$ + cntfu = cnt1*cntfu + cntfu;

零的情况是：ans = (i-lastzero)*(n-i+1);

特殊的看待0，可以发现0可以把这n个序列给分割开，单独求就好，对于每个为0的数可以归纳一个规律（左边的个数+1）*（右边的个数+1 ），其中左边的个数是左边到零的数。

代码：

#include <bits/stdc++.h>

using namespace std;

typedef long long ll;
const int maxn = 2e5+10;
ll n;
ll a[maxn];
ll d[maxn]; // 前缀积

int main()
{
    ll i,j,x;
    cin >> n;
    d[0] = 1; // 前缀积的初始化
    ll cnt = 1;
    ll cnt1 = 0; // 正数的个数
    ll cntfu = 0; // 负数的个数
    ll ans1=0,ans2=0,ans3=0;
    ll lastzero = 0;
    for ( i=1; i<=n; i++ ) {
        scanf("%lld",&x);
        if ( x>0 ) {
            a[cnt] = 1;
            d[cnt] = d[cnt-1]*a[cnt];
            if ( d[cnt]>0 ) cnt1++;
            else cntfu++;
            cnt ++;
        }
        else if ( x<0 ) {
            a[cnt] = -1;
            d[cnt] = d[cnt-1]*a[cnt];
            if ( d[cnt]>0 ) cnt1++;
            else cntfu++;
            cnt ++;
        }
        else if ( x==0 ) { // 到了0，就分割开了，ans+=答案
            ans1 += cnt1*cntfu + cntfu;
            ans2 += (i-lastzero)*(n-i+1);
            ans3 += cnt1 + cnt1*(cnt1-1)/2 + cntfu*(cntfu-1)/2;
            cnt = 1; cnt1=0; cntfu=0;
            lastzero = i;
        }
    }
    if ( cnt!=1 ) {
        ans1 += cnt1*cntfu + cntfu;
        ans2 += (i-lastzero+1)*(n-i+1);
        ans3 += cnt1 + cnt1*(cnt1-1)/2 + cntfu*(cntfu-1)/2;
    }
    cout << ans1 << " " << ans2 << " " << ans3 << endl;

    return 0;
}