UVALive 7040 (容斥)

题意:n朵花,从m种颜色中选择k种颜色使得相邻花不同色的方案数.

对于比较小的数据,这道计数题可以用DP做,DP[i][j]表示染i朵花用了j种颜色的方案数,那么对于后i-1朵花,要么用了j种颜色,要么用了j-1种颜色,转移方程也就很简单了DP[i][j] = j*(DP[i-1][j-1]+DP[i-1][j]).

但是这题的数据比较大,所以考虑容斥原理.

首先组合数递推公式:

显然要先把1-1e6的逆元求出来.

考虑k种颜色涂色时相邻不同色的方案有k*(k-1)^(n-1),这之中含有实际只用了1种颜色,2种颜色....k-1种颜色,需要把他们减去.

所以减去1种颜色没有用的情况,但是这里面包含了2种颜色没有用的情况,所以要加上2中颜色没有用的情况....这么迭代下去就是容斥,把多的减掉少的加上.

式子就是:

有个学长说我每次写代码都自带大常数,果不其然,这一份挫到卡卡过=.=

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <time.h>
#include <cmath>
using namespace std;
#define mod 1000000007
#define maxn 1111111

long long f[maxn];//逆元
void init(){
    f[1] = 1;
    for (int i = 2; i <= 1000000; i++){
        f[i] = (long long)(mod - mod/i) * f[mod % i] % mod;
    }
}

long long n, k, m;
long long c[maxn];

long long qpow (long long a, long long b) {
    a %= mod;
    if (b == 0)
        return 1;
    long long ans = qpow (a, b>>1);
    ans = ans*ans%mod;
    if (b&1)
        ans = ans*a%mod;
    return ans;
}

int main () {
    init ();
    int t, kase = 0;
    scanf ("%d", &t);
    while (t--) {
        scanf ("%lld%lld%lld", &n, &m, &k);
        printf ("Case #%d: ", ++kase);
        if (k == 1 && n == 1) {
            printf ("%lld\n", m);
            continue;
        }
        if (k == 1 && n > 1) {
            printf ("0\n");
            continue;
        }
        long long ans = m%mod;
        for (long long i = 2; i <= k; i++) {
            ans = ans*(m-i+1)%mod*f[i]%mod;
        }
        c[1] = k%mod;
        for (long long i = 2; i <= k; i++) {
            c[i] = c[i-1]*(k-i+1)%mod*f[i]%mod;
        }
        long long res = k*qpow (k-1, n-1) % mod; 
        for (long long i = 1; i < k; i++) {
            long long id = ((i&1) ? -1 : 1);
            long long cur = c[i]*(k-i)%mod*qpow (k-i-1, n-1)%mod;
            cur *= id;
            if (cur < 0)
                cur += mod;
            res += cur;
            res %= mod;
        }
        printf ("%lld\n", ans*res%mod);
    }
    return 0;
}