bzoj 4487: [Jsoi2015]染色问题

先贴一个题解吧，最近懒得要死2333，可能是太弱的原因吧，总是扒题解，（甚至连题解都看不懂了），blog也没更新，GG

http://blog.csdn.net/werkeytom_ftd/article/details/52527740

容斥原理真的很神奇233

#include <bits/stdc++.h>
#define LL long long
using namespace std;

const int maxn=500;
const int mod=1e9+7;

int fac[maxn],inv[maxn];
void pre()
{
    fac[0]=1; for (int i=1; i<=400; i++) fac[i]=(LL)fac[i-1]*i%mod;
    inv[0]=inv[1]=1;
    for (int i=2; i<=400; i++) inv[i]=(LL)(mod-mod/i)*inv[mod%i]%mod;
    for (int i=2; i<=400; i++) inv[i]=(LL)inv[i]*inv[i-1]%mod;
}
int ksm(int x, int p)
{
    int sum=1;
    for (;p;p>>=1,x=(LL)x*x%mod)
        if (p&1) sum=(LL)sum*x%mod;
    return sum;
}
int C(int n, int m)
{
    return (LL)fac[n]*inv[m]%mod*inv[n-m]%mod;
}

int n,m,p,ans;
int main()
{
    cin>>n>>m>>p; pre();
    for (int i=0; i<=n; i++)
        for (int k=0; k<=p; k++)
        {
            int qwq=(LL)C(n,i)*C(p,k)%mod;
            int orz=ksm((1-ksm(k+1,i)+mod)%mod,m);
            qwq=(LL)qwq*orz%mod;
            if ((n+m+p-i-k)&1) qwq=-qwq;
            ans=(ans+qwq)%mod;
        }
    printf("%d
",(ans+mod)%mod);
    return 0;
}