CS Academy 32 Sum of Powers 动态规划

题意

考虑所有的正整数可重集$\{a_1,a_2,a_3...a_k\}$，满足$a_1+a_2...+a_k=n$，求所有$a_1^k+a_2^k...+a_k^m$的和。
$n,m,k<=4096$

分析

先做一次整数划分dp，也就是f[i,j]表示选了i个数和为j的方案。
然后分别考虑$[1,n]$中的每一个数对答案的贡献。
不难发现数$i$在答案中的系数为$\sum_ji至少出现了j次的可重集数量$。
也就是$\sum_jf[k-j,n-i*j]$。
时间复杂度$O(n^2)$

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;

typedef long long LL;

const int N=4105;
const int MOD=1000000007;

int n,m,k,f[N][N];

void mod(int &x) {x-=x>=MOD?MOD:0;}

int ksm(int x,int y)
{
    int ans=1;
    while (y)
    {
        if (y&1) ans=(LL)ans*x%MOD;
        x=(LL)x*x%MOD;y>>=1;
    }
    return ans;
}

int main()
{
    scanf("%d%d%d",&n,&k,&m);
    f[0][0]=1;
    for (int i=1;i<=k;i++)
        for (int j=i;j<=n;j++)
            mod(f[i][j]=f[i-1][j-1]+f[i][j-i]);
    int ans=0;
    for (int i=1;i<=n;i++)
    {
        int s=0;
        for (int j=1;j<=k&&j*i<=n;j++) mod(s+=f[k-j][n-i*j]);
        mod(ans+=(LL)s*ksm(i,m)%MOD);
    }
    printf("%d",ans);
    return 0;
}