题目出处:CF 1215B
给你一个序列包含 \(n\) 个元素的序列 \(a_1, a_2, \dots , a_n\) (每个元素 \(a_i \ne 0\))。
你需要计算如下两个值:
即:这个序列中有多少子串(子串即连续子序列)的乘积为正,有多少子串的乘积为负。
输入的第一行包含一个整数 \(n (1 \le n \le 2 \cdot 10^{5})\) —— 用于表示序列中元素的个数。
输入的第二行包含 \(n\) 个整数 \(a_1, a_2, \dots , a_n (-10^{9} \le a_i \le 10^{9}; a_i \neq 0)\) ,用于表示序列中的元素。
输出两个正数,以一个空格分隔。分别表示乘积为正的子串的个数,以及乘积为负的子串的个数。
5
5 -3 3 -1 1
8 7
10
4 2 -4 3 1 2 -4 3 2 3
28 27
5
-1 -2 -3 -4 -5
9 6
本体涉及算法:动态规划。
这道题目是一个动态规划入门题。
我们定义状态:
可以得到状态转移方程为:
实现代码如下:
#include <bits/stdc++.h>
using namespace std;
const int maxn = 200020;
int n, a[maxn];
long long f[maxn][2], sum[2]; // f[i][0]表示以a[i]结尾整数数量, meanwhile f[i][2]负数。
int main() {
cin >> n;
for (int i = 1; i <= n; i ++) cin >> a[i];
for (int i = 1; i <= n; i ++) {
if (a[i] > 0) {
f[i][0] = f[i-1][0] + 1;
f[i][1] = f[i-1][1];
}
else { // if (a[i] < 0) {
f[i][0] = f[i-1][1];
f[i][1] = f[i-1][0] + 1;
}
for (int j = 0; j < 2; j ++) sum[j] += f[i][j];
}
cout << sum[1] << " " << sum[0] << endl;
return 0;
}
原文:https://www.cnblogs.com/zifeiynoip/p/11570762.html