【题解】Popping Balls AtCoder Code Festival 2017 qual B E 组合计数

蒟蒻__stdcall终于更新博客辣~
一下午+一晚上=一道计数题QAQ
为什么计数题都这么玄学啊QAQ

---

Prelude

题目链接：这里是传送门=￣ω￣=
下面我将分几个步骤讲一下这个题的做法，大家不必一次看完，可以一点一点地推进思路，希望对锻炼大家的思维能力有帮助o(￣▽￣)ブ。

---

Step 1

首先要看出来这是一个计数题对吧。。。
计数题有很多做法，对于这个题，我们考虑合理枚举，即不重复不遗漏地枚举所有情况，然后乘上一个组合数，并通过前缀和优化来降低复杂度。

---

Step 2

枚举什么东西呢？
我们考虑重新定义一下题目中的(s)和(t)。
我们定义(t)为，第一次取的蓝色球的位置，同时，为了构造出尽量多的方案，这个(t)要尽量地靠左。
举个栗子吧，比如我们想第一个取蓝色球，那么就有(t = a + 1)，如果我们想等所有的红色球取完再取蓝色球，那么(t = 1)。
这里说的不太清楚。。。大家意会一下QAQ
于是我们首先枚举(t)，即第一个取的蓝色球的位置，那么我们在取到这个球之前，首先需要从首部取(a + 1 - t)个红色球。
也就是说，我们得到的结果序列，是以(a + 1 - t)个红色球和一个蓝色球开始的，对于不同的(t)，我们得到的序列肯定是不同的，不会重复。
同时，因为我们肯定要取蓝色球，所以也覆盖了所有情况而不会遗漏。

---

Step 3

确定了一个(t)之后，我们取走最前面的(a + 1 - t)个红色球和在(t)位置的一个蓝色球。剩余(t - 1)个红色球和(b - 1)个蓝色球。
接下来该枚举什么？
考虑现在一共有(t + b - 2)个球，我们取走(b - 1)个，使得剩下(t - 1)个球，然后(t)这个位置就没有用了，接下来就可以考虑确定(s)的位置。
于是我们枚举(i)，即剩下的(t - 1)个球中，蓝色球的数量。
然后，确定了(i)之后，我们发现我们需要取走(i)个红色球和(b - 1 - i)个蓝色球。
容易发现，接下来准备取走的球，是不受位置限制的，颜色可以任意取，也就是说，这即将要取走的(b - 1)个球中，红色球和蓝色球可以排在任意位置。
于是，取走这(b - 1)个球的方案数就是(C_{b-1}^{i})。
同样，当我们固定了(t)之后再枚举(i)，也肯定是不重复不遗漏的，原因太显然了我就懒得讲了大家自己体会叭。。。

---

Step 4

现在我们剩了(t - 1 - i)个红色球和(i)个蓝色球。
现在(t)这个位置已经没法用了，我们还想要取蓝色球的话，需要新建一个位置(s)，于是我们枚举(s)的位置。
注意，因为现在红色球只有(t - 1 - i)个，所以(s)最多枚举到(t - i)。
同理，确定了一个(s)之后，我们需要先取走(t - i - s)个红色球，然后再取走一个蓝色球，和上面的情况类似，这里也是不重复不遗漏的。

---

Step 5

现在我们只剩(s - 1)个红色球和(i - 1)个蓝色球了。
类似Step 3，我们考虑取走(i - 1)个球，剩余(s - 1)个球，枚举这剩余(s - 1)个球中，蓝色球的数量(j)，那么我们要取走(j)个红色球和(i - 1 - j)个蓝色球。
同理，取这些球的时候是不受位置限制的，方案数是(C_{i-1}^{j})。

---

Step 6

于是我们就只剩最后(s - 1)个球了，只能从首部一个一个取。

---

Step 7

我们枚举了(t, i, s, j)四个量，算法的时间复杂度为(O(n^4))。
然后我们发现，枚举(j)并累加(C_{i-1}^{j})，实际上是杨辉三角第(i-1)行的一个前缀和，于是可以记(sum[i][j])为杨辉三角第(i)行前(j)项的和，复杂度变为(O(n^3))。
我们还能发现，枚举(s)并累加(sum[i-1][min(s-1,i-1)])，其实也是(sum[i-1])的一个前缀和，于是记(f[i][j])为(sum[i])的前(j)项的和。
于是复杂度就变成了(O(n^2))辣~
撒花花~★,°:.☆(￣▽￣)/$:.°★ 。
（我想了一整个下午加一整个晚上好叭QAQ）

---

Code

代码如下

#include <cstring>
#include <cstdio>
#include <algorithm>

using namespace std;
typedef long long ll;
const int MAXN = 2010;
const int MOD = 1e9+7;
int _w;

int C[MAXN][MAXN], sum[MAXN][MAXN], f[MAXN][MAXN];
void prelude() {
	for( int i = 0; i < MAXN; ++i ) {
		f[i][0] = sum[i][0] = C[i][0] = 1;
		for( int j = 1; j <= i; ++j )
			C[i][j] = (C[i-1][j] + C[i-1][j-1]) % MOD;
		for( int j = 1; j < MAXN; ++j )
			sum[i][j] = (sum[i][j-1] + C[i][j]) % MOD;
		for( int j = 1; j < MAXN; ++j )
			f[i][j] = (f[i][j-1] + sum[i][j]) % MOD;
	}
}

int solvet( int t, int b ) {
	int ans = 1; // i == 0
	for( int i = 1; i <= min(t-1, b); ++i )
		ans = int((ans + (ll)f[i-1][t-i-1] * C[b][i]) % MOD);
	return ans;
}

int main() {
	prelude();
	int a, b;
	_w = scanf( "%d%d", &a, &b );
	if( !a || !b ) return puts("1"), 0;
	int ans = 0;
	for( int t = 1; t <= a+1; ++t )
		ans = (ans + solvet(t, b-1)) % MOD;
	printf( "%d
", ans );
	return 0;
}