「SDWC2018 Day1」网格

Address

LOJ#6374

Solution

题目给出 $3$ 个限制：

1. $x\le x^{\prime }\le x+M_x,y\le y^{\prime }\le y+M_y$
2. 每一步都不能停在原地
3. 每一步 $x,y$ 的增量不能同时等于 $k_i$

先考虑怎么满足限制 $1$：

发现两维之间独立，可以分别计算答案然后相乘。那么问题转化为在数轴上从 $0$ 走到 $T_x$，走 $R$ 步，每一步满足 $x\le x^{\prime }\le x+M_x$。

记 $f(i)$ 表示恰好有 $i$ 步超过步长限制的方案数。$g(i)$ 表示至少有 $i$ 步超过步长限制的方案数，那么有 $g(i)=C_R^i\times C_{T_x-i(M_x+1)+R-1}^{R-1}$。显然每个 $f(j)$ 都在 $g(i)$ 中被算了 $C_j^i$ 次，即 $g(i)={\sum }_{j=i}^R{C}_j^i\times f(j)$。

下面证明 $g(i)={\sum }_{j=i}^{R}f(j)$ 等价于 $f(i)={\sum }_{j=i}^R(-1{)}^{j-i}\times C_j^i\times g(j)$：

$$\sum _{j=i}^R(-1{)}^{j-i}\times C_j^i\times g(j)=\sum _{j=i}^R(-1{)}^{j-i}\times C_j^i\times \sum _{k=j}^R\times C_k^j\times f(k)$$ $$=\sum _{k=i}^R(\sum _{j=i}^k(-1{)}^{j-i}\times C_j^i\times C_k^j)\times f(k)$$

把 ${\sum }_{j=i}^k(-1{)}^{j-i}\times C_j^i\times C_k^j$ 单独拿出来考虑：
$$\sum _{j=i}^k(-1{)}^{j-i}\times C_j^i\times C_k^j=C_k^i\sum _{j=i}^k(-1{)}^{j-i}\times C_{k-i}^{j-i}=C_k^i\sum _{j=0}^{k-i}(-1{)}^j\times C_{k-i}^j$$ 发现只有在 $k=i$ 的时候上式为 $1$，$k\ne i$ 时上式都为 $0$，那么：$$\sum _{k=i}^R(\sum _{j=i}^k(-1{)}^{j-i}\times C_j^i\times C_k^j)\times f(k)=f(i)$$

证毕。

我们要求的就是 $$f(0)=\sum _{i=0}^R(-1{)}^{i-0}\times C_i^0\times g(i)=\sum _{i=0}^R(-1{)}^i\times g(i)$$

接下来考虑限制 $2$。

记 $h(i)$ 表示走 $i$ 步到终点，每步都满足限制 $1,2$ 的方案数，$p(i)$ 表示走 $i$ 步到终点，每步都满足限制 $1$ 的方案数，可以直接由 $f$ 得出。枚举强制 $j$ 步必须满足限制 $2$，剩下 $i-j$ 步停在原地，那么有：
$$p(i)=\sum _{j=0}^i{C}_i^j\times h(j)$$
反演可得（证明类似 $f,g$ 的关系式）：$$h(R)=\sum _{j=0}^i(-1{)}^{R-i}\times C_R^i\times p(i)$$

接下来考虑限制 $3$。

记 $a(i)$ 表示走 $R$ 步到终点，每步都满足限制 $1,2$，至少有 $i$ 步走了非法向量的方案数，记 $calc(T_x,T_y,M_x,M_y,R)$ 表示题目去掉第 $3$ 个限制之后，走$R$ 步到 $T_x,T_y$，每一步满足 $x\le x^{\prime }\le x+M_x,y\le y^{\prime }\le y+M_y$ 的方案数。

先用背包预处理出 $S(i,j)$ 表示选出 $i$ 个非法向量，$\sum k_i=jG$ 的方案数，然后有：$$a(i)=\sum _{j}S(i,j)\times calc(T_x-jG,T_y-jG,M_x,M_y,R-j)\times C_R^i$$

记 $b(i)$ 表示走 $R$ 步到终点，每步都满足限制 $1,2$，恰好有 $i$ 步走了非法向量的方案数，显然有 $$a(i)=\sum _{j=i}^R{C}_j^i\times b(j)$$ 同样反演可得 $$b(0)=\sum _{i=0}^R(-1{)}^{i}a(i)$$

时间复杂度 $\mathcal{O}($跑的飞快$)$。

Code

#include <bits/stdc++.h>

using namespace std;

#define ll long long

const int N = 1e6, e = 1005, o = 105, mod = 1e9 + 7;
int Tx, Ty, Mx, My, R, G, n, a[e], ans, S[e][e], Lim, g[e];
int fac[N + 5], inv[N + 5];

inline int ksm(int x, int y)
{
	int res = 1;
	while (y)
	{
		if (y & 1) res = (ll)res * x % mod;
		y >>= 1;
		x = (ll)x * x % mod; 
	}
	return res;
}

inline void add(int &x, int y)
{
	(x += y) >= mod && (x -= mod);
}

inline int plu(int x, int y)
{
	add(x, y);
	return x;
}

inline int c(int x, int y)
{
	if (x < y) return 0;
	if (x == y) return 1;
	return (ll)fac[x] * inv[x - y] % mod * inv[y] % mod;
}

inline int solve(int Tx, int Mx, int R)
{
	int res = 0;
	for (int i = 0; i <= R; i++)
		if ((Mx + 1) * i <= Tx)
		{
			int v = (ll)c(R, i) * c(Tx - i * (Mx + 1) + R - 1, R - 1) % mod;
			if (i & 1) add(res, mod - v);
			else add(res, v);
		}
		else break;
	return res;
}

inline int calc(int Tx, int Ty, int Mx, int My, int R)
{
	if (Tx < 0 || Ty < 0) return 0;
	int res = 0;
	for (int i = 0; i <= R; i++)
	{
		int v = (ll)solve(Tx, Mx, i) * solve(Ty, My, i) % mod * c(R, i) % mod;
		if ((R - i) & 1) v = mod - v;
		add(res, v);
	}
	return res;
}

int main()
{
	cin >> Tx >> Ty >> Mx >> My >> R >> G >> n;
	int i, j, k, h; 
	Lim = min(Tx, Ty); Lim /= G;
	for (i = 1; i <= n; i++)
		cin >> a[i], a[i] /= G;
	sort(a + 1, a + n + 1);
	n = unique(a + 1, a + n + 1) - a - 1;
	
	fac[0] = 1;
	for (i = 1; i <= N; i++)
		fac[i] = (ll)fac[i - 1] * i % mod;
	inv[N] = ksm(fac[N], mod - 2);
	for (i = N - 1; i >= 0; i--)
		inv[i] = (ll)inv[i + 1] * (i + 1) % mod;
	
	S[0][0] = 1;
	for (i = 1; i <= R; i++)
		for (j = 1; j <= n; j++)
			for (k = 0; k <= Lim - a[j]; k++)
				add(S[i][k + a[j]], S[i - 1][k]);
	
	for (i = 0; i <= min(R, Lim); i++)
		for (j = 0; j <= Lim; j++)
			if (S[i][j]) g[i] = (g[i] + (ll)S[i][j] * 
			calc(Tx - j * G, Ty - j * G, Mx, My, R - i) % mod * c(R, i)) % mod;
	
	for (i = 0; i <= min(R, Lim); i++)
		if (i & 1) add(ans, mod - g[i]);
		else add(ans, g[i]);
	
	cout << ans << endl;
	return 0;
}