码迷,mamicode.com
首页 > 其他好文 > 详细

组合数+容斥原理 UVALive 7040 Color(14西安F)

时间:2015-10-02 14:54:54      阅读:215      评论:0      收藏:0      [点我收藏+]

标签:

 

题目传送门

题意:n盆花涂色,相邻不能涂相同的颜色,从m中颜色选取k种颜色涂,保证正好有k种颜色

分析:从m中颜色选取k种就是C (m, k),然后第一个有k种选择,之后的都有k-1种选择,这样是不超过k种颜色的方案,那么减去少了Ai颜色的方案数,用容斥原理,最后答案是C(m,k) × ( k × (k-1)^(n-1) + ∑((-1)^p × C(k, p) × p × (p-1)^(n-1) ) (2 <= p <= k-1)

 

#include <cstdio>
#include <cmath>
#include <cstring>
#include <algorithm>
#include <iostream>
#include <vector>
#include <set>
#include <map>
#include <queue>
using namespace std;

#define lson l, mid, rt << 1
#define rson mid + 1, r, rt << 1 | 1
typedef long long ll;
const int N = 1e6 + 10;
const int INF = 0x3f3f3f3f;
const int MOD = 1e9 + 7;
const int EPS = 1e-8;
int inv[N];

void init(int n)	{
	inv[1] = 1;
	for (int i=2; i<=n; ++i)	{
		inv[i] = (MOD - MOD / i) * 1ll * inv[MOD%i] % MOD;		//inv[n]
	}
}

int pow_mod(int x, int n, int p)	{
	int ret = 1;
	while (n)	{
		if (n & 1)	ret = ret * 1l * x % p;
        x = x * 1ll * x % p;
        n >>= 1;
	}
	return ret;
}

int f(int n, int k)	{
	if (!k)	return 0;
	else	return k * 1ll * pow_mod (k-1, n-1, MOD) % MOD;
}

int main(void)	{
	init (1000000);
	int T, cas = 0;	scanf ("%d", &T);
	while (T--)	{
		int n, m, k;	scanf ("%d%d%d", &n, &m, &k);
		ll ans = 0;	int res = 1;
		for (int i=1; i<=k; ++i)	{
			res = res * 1ll * (k - i + 1) % MOD * inv[i] % MOD;
			if (k - i & 1)	{
				ans -= res * 1ll * f (n, i) % MOD;
				if (ans < 0)	ans += MOD;
			}
			else	{
				ans += res * 1ll * f (n, i) % MOD;
				if (ans >= MOD)	ans -= MOD;
			}
		}
		for (int i=1; i<=k; ++i)	{
			ans = ans * 1ll * (m - i + 1) % MOD * inv[i] % MOD;
		}
		printf ("Case #%d: %lld\n", ++cas, ans);
	}

	return 0;
}

  

组合数+容斥原理 UVALive 7040 Color(14西安F)

标签:

原文地址:http://www.cnblogs.com/Running-Time/p/4852200.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!