[ACNOI2022]张士超的钥匙

OneInDark

已于 2022-08-05 21:07:53 修改

阅读量690

点赞数 1

分类专栏： C++ 数学 # 多项式/生成函数文章标签：算法我不会做小游戏又杀穿了

于 2022-02-21 15:18:14 首次发布

本文链接：https://blog.csdn.net/qq_42101694/article/details/123047901

版权

C++ 同时被 3 个专栏收录

547 篇文章 3 订阅

订阅专栏

数学

280 篇文章 1 订阅

订阅专栏

多项式/生成函数

51 篇文章 1 订阅

订阅专栏

题目

题目背景
“想成为标杆的话，绝对没有捷径可走；成为标杆之后，更是无路可退！” $\sf DD(XYX)$ 从来没有忘记自己的使命。“所谓标杆，就是忍受一切赞誉走在所有人前面的人！”

我最大的错误就是试图与 $\sf DD(XYX)$ 相比较。而有的错误需要用一生来偿还。

题目描述
在第 $i\;(1\leqslant i\leqslant m)$ 天， $\sf OneInDark$ 有 $p_i$ 的概率参加考试，然后被 $\sf DD(XYX)$ 超越 $b_i$ 道题；也有 $1-p_i)$ 的概率不参加考试，什么事儿也不发生。若 $m$ 天过后， $\sf OneInDark$ 被 $\sf DD(XYX)$ 总共超越了至少 $n$ 道题，则称为 “太阳打东边出来”。

现在请对所有 $\{b\}\;(0\leqslant b_i\leqslant a_i)$ 满足 $\sum b_i=N$ 求出 “太阳打东边出来” 的概率之和。

数据范围与提示
$m\leqslant 100$ 但 $1\leqslant n\leqslant N\leqslant \min(10^9,\sum a_i)$ 。输入中还会给出非负整数 $d$ 使得 $d\mid(a_i+1)$ 且 $N\leqslant 100d$ 。其余的范围如 $a_i\leqslant 10^9,\;d\leqslant 10^7$ 无甚用处。输出对 $998244353$ 取模。

思路

有一个奇怪的条件 $d\mid(a_i+1)$ 。它会在哪里用上呢？

解决 $n = 1$ 或 $n = N$ 的子任务时，就发现总方案数的计算可以是 $[x^N]\prod\frac{1-x^{a_i+1}}{1-x}$ ，那么分子实际上是一个关于 $x^d$ 的多项式，长度不超过 ${N\over d}\leqslant 100$ 了。最后乘一个 $1-x)^{-m}$ ，简单组合数即可解决。

再接再厉，继续考虑生成函数。可以发见，目前的 $\sum b$ 和 “已经拿到手的钥匙” 数量是有用而独立的，于是需要写成二元生成函数
$\begin{aligned} F_i(x,y) &=\sum_{j=0}^{a_i}x^j(p_iy^j+1-p_i)\\ &=p_i\cdot {1-(xy)^{a_i+1}\over 1-xy}+(1-p_i)\cdot\frac{1-x^{a_i+1}}{1-x} \end{aligned}$

然后答案就是 $\sum_{i\geqslant n}[x^Ny^i]\prod_j F_j(x,y)$ 。这东西怎么求解呢？

显然我们想要保留分子的指数 $(a_i{\rm+}1)$ ，所以分母就应单独记录。即，记录 $\mathcal O(m)$ 个多项式（的分子），分母为 $(1-xy)^i\cdot(1-x)^{m-i}$ 。

那么我们顺序做乘法，得到的结果等价于求出了 $f (a, b, c, d)$ 表示分母为 $1-xy)^a(1-x)^b$ 时 $x^cy^d$ 的系数。由于是 $\mathcal O(1)$ 转移，复杂度当然是状态数 $\mathcal O[m^2({N\over d})^2]$ 。

最后需要除以 $1-xy)^a(1-x)^b$ ，等价于乘上 $\sum_{i\geqslant 0}{i+a-1\choose i}(xy)^i$ 和 $\sum_{i\geqslant 0}{i+b-1\choose i}x^i$ ，即乘上
$\sum_{i\geqslant 0}x^i\sum_{j=0}^{i}{j+a-1\choose j}{i-j+b-1\choose i-j}y^j$

