[CSAcademy]Sum of Powers

[CSAcademy]Sum of Powers

题目大意：

给定(n,m,k(n,m,kle4096))。一个无序可重集(A)为合法的，当且仅当(|A|=m)且(sum A_i=n)。定义一个集合的贡献为(sum A_i^k)，求所有满足条件的集合的贡献之和。

思路：

(f[i][j])表示将(j)个数之和为(i)的方案数，有如下两种转移：

1. (f[i][j]+=f[i-1][j-1])，表示新加入一个元素(1)；
2. (f[i][j]+=f[i-j][j])，表示集合内每个元素(+1)。

可以证明这样就不重复、不遗漏地包含了所有的集合。

由于每个元素的贡献独立，最后枚举每种元素及其出现次数并计算贡献即可。

时间复杂度(mathcal O(nm))。

源代码：

#include<cstdio>
#include<cctype>
inline int getint() {
	register char ch;
	while(!isdigit(ch=getchar()));
	register int x=ch^'0';
	while(isdigit(ch=getchar())) x=(((x<<2)+x)<<1)+(ch^'0');
	return x;
}
const int N=4097,mod=1e9+7;
int f[N][N];
inline int power(int a,int k) {
	int ret=1;
	for(;k;k>>=1) {
		if(k&1) ret=1ll*ret*a%mod;
		a=1ll*a*a%mod;
	}
	return ret;
}
int main() {
	const int n=getint(),m=getint(),k=getint();
	f[0][0]=1;
	for(register int i=1;i<=n;i++) {
		for(register int j=1;j<=m&&j<=i;j++) {
			f[i][j]=(f[i-1][j-1]+f[i-j][j])%mod;
		}
	}
	int ans=0;
	for(register int i=1;i<=n-m+1;i++) {
		const int pwr=power(i,k);
		for(register int j=1;j<=m&&i*j<=n;j++) {
			(ans+=1ll*pwr*f[n-i*j][m-j]%mod)%=mod;
		}
	}
	printf("%d
",ans);
	return 0;
}