数数

Gig Combinatorics

• 题意：给(n <= 1e6)的一串序列，只包含(1, 2, 3)三种元素，然后问，一共有多少种序列，序列开头必须只有一个(1),结尾必须只有一个(3)，之间至少有一个(2)。

• 题解：只能说自己见的少了吧，只会统计数量，然后乘或者其他的，其实这道题，先统计前面(1)的数量，然后再统计到(2)的方案数，最后遇到三就加上到(2)的方案数。主要是(2)如何理解。当前元素为(2)，那么前面(1)的数量必然因为这个(2)多做了一次贡献，因此(sum_2[i] += sum_1[i-1]),除此之外，之前的(sum_2)还需要( imes 2)，想一下就知道了。

• 代码：

#include <iostream>
using namespace std;
const int N = 1e6 + 99;
typedef long long ll;
const ll mod = 1e9 + 7;
ll a[N];
ll sum1[N];
ll sum2[N];
int main() {
	int n;
	cin >> n;
	for (int i = 1; i <= n; i++) cin >> a[i];
	ll ans = 0;
	for (int j = 1; j <= n; j++) {
		if (a[j] == 1) {
			sum1[j] = sum1[j-1] + 1;
		} else sum1[j] = sum1[j-1];
		if (a[j] == 2) {
			sum2[j] = sum1[j-1] + sum2[j-1] * 2;
		} else sum2[j] = sum2[j-1];
		sum1[j] %= mod;
		sum2[j] %= mod;
		if (a[j] == 3) {
			(ans += sum2[j]) %= mod;
		}
	}
	cout << ans << endl;
}