不幸的是，我们要计算 $\sum_{i\geqslant n}[y^i]F(y)$ 。也就是说， $\mathcal O(m^3)$ 枚举 $a, b$ 和 $x, y$ 的指数后，我们面临的问题实际上是
$\sum_{j=l}^{i}{j+a-1\choose j}{i-j+b-1\choose i-j}$

因为 $y$ 的指数不小于某个阈值 $l$ 即可。由于 $i$ 的范围是 $\mathcal O(n)$ 的，绝对不能这样计算。怎么办？

忠告：组合数应当递推计算。譬如这道题。尽管这好像并不很有逻辑。

上式也可写为 $\sum_{j=0}^{i-l}{j+l+a-1\choose j+l}{i-l-j+b-1\choose i-l-j}$ ，为了递推的方便，不妨简记
$S(n_1,m_1)=\sum_{j=0}^{n_2}{j+n_1+m_1\choose j+n_1}{n_2-j+m_2\choose n_2-j}$

也就是用 $n_1=l,\;m_1=a-1,\;n_2=i-l,\;m_2=b-1$ 替换了一下。这里没有将 $m_2,n_2$ 设为自变量，因为后面的递推中二者是不变的。

最简单的方法：利用帕斯卡法则 减小上指标。比如将 ${j+n_1+m_1\choose j+n_1}$ 用 ${j+n_1+m_1-1\choose j+n_1}+{j+n_1+m_1-1\choose j+n_1-1}$ 换掉，前者就得到 $S(n_1,m_1{\rm-}1)$ ，而后者得到 $S(n_1{\rm-}1,m_1)$ 。这相当于在坐标系内，沿着四连通的边移动；只需要任选一个边界。由于 $(j{\rm+}n_1)$ 是下指标，必然有 $n_1\geqslant 0$ ，取边界 $n_1=0$ 。但 $m_1$ 可以取任意值。观察发现， $m_1=-1$ 可以使得计算量最小。

$n_1=0$ ，则
$\begin{aligned} S(0,m_1) &=\sum_{j=0}^{n_2}{j+m_1\choose j}{n_2-j+m_2\choose n_2-j}\\ &=\sum_{j=0}^{n_2}{j+m_1\choose m_1}{n_2-j+m_2\choose m_2}\\ &={m_1+m_2+n_2+1\choose m_1+m_2+1} \end{aligned}$

最后一步可以由 $\text{constructive}$ 的组合意义得到：在 $(j{\rm+}m_1)$ 元素后，插入一个被选择元素。或者，从第一行直接交换上下标得到简单的二项式系数卷积恒等式。

$m_1=-1$ ，则
$S(n_1,-1) =\sum_{j=0}^{n_2}{j+n_1-1\choose j+n_1}{n_2-j+m_2\choose n_2-j}\\ =[n_1=0]{n_2+m_2\choose n_2}$

因为左式只在 $j=n_1=0$ 时，得出非零值 $1$ 。

枚举边界值，考虑其对 $S(n_1,m_1)$ 的贡献，则
$\begin{aligned} S(n_1,m_1) &=\sum_{i=1}^{n_1}S(i,-1){n_1+m_1-i\choose n_1-i} +\sum_{j=0}^{m_1}S(0,j){n_1+m_1-j-1\choose m_1-j}\\ &=\sum_{j=0}^{m_1}{j+m_2+n_2+1\choose j+m_2+1}{n_1+m_1-j-1\choose m_1-j} \end{aligned}$

注意 “坐标系内行走” 的系数，是 $(1,j)\to(n_1,m_1)$ ，因为不能再经过 $(0,j{\rm+}1)$ 了。

这是一个 $\mathcal O(m)$ 的枚举，所以总时间复杂度 $\mathcal O(m^4)$ ， $\text{case closed}$ ！

再闲聊两句：原式 $\sum_{j=l}^{i}{j+a-1\choose j}{i-j+b-1\choose i-j}$ 有组合意义，即前 $a$ 个盒子放 $j$ 个球、后 $b$ 个盒子放 $(i{\rm-}j)$ 个球、可以空盒的方案数。唯一的限制是 $j\geqslant l$ ，即前 $a$ 个盒子里放了至少 $l$ 个球。

