The Three Little Pigs(数论/逆元/dp)

题目
题意：给定n和q，q次查询，每次查询$x_i，1<=x_i<=3\ast n$，求${\sum }_{i=1}^n{C}_{3\ast i}^{x_i}$，其中$1<=n<=10^6,1<=q<=2\ast 10^5$，答案对1e9+7取模。
参考

题解：不会做，只能翻译下官方的233，最难的还是构造dp方程。
定义$dp[x][m]={\sum }_{i=0}^{n-1}{C}_{3\ast i+m}^x$，则$ans[x]=dp[x][0]+C_{3\ast n}^x$，此外，有
$dp[x][0]+dp[x][1]+dp[x][2]={\sum }_{i=0}^{3\ast n-1}{C}_i^x$，而
$C_{3\ast n}^{x+1}$
$=C_{3\ast n-1}^x+C_{3\ast n-1}^{x+1}$
$=C_{3\ast n-1}^x+C_{3\ast n-2}^x+C_{3\ast n-2}^{x+1}$
$={\sum }_{i=0}^{3\ast n-1}{C}_i^x$
所以有以下（1）式。加上组合数的$C_m^x=C_{m-1}^x+C_{m-1}^{x-1}$，可以得到（2）和（3）式。
$dp[x][0]+dp[x][1]+dp[x][2]=C_{3\ast n}^{x+1}$ （1）
$dp[x][1]=dp[x][0]+dp[x-1][0]$ （2）
$dp[x][2]=dp[x][1]+dp[x-1][1]$ （3）
$dp[0][0]=dp[0][1]=dp[0][2]=n$（4）
利用上面3个式子，和初始式子（4），我们就可以愉快的递推了。
但还有我们还需要预处理$C_{3\ast n}^i$，考虑到取模运算，我们还需要预处理1到3*n关于mod的逆，这里要用上线性递推，预处理线性逆元（图源）。

#include<bits/stdc++.h> 
using namespace std;
#define ll long long
const int maxn = 1000010;
const int mod = 1e9 + 7;

int n, q, x;
int C[3 * maxn];
int inv[3 * maxn];
int dp[3 * maxn][3];
/*
 * 0 <= a, b < mod
*/
int sub_mod(int a, int b) {
	int res = a - b;
	if (res < 0) res += mod;
	return res;
}
/*
 * 0 <= a, b < mod
*/
int add_mod(int a, int b) {
	int res = a + b;
	if (res >= mod) res -= mod;
	return res; 
}
int mul(int a, int b) {
	return 1LL * a * b % mod; 
}

void print(int N) {
	printf("C:\n");
	for (int i = 0; i <= N; ++i) {
		printf("%d\n", C[i]);
	}
	printf("dp:\n");
	for (int i = 0; i <= N; ++i) {
		printf("%d %d %d\n", dp[i][0], dp[i][1], dp[i][2]);
	}
	printf("-------------\n");
}
void init() {
	
	C[0] = 1;
	int N = 3 * n;
	inv[0]=1;
    inv[1]=1;
    for (int i = 2; i <= N; ++i) {
        inv[i] = mul(mod - mod / i, inv[mod%i]);
    }

	for (int i = 1; i <= N; ++i) {
//		C[i] = C[i-1] * (N - i + 1) / i;
		C[i] = mul(mul(C[i-1], (N - i + 1)), inv[i]);
	}

	memset(dp, 0, sizeof(dp));
	dp[0][0] = dp[0][1] = dp[0][2] = n;
	for (int i = 1; i <= N; ++i) {
//		dp[i][0] = (C[i+1] - 2 * dp[i-1][0] - dp[i-1][1]) / 3;
		int tmp = sub_mod(C[i+1], mul(2, dp[i-1][0]));
		tmp = sub_mod(tmp, dp[i-1][1]);
		dp[i][0] = mul(tmp, inv[3]);
		dp[i][1] = add_mod(dp[i][0], dp[i-1][0]);
		dp[i][2] = add_mod(dp[i][1], dp[i-1][1]);
	}
	
}
int main() {
	scanf("%d%d", &n, &q);
	init();
//	print(3 * n);
	while (q--) {
		scanf("%d", &x);
		printf("%d\n", add_mod(dp[x][0], C[x]));
	}
}