Wannafly挑战赛24 D 无限手套

$Result$

---

$Hyperlink$

https://ac.nowcoder.com/acm/contest/186/D

---

$Description$

一个无限手套，有$n$个槽，第$i$个槽有属性$a_i,b_i$
若手套第$i$个槽装上了$x$颗宝石，这个槽会产生$(a_i{x}^2+b_{i}x+1)$的能量
无限手套的能量是它每个槽上面的能量的积

$q$个询问，每个询问给定一个$m$，询问如果有$m$个宝石，所有情况产生的能量的和是多少？
答案对998244353取模

数据范围：$n\le 10^3,m\le 10^4$

---

$Solution$

据说有生成函数的做法，但是我不会【我太菜了】
设$f_{i,j}$表示处理到第$i$个槽，$j$颗宝石能产生的能量之和

则我们显然是枚举上一个槽选了多少个宝石$k$，有转移
$f_{i,j}={\sum }_{k=0}^j{f}_{i-1,j}\times y(j-k)$，$y(x)$表示在第$i$个槽下装$x$个宝石会产生的能量

直接这样做时间复杂度为$O(n{m}^2)$显然会$T$，我们考虑优化它

观察$f_{i,j}$的式子可以发现，实际上它和上一次枚举的$f_{i,j-1}$只有两个地方不同，我们写出来看一下
$f_{i,j}={\sum }_{k=0}^j{f}_{i-1,j}\times y(j-k)$
$f_{i,j-1}={\sum }_{k=0}^{j-1}{f}_{i-1,j}\times y(j-k-1)$

$f_{i,j}$在求和相差的那一部分多枚举了一个$k=j$的情况
这种情况的差值显然就是
$A=f_{i-1,j}\times y(j-k)=f_{i-1,j}$

考虑计算求和中间那一段的差值，试着上下相减，看一下每一项的差值
$f_{i-1,j}\times y(j-k)-f_{i-1,j}\times y(j-k-1)$

分配率
$f_{i-1,j}\times [y(j-k)-y(j-k-1)]$

把$y$拆开
$f_{i-1,j}\times [a_i(j-k{)}^2+b_i(j-k)-a_i(j-k-1{)}^2-b_i(j-k-1)]$（常数项显然抵消掉了）
由$x^2-(x-1{)}^2=2x-1$，可以得到

$f_{i-1,j}\times [a_i(2j-2k-1)+b_i]$

考虑把与$j$相关的写一起，与$k$相关的写一起
得到了第二部分的差值
$B=f_{i-1,j}\times [b_i+(2j-1)a_i-2k{a}_i]$

$b_i,a_i$是常量，换言之这些是我们已知的
也就是每一个$f_{i,j}$都和$f_{i,j-1}$差了$A+B$

$A$显然很好求，考虑求$B$

对于每一个$k$，都会有$f_{i-1,j}[b_i+(2j-1)a_i-2k{a}_i]$的差值

观察到我们始终只关心$f_i$和$f_i-1$，为了方便我们把它们分别设成$f$和$g$【这次的和上一次的】
则$B=g_j[b_i+(2j-1)a_i-2k{a}_i]$
拆成两坨$B=g_j[b_i+(2j-1)a_i]-g_i\times 2k{a}_i$
记录前缀和$s_j={\sum }_{k=0}^j{g}_j,S_k={\sum }_{j=0}^k{g}_j\times 2j$

这样就可以从$f_{j-1}$快速的转移到$f_j$了，因为$f_j$和$f_{j-1}$的差值就是$A+B$

即$f_j=f_{j-1}+A+B$，这是笼统的总式子，接下来我们把它写细一点
首先$A$是以前的$f_{i-1,j}$，现在换了$dp$后是$g_j$
$B$可以表示成两段前缀和，即$s_{j-1}\times (b_i+(2j-1)a_i)-S_{j-1}\times a_i$【之所以是$j-1$是因为$f_j-1$只计算到$j-1$，所以这一部分的差值应该是到$j-1$】

写成真正的$dp$转移方程
$f_j=f_{j-1}+g_j+s_{j-1}\times (b_i+(2j-1)a_i)-S_{j-1}\times a_i$【前面有一个$f_{j-1}$是因为我们后面算的是和上一次的差值】
有点丑，将常量$b_i+(2j-1)a_i$用变量$tmp$表示起来，就舒服多了

$f_j=f_{j-1}+g_j+s_{j-1}\times tmp-S_{j-1}\times a_i$

最后答案就是$f_m$，时间复杂度：$O(nm)$

---

$Code$

#include<cctype>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define LL long long
#define N 10001
#define mod 998244353
using namespace std;int n,q;
LL a,b,f[N],g[N],s[N],S[N],tmp;
inline LL read()
{
	char c;LL d=1,f=0;
	while(c=getchar(),!isdigit(c)) if(c=='-') d=-1;f=(f<<3)+(f<<1)+c-48;
	while(c=getchar(),isdigit(c)) f=(f<<3)+(f<<1)+c-48;
	return d*f;
}
signed main()
{
	f[0]=g[0]=s[0]=1;n=read();
	for(register int i=1;i<=n;i++)
	{
		a=read();b=read();
		for(register int j=1;j<N;j++) (s[j]=s[j-1]+g[j])%=mod,(S[j]=S[j-1]+2*g[j]*j%mod)%=mod;
		for(register int j=1;j<N;j++) tmp=(b+(2*j-1)*a%mod)%mod,f[j]=(f[j-1]+g[j]+(s[j-1]*tmp%mod-S[j-1]*a%mod+mod)%mod)%mod;
		for(register int j=1;j<N;j++) g[j]=f[j];
	}
	q=read();
	while(q--) printf("%lld\n",f[read()]);
}