高明的想法：转而枚举第 $l$ 个球的位置。设为第 $j\;(j\geqslant a)$ 个盒子，则前 $(l{\rm-}1)$ 个球也在这 $j$ 个盒子中；后 $(i{\rm-}l)$ 个球则在后 $[j,a{\rm+}b]$ 号盒子中，于是
$\sum_{j=a}^{a+b}{l-1+j-1\choose j-1}{a+b-j+1+i-l-1\choose a+b-j+1-1}$

你会发现，它和 $S(l,a{\rm-}1)$ 完全相同，只不过 $S(l,a{\rm-}1)$ 中枚举的 $j$ 是这里的 $(a - j)$ 罢了。所以上面的代数推演都没必要，组合意义有时比数学推导更好用。倒不如说，只是因为它恰好有这样一个组合意义，数学推导才可能得到一个简单的式子呢！

代码

#include <cstdio> // JZM yydJUNK!!!
#include <iostream> // XJX yyds!!!
#include <algorithm> // XYX yydLONELY!!!
#include <cstring> // (the STRONG long for LONELINESS)
#include <cctype> // ZXY yydSISTER!!!
using namespace std;
# define rep(i,a,b) for(int i=(a); i<=(b); ++i)
# define drep(i,a,b) for(int i=(a); i>=(b); --i)
typedef long long llong;
inline int readint(){
	int a = 0; int c = getchar(), f = 1;
	for(; !isdigit(c); c=getchar())
		if(c == '-') f = -f;
	for(; isdigit(c); c=getchar())
		a = (a<<3)+(a<<1)+(c^48);
	return a*f;
}

const int MAXN = 105, MOD = 998244353;
int inv[MAXN];
inline int getC(int n, int m){
	llong c = 1;
	for(int i=0; i!=m; ++i)
		c = c*(n-i)%MOD*inv[i+1]%MOD;
	return int(c);
}
inline int calc(int n1, int m1, int n2, int m2){
	if(m1 == -1) return n1 ? 0 : int(getC(n2+m2,m2)); // m2 != -1
	static int t1[MAXN], t2[MAXN]; int res = 0;
	t1[0] = 1, t2[0] = getC(m2+n2+1,m2+1);
	rep(j,1,m1) t1[j] = int(llong(n1+j-1)*t1[j-1]%MOD*inv[j]%MOD);
	rep(j,1,m1) t2[j] = int(llong(m2+n2+1+j)*t2[j-1]%MOD*inv[j+m2+1]%MOD);
	rep(j,0,m1) res = int((res+llong(t1[j])*t2[m1-j])%MOD);
	return res;
}

int dp[2][MAXN][MAXN][MAXN];
int main(){
	int m = readint(), d = readint();
	int N = readint(), n = readint();
	rep(i,(inv[1]=1)+1,m) inv[i] = int(
		llong(MOD-MOD/i)*inv[MOD%i]%MOD);

	const int Nd = N/d;
	****dp = 1; // x^0 y^0
	for(int i=1,fr=0,a; i<=m; ++i,fr^=1){
		a = (readint()+1)/d; llong p = readint();
		rep(j,0,i) rep(x,0,Nd) rep(y,0,x){
			int &v = dp[i&1][j][x][y] = int( (
				(x < a ? 0 : (p-1)*dp[fr][j][x-a][y])
				+ (MOD+1-p)*dp[fr][j][x][y] ) %MOD );
			if(!j) break; // cannot choose it here
			v = int((v+p*dp[fr][j-1][x][y])%MOD);
			if(x >= a && y >= a) v = int((v+
				(MOD-p)*dp[fr][j-1][x-a][y-a])%MOD);
		}
	}

	int ans = 0;
	rep(a,0,m) rep(x,0,Nd) rep(y,0,x) if(dp[m&1][a][x][y]){
		const int l = std::max(0,n-y*d);
		if(l > N-x*d) continue; // invalid
		ans = int((ans+llong(dp[m&1][a][x][y])
			*calc(l,a-1,N-x*d-l,m-a-1))%MOD);
	}
	printf("%d\n",ans);
	return 0;
}