[SRM 498 Div.1] Hard FoxJumping
题意
有一个网格,每一步可以走\((0\cdots M_x,0\cdots M_y)\)中的任意非零向量。
有\(K\)种向量不能走,为\((k_i,k_i)\),其中\(k_i\)为\(10\)的倍数。
求从\((0,0)\)走\(R\)步到\((T_x,T_y)\)的方案数。对\(10007\)取模。
\(,1\leq T_x,T_y,M_x,M_y\leq800,1\leq R\leq1600,0\leq K\leq50\)
分析
这个题目其实是个套路题。首先第一眼我们看到这个题目,想到的是FFT,但是我们注意到\(K\)很小,这个就很可能是个套路题。我们考虑先放掉这个不能走向量的约束,然后发现非常好做:考虑设一个\(f_{i,x,y}\)表示这个走了\(i\)次的答案,那么显然有\(f_{i,x,y}\rightarrow f_{i+1,x+dx,y+dy}\)。进一步地,我们如果考虑到这里实际上两个维度上是无关的,也就是说,我们完全可以独立考虑,最后乘起来即可(因为没有了限制,我们可以相当于是两个方向上分别走)。这样我们就可以用前缀和轻松解决(对于每一个位置,都是往前\(T_x\)或\(T_y\)长度区间中的和转移过来)。
然后我们回到原来的问题,考虑有这样的限制该怎么做?对此,我们需要利用类似二项式反演推出来的时候的思路:选一些不合法的,然后按照选\(\times\)不合法方案数\(\times\)全合法方案数的思路容斥出结果。不合法方案数直接推,于是我们最终能够得出式子是:
\[Ans=\sum_{i,z}\limits(-1)^i\binom{R}{i}g_{i,z}f_{R-i,T_x-10z,T_y-10z}
\]
感性理解一下,就能写出程序辣!非常好实现
# include <bits/stdc++.h>
using namespace std;
const int N = 888;
const int MOD = 10007;
int dp[N << 1][N];
int x[N << 1][N], y[N << 1][N];
int gx[N << 1][N], gy[N << 1][N];
int C[N << 1][N << 1];
int n, m, mn, dx, dy, R, cnt, ans;
int a[N];
inline void Add(int &x, int a) {
x = (x + a) % MOD;
}
int getCount(int Tx, int Ty, int Mx, int My, int _R, vector <int> bad) {
n = Tx; m = Ty; dx = Mx; dy = My; R = _R; mn = min(m, n);
bad.push_back(0); cnt = bad.size();
for (int i = 0; i < cnt; i++) a[i + 1] = bad[i];
sort(a + 1, a + cnt + 1);
cnt = unique(a + 1, a + cnt + 1) - a - 1;
ans = 0;
memset(x, 0, sizeof x);
memset(y, 0, sizeof y);
memset(dp, 0, sizeof dp);
for (int i = 0; i <= R; i++) {
C[i][0] = 1;
for (int j = 1; j <= i; j++)
C[i][j] = (C[i - 1][j - 1] + C[i - 1][j]) % MOD;
}
x[0][0] = 1;
for (int i = 0; i <= R; i++)
for (int j = 0; j <= n; j++) {
gx[i][j + 1] = (gx[i][j] + x[i][j]) % MOD;
Add(x[i + 1][j], (gx[i][j + 1] - gx[i][max(j + 1 - dx - 1, 0)] + MOD) % MOD);
}
y[0][0] = 1;
for (int i = 0; i <= R; i++)
for (int j = 0; j <= m; j++) {
gy[i][j + 1] = (gy[i][j] + y[i][j]) % MOD;
Add(y[i + 1][j], (gy[i][j + 1] - gy[i][max(j + 1 - dy - 1, 0)] + MOD) % MOD);
}
dp[0][0] = 1;
for (int i = 0; i <= R; i++)
for (int j = 0; j * 10 <= mn; j++)
for (int k = 1; k <= cnt; k++)
if (j * 10 + a[k] <= mn) Add(dp[i + 1][j + a[k] / 10], dp[i][j]);
else break;
for (int p = 0; p <= R; p++)
for (int q = 0; 10 * q <= mn; q++) {
int res = x[R - p][n - 10 * q] * y[R - p][m - 10 * q] % MOD * dp[p][q] % MOD * C[R][p] % MOD;
if (p & 1) Add(ans, MOD - res);
else Add(ans, res);
}
return ans;
}
vector<int> bad;
int main(){
freopen("jump.in","r",stdin);
freopen("jump.out","w",stdout);
int Tx,Ty,Mx,My,_R,_K;
scanf("%d%d%d%d%d%d",&Tx,&Ty,&Mx,&My,&_R,&_K);
for (int i=0;i<_K;++i){ int x; scanf("%d",&x); bad.push_back(x); }
printf("%d\n",getCount(Tx,Ty,Mx,My,_R,bad));
}