HDU 5622 KK's Chemical（dp+组合数学）

Description
我们可爱的KK有一道困难的化学题目：实验室有N(1≤N≤100)瓶化学药品，编号为0到N-1，现在KK知道第i瓶当且仅当和第c[i]瓶放在一起时会发生爆炸。KK得到任务要整理实验室，他需要将他们装进k个不同的盒子里。显然，为了KK的生命安全，你不能把两瓶会造成爆炸的药品放进同一个箱子。现在KK想知道有多少种不同的方案。由于答案可能会非常非常的大，所有将最后答案取1e9+7的模即可。保证c[i]≠i。
Input
第一行一个数T(1≤T≤200)，表示数据组数。
每组数据两行，第一行两个整数N(1≤N≤100)和K(1≤K≤1000)，表示化学药品的个数和需要放入的盒子数。
接下来第二行N个整数c[i] (0≤c[i]≤N−1)。
Output
对于每一个数据输出一个整数，表示方案数对1e9+7取模后的结果。
Sample Input
3
3 3
1 2 0
4 3
1 2 0 0
3 2
1 2 0
Sample Output
6
12
0
Solution
根据所给关系可以建一张图，问题转化成对这张图k染色，且相邻点颜色不能相同，考虑到第i瓶当且仅当和第c[i]瓶放在一起会爆炸，可知这张图是由若干连通块组成，而且每个连通块的点数都等于边数，即每个连通块只有一个环，那么只需要dfs找连通块以及连通块中的环，对于不在环中的点，每个点只需要和其相邻的点颜色不同即可，故有k-1种染色方案，对于一个环上的染色，考虑一个有i个点的环，任选一点j，如果j两端的点颜色不同，那么j本身有k-2种染色方案，其他点就是i-1个点的环的染色情况，如果j两端的点颜色相同，那么j本身有k-1种染色方案，而剩下的点可以看做是i-2个点的环的染色情况（将j两端颜色相同的点看作一个点），以dp[i]表示i个点的环的染色方案数，由上面的分析可得转移方程：dp[i]=(k-2)*dp[i-1]+(k-1)*dp[i-2]
对这个dp数组预处理，对于每张图统计每个连通块中点的数量以及环的点数即可
Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
typedef long long ll;
#define mod 1000000007ll
#define maxn 111
ll dp1[maxn],dp2[maxn];
int T,n,k,vis[maxn],cnt,res;
vector<int>v[maxn];
void init()
{
    memset(vis,0,sizeof(vis));
    for(int i=0;i<n;i++)v[i].clear();
    dp1[1]=k,dp1[2]=k*(k-1),dp1[3]=k*(k-1)*(k-2);
    for(int i=4;i<=n;i++)dp1[i]=((k-2)*dp1[i-1]%mod+(k-1)*dp1[i-2]%mod)%mod;
    dp2[0]=1;
    for(int i=1;i<=n;i++)dp2[i]=(k-1)*dp2[i-1]%mod;
}
void dfs(int u,int fa,int deep)
{
    vis[u]=deep,cnt++;
    for(int i=0;i<v[u].size();i++)
        if(v[u][i]!=fa)
        {
            if(vis[v[u][i]]&&vis[v[u][i]]<deep)
                res=deep-vis[v[u][i]]+1;
            if(!vis[v[u][i]])dfs(v[u][i],u,deep+1);
        }
}
int main()
{
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&k);
        if(k==1)
        {
            printf("%d\n",n-1?0:1);
            continue;
        }
        init();
        for(int i=0;i<n;i++)
        {
            int c;
            scanf("%d",&c);
            v[i].push_back(c),v[c].push_back(i);
        }
        ll ans=1ll;
        for(int i=0;i<n;i++)
            if(!vis[i])
            {
                cnt=res=0;
                dfs(i,-1,1);
                if(!res)res=2;
                ans=ans*dp1[res]%mod*dp2[cnt-res]%mod;
            }
        printf("%I64d\n",ans);
    }
    return 0;
}