Codeforces GYM 100548 F - Color （2014年西安站F题）

最新推荐文章于 2019-02-09 20:05:59 发布

oilover

最新推荐文章于 2019-02-09 20:05:59 发布

阅读量1.4k

点赞数 3

分类专栏：数学文章标签：容斥原理组合数学

本文链接：https://blog.csdn.net/oilover/article/details/45041245

版权

数学专栏收录该内容

15 篇文章 0 订阅

订阅专栏

这是一道关于涂色问题的竞赛题目，要求在N朵花上使用M种颜色，相邻花朵颜色不能相同，并恰好使用K种颜色。通过组合数学和容斥原理，可以得出公式：F(i)=Ciki(i−1)n−1，并通过求和计算出使用K种颜色的方案数。

摘要由CSDN通过智能技术生成

题意： $N$ 朵花，有 $M$ 种颜色可用，要求给他们涂色，相邻的话颜色不同。要求正好使用 $K$ 种颜色。
题解：选 $K$ 种颜色是 $C^k_m$ ，记 $F(i)$ 为选了 $K$ 种颜色后，使用不超过 $i$ 种颜色的方案。第一朵花有 $i$ 种选择，后面 $N-1$ 朵的花的颜色除了前一朵花的颜色之外都可以选，总的就是 $F(i)=C^i_ki(i-1)^{n-1}$
要求出正好 $K$ 种颜色的，就要利用容斥原理，用不超过 $K$ 种的 - 不超过 $(K-1)$ 种的 + 不超过 $(K-2)$ 种的 - 不超过 $(K-3)$ 种的…
答案就是 $C^k_m(F(k)-F(k-1)+F(k-2)...)=C^k_m\sum^k_{i=1}(-1)^{k-i}F(i)$

题目地址
这里写图片描述

#include <iostream>
#include <cstring>
#include <cstdio>
#define prt(k) cout<<#k" = "<<k<<endl;
using namespace std;
typedef long long ll;
const ll mod = 1000000007;
const int N = 1000005;
ll inv[N];
ll C[N];
ll n, m, k;
ll powmod(ll a, ll n)
{
    ll ret = 1;
    for (; n; n>>=1, a=a*a%mod)
        if (n&1)
        ret = ret*a%mod;
    return ret % mod;
}
void init(ll m, ll k)
{
    C[0] = 1;
    for (ll i=1;i<=k;i++)
        C[i] = ( C[i-1] * (m-i+1) %mod )* inv[i]% mod;
}
ll F(ll i)
{
    return (i*C[i] %mod )* powmod(i-1, n-1) % mod;
}
int main()
{
    memset(inv, 0, sizeof inv);
    for (int i=1;i<N;i++) inv[i] = powmod(i, mod-2)%mod;
    int re, ca=1; scanf("%d", &re);
    while (re--)
    {
        scanf("%I64d %I64d %I64d", &n, &m, &k);
        ll sig = 1;
        init(m, k);
        ll ans = C[k]%mod;
        ll tmp = 0;
        memset(C, 0, sizeof C);
        init(k, k);
        for (ll i=k;i>=1;i--)
        {
            tmp = (tmp + sig * F(i) + mod) % mod;
            sig = -sig;
        }
        ans = ans * tmp % mod;
        printf("Case #%d: %I64d\n", ca++, ans);
    }
    return 0;
}