CodeForces 111 D.Petya and Coloring（组合数学+dp）

Description

用$k$种颜色去给一个$n\times m$的棋盘，要求以任一非边界的列分界线将棋盘分成两块后，这两块中颜色数量相等，问方案数

Input

三个整数$n,m,k(1\le n,m\le 1000,1\le k\le 10^6)$

Output

输出方案数，结果模$10^9+7$

Sample Input

2 2 1

Sample Output

1

Solution

令$S_i$为第$i$列的颜色集，则由条件知$|S_1\cup S_2\cup...\cup S_j|=|S_{j+1}\cup S_{j+2}\cup ...\cup S_m|,j=2,...,m-1$

进而得到$S_2,S_3,...,S_{m-1}\in S_1,S_m$且$|S_1|=|S_m|$

首先枚举$|S_1\cap S_m|$的大小$i$，以及两边两列的颜色数$j$，那么两列会用$j$种颜色染，中间$m-2$列只能用$i$种颜色，故方案数为$\sum\limits_{i=1}^n\sum\limits_{j=i}^nC_k^iC_{k-i}^{2(j-i)}C_{2(j-i)}^{j-i}i^{n(m-2)}f^2[j]$

其中$f[i]$为用$i$种颜色给$n$个格子染色的方案数，显然$f[1]=1,f[i]=i^n-\sum\limits_{j=1}^{i-1}C_i^jf[j],i\ge 2$

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int INF=0x3f3f3f3f,maxn=1000005;
#define mod 1000000007
int fact[maxn],inv[maxn];
void init(int n=1e6)
{
    fact[0]=1;
    for(int i=1;i<=n;i++)fact[i]=(ll)i*fact[i-1]%mod;
    inv[1]=1;
    for(int i=2;i<=n;i++)inv[i]=mod-(ll)(mod/i)*inv[mod%i]%mod;
    inv[0]=1;
    for(int i=1;i<=n;i++)inv[i]=(ll)inv[i-1]*inv[i]%mod;
}
int C(int n,int m)
{
    if(m>n)return 0;
    return (ll)fact[n]*inv[m]%mod*inv[n-m]%mod;
}
int Pow(int a,int b)
{
    int ans=1;
    while(b)
    {
        if(b&1)ans=(ll)ans*a%mod;
        a=(ll)a*a%mod;
        b>>=1;
    }
    return ans;
}
int f[1005];
int main()
{
    init();
    int n,m,k;
    while(~scanf("%d%d%d",&n,&m,&k))
    {
        if(m==1)
        {
            printf("%d\n",Pow(k,n));
            continue;
        }
        f[1]=1;
        for(int i=2;i<=n;i++)
        {
            f[i]=Pow(i,n);
            for(int j=1;j<i;j++)
            {
                f[i]-=(ll)C(i,j)*f[j]%mod;
                if(f[i]<0)f[i]+=mod; 
            }
        }
        int ans=0;
        for(int i=0;i<=n;i++)
        {
            int res=0;
            for(int j=i;j<=n;j++)
            {
                int temp=(ll)f[j]*f[j]%mod;
                temp=(ll)temp*C(k-i,2*(j-i))%mod*C(2*(j-i),j-i)%mod;
                res+=temp;
                if(res>=mod)res-=mod; 
            }
            res=(ll)res*C(k,i)%mod*Pow(i,n*(m-2))%mod;
            ans+=res;
            if(ans>=mod)ans-=mod;
        }
        printf("%d\n",ans);
    }
    return 0;
}