[DP] ZROI 2017 提高6 T2 异或统计

这题如果能知道一个答案的式子就很简单了,设 $g[i][j]$ 表示取 $i$ 个数, 加和为 $j$ 的方案数。g数组直接 $DP$ 出来就好了。则最终的答案就为：

$$\sum_{i=1}^n i^m \sum_j g[K-j][n-i*j]$$
为什么是对的呢？注意到 $g[K-j][n-i*j]$ 的意义是至少取了 $j$ 个 $i$ 的方案数，贡献了一个 $i^m$ 。

考虑一个方案，若$i$ 选了$t$ 个，则这个方案满足：$i$ 至少取1个，至少取2个…，至少取t个，刚好被算了$t$ 次。

以前没见过，感觉很神奇。这大概是一种套路吧，这里先记一下。

#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn=5005,MOD=1e9+7;
typedef long long LL;
int n,m,K,g[maxn][maxn],ans;
int Pow(LL a,int b){
    LL res=1;
    for(;b;b>>=1,a=a*a%MOD) if(b&1) res=(res*a)%MOD;
    return res;
}
int main(){ 
    scanf("%d%d%d",&n,&K,&m);
    g[1][1]=1; 
    for(int i=1;i<=K;i++) 
     for(int j=1;j<=n;j++) if(g[i][j]){
        (g[i+1][j+1]+=g[i][j])%=MOD;
        if(i+j<=n) (g[i][i+j]+=g[i][j])%=MOD;
      }
    for(int i=1;i<=n;i++){
        int res=0;
        for(int j=1;(LL)i*j<=n&&j<=K;j++) (res+=g[K-j][n-i*j])%=MOD;
        (ans+=(LL)res*Pow(i,m)%MOD)%=MOD;  
    }
    printf("%d\n",ans);
    return 0;
}