The Number of Products (dp)

The Number of Products

题目链接

大致题意:

给出一个序列,非零组成,求有多少个连续的子区间乘积小于零，大于零

---

解题思路:

对于第i个数
如果a[i]>0,它对正数区间的贡献是前面子区间乘积>0的个数
如果a[i]>0,它对负数区间的贡献是前面子区间乘积>0的个数+1
如果a[i]<0,它对正数区间的贡献是前面子区间乘积<0的个数
如果a[i]<0,它对负数区间的贡献是前面子区间乘积>0的个数+1

状态表示:f[i][0]表示以i结尾,乘积是负,即奇数个负数的区间个数 f[i][1]表示以i结尾,乘积是正,即偶数个负数的区间个数

转移方程:
a[i]>0 f[i][0] = f[i - 1][0] f[i][1] = f[i - 1][1] + 1
a[i]<0 f[i][0] = f[i - 1][1] + 1 f[i][1] = f[i - 1][0]

---

AC代码:

#include <bits/stdc++.h>
#define rep(i, n) for (int i = 1; i <= (n); ++i)
#define debug(a) cout << #a << " = " << a << endl;
using namespace std;
typedef long long ll;
const int N = 2e5 + 10;
int n, m, k;
int a[N];
ll f[N][2]; //f[i][0]表示以i结尾,乘积是负,即奇数个负数的区间个数  f[i][1]表示以i结尾,乘积是正,即偶数个负数的区间个数
int main(void)
{
	ios::sync_with_stdio(0); cin.tie(0);
	cin >> n;
	for (int i = 1; i <= n; ++i)cin >> a[i];
	ll res1 = 0, res2 = 0;
	for (int i = 1; i <= n; ++i) {
		if (a[i] < 0) {
			f[i][0] = f[i - 1][1] + 1;
			f[i][1] = f[i - 1][0];
		}
		else {
			f[i][0] = f[i - 1][0];
			f[i][1] = f[i - 1][1] + 1;
		}
		res1 += f[i][0]; res2 += f[i][1];
	}
	cout << res1 << ' ' << res2 << endl;
	return 0;
}

---