Wannafly挑战赛24D-无限手套【dp】【数学】

---

题目：

传送门

---

题意：

有$m$种物品，数量无限，用$(a_i,b_i,x_i)$三元组来形容物品，$x_i$表示我们选出该种物品多少个
对答案的贡献为${\prod }_{i=1}^{n}1+a_i{x}_i^2+b_i{x}_i$
求所有方案的贡献和

---

分析：

设$f_{i,j}$表示考虑到第$i$位已经考虑完前$j$种物品
会有$f_{i,j}={\sum }_{k=0}^i{f}_{i-k,j-1}\ast (a_j{k}^2+b_{j}k+1)$
此时复杂度为$O(n^{2}m)$
我们不妨将方程逆过来推
$f_{i,j}\ast (a_j{k}^2+b_{j}k+1)\to f_{i+k,j+1}$
式子的左边展开得
$f_{i,j}\ast a_j{k}^2+f_{i,j}\ast b_{j}k+f_{i,j}$
我们考虑将三个单项式分开求出最后合并
阔以运用前缀和的芝士，将$k^2,k,1$分别用三遍，两遍，一遍前缀和来代替这个系数
其他照搬式子
复杂度就来到优秀的$O(nm)$了
当然还有更秀的生成函数，我不会，爬了

---

代码：

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>  
#include<string>
#include<algorithm>
#include<queue>
#define LL long long
#define mo 998244353
using namespace std;
inline LL read() {
    LL d=0,f=1;char s=getchar();
    while(s<'0'||s>'9'){if(s=='-')f=-1;s=getchar();}
    while(s>='0'&&s<='9'){d=d*10+s-'0';s=getchar();}
    return d*f;
}
LL a[1005],b[1005];
LL sum1[10005],sum2[10005],sum3[10005],f[10005][1005];
int main()
{
	LL m=read();
	for(LL i=1;i<=m;i++) a[i]=read(),b[i]=read();
	f[0][0]=1;
	for(LL j=0;j<m;j++)
	{
		memset(sum1,0,sizeof(sum1));
		memset(sum2,0,sizeof(sum2));
		memset(sum3,0,sizeof(sum3));
		for(LL i=0;i<=10000;i++) 
		{
			(sum1[i]+=f[i][j])%=mo;
			(sum2[i+1]+=f[i][j]*b[j+1]%mo)%=mo;
			(sum3[i+1]+=f[i][j]%mo*a[j+1]%mo)%=mo;
			(sum3[i+2]+=f[i][j]%mo*a[j+1]%mo)%=mo;
		}
		for(LL i=1;i<=10000;i++) (sum1[i]+=sum1[i-1])%=mo;
		for(LL i=1;i<=10000;i++) (sum2[i]+=sum2[i-1])%=mo;
		for(LL i=1;i<=10000;i++) (sum2[i]+=sum2[i-1])%=mo;
		for(LL i=1;i<=10000;i++) (sum3[i]+=sum3[i-1])%=mo;
		for(LL i=1;i<=10000;i++) (sum3[i]+=sum3[i-1])%=mo;
		for(LL i=1;i<=10000;i++) (sum3[i]+=sum3[i-1])%=mo;
		for(LL i=0;i<=10000;i++) f[i][j+1]=(sum1[i]+sum2[i]+sum3[i])%mo;
	}
	LL q=read();
	while(q--) printf("%lld\n",f[read()][m]);
	return 0;
}