K - King's Colors Gym - 101933K 容斥原理水题..

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

 

题意：

        给你一棵树，问用恰好k种颜色将其涂成两个相邻点颜色不同的方法有多少种。

 

用法：

       如果是一个环，情况就会比较不一样，但是这里是树，所以不会有最后一个点和第一个点的链接出问题的情况。那么当不是恰好K种而是最多K种时，方法数就是,因为第一个点有K种涂法，之后的n-1个点每一个只要和前一个不同就好，那么就是k-1种涂法。但是这些方法里包含了恰好用k-1,k-2,k-3...种涂法所以要依此加减，容斥的加减原因就不说了，自己百度吧。想到了就很好做了，其实是一道水题啊。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll mod=(ll)1e9+7;
ll C[2505][2505],n,k;
ll quick(ll a,ll b){
    ll ans=1;
    while(b){
        if(b&1) ans=ans*a%mod;
        a=a*a%mod;
        b>>=1;
    }
    return ans;
}
void init(){
    for(int i=0;i<=2500;i++)C[i][0]=1,C[i][1]=i;
    for(int i=1;i<=2500;i++)
        for(int j=2;j<=i;j++)
            C[i][j]=(C[i-1][j]+C[i-1][j-1])%mod;
}

int main(){
    init();
    scanf("%lld%lld",&n,&k);
    for(int i=1;i<n;i++){
        ll x;scanf("%lld",&x);
    }
    ll ans=0,flag=1;
    for(int i=k;i>=1;i--){
        ll aim=flag*i*quick(i-1,n-1)%mod*C[k][i];
        ans=(ans+aim+mod)%mod;
        flag=-flag;
    }
    printf("%lld\n",ans);

    return 0;
}