[JSOI2015] bzoj 4487 染色问题 - 容斥

连着三个容斥最后一个用二项式定理缩起来即可。
(或者只容斥两层亦可)

#include
#include
#include
#include
#define lint long long
#define mod 1000000007
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define N 410
using namespace std;
int mi[N*N],C[N][N];
inline int fast_pow(int x,int k,int ans=1)
{   for(;k;k>>=1,x=(lint)x*x%mod) (k&1)?ans=(lint)ans*x%mod:0;return ans; }
inline int sol(int x,int s) { return (s&1)?(x?mod-x:0):x; }
inline int F(int c,int n,int m)
{
    mi[0]=1;rep(i,1,n*m) mi[i]=mi[i-1]*(c+1ll)%mod;lint ans=0ll,t;
    rep(i,0,n) t=mi[n-i]-1,(t<0?t+=mod:0),ans+=sol((lint)C[n][i]*fast_pow(t,m)%mod,i);
    return (int)(ans%mod);
}
inline int preludec(int n)
{
    rep(i,0,n) C[i][0]=1;
    rep(i,1,n) rep(j,1,i) C[i][j]=C[i-1][j]+C[i-1][j-1],(C[i][j]>=mod?C[i][j]-=mod:0);
    return 0;
}
int main()
{
    int n,m,c;scanf("%d%d%d",&n,&m,&c);lint ans=0ll;
    preludec(max(max(n,m),c));
    rep(i,0,c) ans+=sol((lint)C[c][i]*F(c-i,n,m)%mod,i);
    return !printf("%d\n",(int)(ans%mod));
}

你可能感兴趣的:(组合计数,容斥原理)