【LOJ577】「LibreOJ NOI Round #2」简单算术（Lucas定理）（DP）

传送门

---

题解：

其实这道题还挺简单的。。。

首先考虑如何表示$m$次幂的$k$次项，显然是若干项的乘积之和。

考虑枚举拆分方案，设$i$次项出现了$b_i$次，则$\sum _{i=0}^{n}i{b}_i=k,\sum _{i=0}^n{b}_i=m$，并且这个拆分对应的排列数量为$\prod _{i=0}^n\left(\genfrac{}{}{0ex}{}{m-\sum _{j=0}^{i-1}{b}_j}{b_i}\right)$

设已知多项式为$A(x)$，我们要求的是$[x^k]A(x{)}^m$。当$p\mid m$，由Lucas定理，上面对应的排列数量非$0$当且仅当所有$p\mid b_i$成立，则$p\mid k$。

那么在$p\mid m$的情况下我们有：

$$[x^k]A(x{)}^m=\{\begin{array}{rlrlr}& [x^{k/p}]A(x{)}^{m/p}& & & p\mid k\\ & 0& & & p\nmid k\end{array}$$

更一般的，设$m=ap+b,0\le b\le p-1$，我们有如下转移式子：

$$[x^k]A(x{)}^{ap+b}=\sum _{i\equiv k(\mathrm{m}\mathrm{o}\mathrm{d}p)}[x^{\frac{k-i}{p}}]A(x{)}^a\cdot [x^i]A(x{)}^b$$

预处理$A(x)$的$0$到$p-1$次幂，然后记忆化搜索即可。

复杂度$O(n^2{p}^2+T{n}^2{\log }_{p}m)$

---

代码：

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

int n,m,p;
int a[55],b[55][3005],len[55];
cs int M=19260817;
std::unordered_map<ll,int> ma;

int F(ll m,ll k){
	if(!m)return !k;
	ll id=m*M+k;if(ma.count(id))return ma[id];
	int ans=0,a=m%p;ll c=m/p,d=k/p;
	for(int re i=k%p;i<len[a]&&i<=k;i+=p,--d)
	if(b[a][i])ans+=F(c,d)*b[a][i];
	return ma[id]=ans%p;
}

signed main(){
#ifdef zxyoi
	freopen("calc.in","r",stdin);
#endif
	scanf("%d%d",&n,&p);
	for(int re i=0;i<=n;++i)scanf("%d",a+i);
	b[0][0]=len[0]=1;
	for(int re i=1;i<p;++i){len[i]=len[i-1]+n;
		for(int re j=0;j<len[i];++j){ll val=0;
			for(int re k=0;k<=std::min(n,j);++k)val+=a[k]*b[i-1][j-k];
			b[i][j]=val%p;
		}
	}
	int T;scanf("%d",&T);
	while(T--){
		ll m,k;scanf("%lld%lld",&m,&k);
		std::cout<<F(m,k)<<"\n";
	}
	return 0;
}