CF gym101933 K King's Colors——二项式反演

题目：http://codeforces.com/gym/101933/problem/K

每个点只要和父亲不同色就行。所以 “至多 i 种颜色” 的方案数就是 i * ( i-1 )^n-1 。

#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const int N=2505,mod=1e9+7;
int n,k,g[N],c[N][N];
void upd(int &x){x>=mod?x-=mod:0;}
int pw(int x,int k)
{int ret=1;while(k){if(k&1)ret=(ll)ret*x%mod;x=(ll)x*x%mod;k>>=1;}return ret;}
int main()
{
  scanf("%d%d",&n,&k);
  for(int i=1,d;i<n;i++)scanf("%d",&d);
  for(int i=0;i<=n;i++)c[i][0]=1;
  for(int i=1;i<=k;i++)
    for(int j=1;j<=i;j++)c[i][j]=c[i-1][j]+c[i-1][j-1],upd(c[i][j]);
  for(int i=2;i<=k;i++)g[i]=(ll)i*pw(i-1,n-1)%mod;
  int ans=0;
  for(int i=0,j=(k&1?-1:1);i<=k;i++,j=-j)
    ans=(ans+(ll)j*c[k][i]*g[i])%mod+mod,upd(ans);
  printf("%d\n",ans);
  return 0;
}