2014ACM/ICPC亚洲区西安站 F题 color (组合数学,容斥原理)

传送门:https://codeforces.com/gym/100548/attachments

题意:

有n个点m种颜料,问你用k种颜色,有多少种方法,使得相邻两点的颜色不一样

题解:

容斥+组合数

我们很容易知道,用t种颜色去涂n个格子,方案数为

[t*(t-1)^{n-1} ]

并不是,因为我们可以这样想

如果我选用最多的五种颜色1,2,3,4,5去染色的话,我们可以会染出1,2,1,2这样的结果

而如果只用两种颜色1,2去染色的话,我们也会染出1,2,1,2这样的结果

因此我们的答案是错误的

也就是说,用k种颜色去染色的时候,我们的结果是包含了k,k-1,,,1种颜色染色的结果的

因此我们需要容斥一下

这个容斥的证明就是传说中的二项式反演

关于二项式反演:[https://www.cnblogs.com/buerdepepeqi/p/10908022.html]

[那么设 f_i为恰好使用了 i 种颜色的方案数,可以得到\ k(k−1)^{n−1}=sum_{i=0}^{k}C_{k}^{i}fi\ 经过反演得到 f_k=sum_{i=2}^{k}(-1)^{k+i}C_{k}^{i}i(i-1)^{n-1} ]

/**
 *        ┏┓    ┏┓
 *        ┏┛┗━━━━━━━┛┗━━━┓
 *        ┃       ┃  
 *        ┃   ━    ┃
 *        ┃ >   < ┃
 *        ┃       ┃
 *        ┃... ⌒ ...  ┃
 *        ┃       ┃
 *        ┗━┓   ┏━┛
 *          ┃   ┃ Code is far away from bug with the animal protecting          
 *          ┃   ┃   神兽保佑,代码无bug
 *          ┃   ┃           
 *          ┃   ┃        
 *          ┃   ┃
 *          ┃   ┃           
 *          ┃   ┗━━━┓
 *          ┃       ┣┓
 *          ┃       ┏┛
 *          ┗┓┓┏━┳┓┏┛
 *           ┃┫┫ ┃┫┫
 *           ┗┻┛ ┗┻┛
 */
// warm heart, wagging tail,and a smile just for you!
//
//                            _ooOoo_
//                           o8888888o
//                           88" . "88
//                           (| -_- |)
//                           O  =  /O
//                        ____/`---'\____
//                      .'  |     |//  `.
//                     /  |||  :  |||//  
//                    /  _||||| -:- |||||-  
//                    |   | \  -  /// |   |
//                    | \_|  ''---/''  |   |
//                      .-\__  `-`  ___/-. /
//                  ___`. .'  /--.--  `. . __
//               ."" '<  `.___\_<|>_/___.'  >'"".
//              | | :  `- \`.;` _ /`;.`/ - ` : | |
//                 `-.   \_ __ /__ _/   .-` /  /
//         ======`-.____`-.___\_____/___.-`____.-'======
//                            `=---='
//        ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
//                     佛祖保佑      永无BUG
#include <set>
#include <map>
#include <deque>
#include <queue>
#include <stack>
#include <cmath>
#include <ctime>
#include <bitset>
#include <cstdio>
#include <string>
#include <vector>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;

typedef long long LL;
typedef long long ll;
typedef pair<LL, LL> pLL;
typedef pair<LL, int> pLi;
typedef pair<int, LL> pil;;
typedef pair<int, int> pii;
typedef unsigned long long uLL;
#define ls rt<<1
#define rs rt<<1|1
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define bug printf("*********
")
#define FIN freopen("input.txt","r",stdin);
#define FON freopen("output.txt","w+",stdout);
#define IO ios::sync_with_stdio(false),cin.tie(0)
#define debug1(x) cout<<"["<<#x<<" "<<(x)<<"]
"
#define debug2(x,y) cout<<"["<<#x<<" "<<(x)<<" "<<#y<<" "<<(y)<<"]
"
#define debug3(x,y,z) cout<<"["<<#x<<" "<<(x)<<" "<<#y<<" "<<(y)<<" "<<#z<<" "<<z<<"]
"

const double eps = 1e-8;
const int mod = 1e9 + 7;
const int maxn = 1e6 + 5;
const int INF = 0x3f3f3f3f;
const LL INFLL = 0x3f3f3f3f3f3f3f3f;
LL qpow(LL n, LL m) {
    LL ans = 1;
    while(m) {
        if(m & 1) ans = ans * n % mod;
        n = n * n % mod;
        m >>= 1;
    }
    return ans;
}


LL inv[maxn];
void init() {
    inv[1] = 1;
    for(int i = 2; i < maxn; i++) {
        inv[i] = (mod - mod / i) * inv[mod % i] % mod;
    }
}
LL n, m, k;
LL Cm[maxn], Ck[maxn];
void get_C(int k) {
    Ck[0] = Cm[0] = 1;
    for(int i = 1; i <= k; i++) {
        Cm[i] = Cm[i - 1] % mod * (m - i + 1) % mod * inv[i] % mod;
        Ck[i] = Ck[i - 1] % mod * (k - i + 1) % mod * inv[i] % mod;
    }
}
int main() {
#ifndef ONLINE_JUDGE
    FIN
#endif
    init();
    int T;
    scanf("%d", &T);
    int cas = 1;
    while(T--) {
        scanf("%lld%lld%lld", &n, &m, &k);
        get_C(k);
        int flag = 1;
        LL tot = 0;
        for(LL i = k; i >= 1; i--) {
            tot = (tot + 1LL * flag * Ck[k - i] * (i) % mod * qpow(i - 1, n - 1) % mod + mod) % mod;
            flag = -1 * flag;
        }
        tot = (tot * Cm[k]) % mod;
        printf("Case #%d: %lld
", cas++, tot);
    }
    return 0;
}
每一个不曾刷题的日子 都是对生命的辜负 从弱小到强大,需要一段时间的沉淀,就是现在了 ~buerdepepeqi
原文地址:https://www.cnblogs.com/buerdepepeqi/p/10902812.html