【BZOJ2655】calc（DP）（拉格朗日插值）

传送门

---

题解：

钦定大小关系，算出来后乘上阶乘就是答案。

显然有DP：$f(A,n)=Af(A-1,n-1)+f(A-1,n)$

求前缀和会将次数+1，每次n+1的时候会乘上一个A，次数会+1，$f(A,n)$是一个关于$A$的$2n$次多项式。

暴力计算前面的项，然后拉格朗日插值就行了。

---

代码：

#include<bits/stdc++.h>
#define ll long long
#define re register
#define cs const

using std::cerr;
using std::cout;

int mod;
inline int add(int a,int b){a+=b-mod;return a+(a>>31&mod);}
inline int dec(int a,int b){a-=b;return a+(a>>31&mod);}
inline int mul(int a,int b){ll r=(ll)a*b;return r>=mod?r%mod:r;}
inline int power(int a,int b,int res=1){
	for(;b;b>>=1,a=mul(a,a))(b&1)&&(res=mul(res,a));
	return res;
}
inline void Inc(int &a,int b){a+=b-mod;a+=a>>31&mod;}
inline void Dec(int &a,int b){a-=b;a+=a>>31&mod;}
inline void Mul(int &a,int b){a=mul(a,b);}

cs int N=1007,M=N<<1|1;

int y[M],fac[M],ifac[M];
int pre[M],suf[M];

inline int calc(int n,int x){
	if(x<=n)return y[x];
	pre[0]=1;for(int re i=1;i<=n;++i)pre[i]=mul(pre[i-1],dec(x,i));
	suf[n+1]=1;for(int re i=n;i>=1;--i)suf[i]=mul(suf[i+1],dec(x,i));
	int ans=0;
	for(int re i=1;i<=n;++i){
		int coef=mul(mul(pre[i-1],suf[i+1]),mul(ifac[i-1],ifac[n-i]));
		if((n^i)&1)coef=mod-coef;
		Inc(ans,mul(coef,y[i]));
	}
	return ans;
}

int f[M][N];
int n,x;
signed main(){
#ifdef zxyoi
	freopen("calc.in","r",stdin);
#endif
	scanf("%d%d%d",&x,&n,&mod);
	fac[0]=fac[1]=ifac[0]=ifac[1]=1;
	for(int re i=2;i<M;++i)fac[i]=mul(fac[i-1],i),ifac[i]=power(fac[i],mod-2);
	f[0][0]=1;
	for(int re i=1;i<=n*2+1;++i){
		f[i][0]=1;
		for(int re j=1;j<=n;++j)f[i][j]=add(mul(i,f[i-1][j-1]),f[i-1][j]);
	}
	for(int re i=1;i<=n*2+1;++i)y[i]=f[i][n];
	cout<<mul(fac[n],calc(2*n+1,x))<<"\n";
	return 0;
}