Codeforces Round #585 (Div. 2), problem: (B) The Number of Products

Codeforces Round #585 (Div. 2), problem: (B) The Number of Products

大致题意：给出一个数组，问一共有多少个区间$[l,r]$满足区间乘积是正数

做法：一开始看到这题B我都懵了，以为是组合数学的东西，但是后来冷静下来一想好像是道dp

其实写出dp状态这题基本上就做完了

1.dp[i][0]表示[1,i]一共有多少段区间乘积是正数

2.dp[i][1]表示[1,i]一共有多少段以i为结尾的区间乘积是正数

3.dp[i][2]表示[1,i]一共有多少段以i为结尾的区间乘积是负数

显然，每往末端添加一个数，状态转移方程很容易得出

/**
 *  Author1: low-equipped w_udixixi
 *  Author2: Sherؼlock
 *  Date :2019-09-15
 **/
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<string>
#include<vector>
#include<stack>
#include<bitset>
#include<cstdlib>
#include<cmath>
#include<set>
#include<list>
#include<deque>
#include<queue>
#include<map>
#define ll long long
#define pb push_back
#define rep(x,a,b) for (int x=a;x<=b;x++)
#define repp(x,a,b) for (int x=a;x<b;x++)
#define W(x) printf("%d\n",x)
#define WW(x) printf("%lld\n",x)
#define pi 3.14159265358979323846
#define mem(a,x) memset(a,x,sizeof a)
#define lson rt<<1,l,mid
#define rson rt<<1|1,mid+1,r
using namespace std;
const int maxn=2e5+7;
const int INF=1e9;
const ll INFF=1e18;
ll a[maxn];
ll dp[maxn][5];
ll f(ll n){return (n+1)*n/2;}
int main()
{
    ll n;
    scanf("%lld",&n);
    rep(i,1,n)
    {
        scanf("%lld",&a[i]);
        if (a[i]>0)a[i]=1;
        else a[i]=-1;
    }
    if (a[1]==1){dp[1][0]=1;dp[1][1]=1;dp[1][2]=0;}
    else {dp[1][0]=0;dp[1][1]=0;dp[1][2]=1;}
    rep(i,2,n)
    {
        if (a[i]==1)
        {
            dp[i][1]=dp[i-1][1]+1;
            dp[i][2]=dp[i-1][2];
            dp[i][0]=dp[i-1][0]+dp[i-1][1]+1;
        }
        else
        {
            dp[i][2]=dp[i-1][1]+1;
            dp[i][1]=dp[i-1][2];
            dp[i][0]=dp[i-1][0]+dp[i-1][2];
        }
    }
    printf("%lld %lld\n",f(n)-dp[n][0],dp[n][0]);
    return 0;
}