2014-2015 ACM-ICPC, Asia Xian Regional Contest F - Color

szy10010

已于 2024-08-09 20:59:07 修改

阅读量229

点赞数 9

文章标签： c++ 算法开发语言

于 2024-08-09 20:54:39 首次发布

本文链接：https://blog.csdn.net/s1379659220/article/details/141070340

版权

题意:给定T组样例,接下来每组输入n,m,k：n朵鲜花排成一排,m种颜色,要求恰好使用k种颜色,并且相邻两朵花颜色不能相同

思路:首先m中选k个颜色是必须有的 $C\binom{k}{m}$ ，然后思考一下k种颜色如何做到不相邻？从左到右第一个位置开始,第一个位置有k种选择,接下来n-1个位置由于前一个位置的选择,只剩下k-1种选择,也就是: $k*k^{(n-1)}$ 但是发现,这种计算方式只能做到使用小于等于k种颜色,但是本题需要恰好k种,所以要减去小于等于k-1种的情况,这里发现这样减去一定会使得小于等于k-2种的情况被多减了,于是要加上小于等于k-2种的情况（容斥原理）

那么恰好使用k种颜色的情况就可以这么计算

$k*(k-1)^{n-1}-C\binom{1}{k}(k-1)*(k-2)^{n-1}+C\binom{2}{k}(k-2)*(k-3)^{n-1}-...$

容斥原理的证明比较繁琐,这里可以简单计算证明一下公式的正确性

就拿前三项举例子,假设此时k=6

那么小于等于6种颜色的方案数的计算公式就是 $k*(k-1)^{6-1}$

此时小于等于5种颜色的方案数量就是在小于等于6种颜色的情况下选一种颜色删除也就是

$C\binom{1}{6}$ ,那么小于等于5种颜色的方案数量就是 $C\binom{1}{6}(k-1)*(k-2)^{6-1}$ 也就是理论上要删去的小于等于5种颜色的方案数量

此时小于等于4种颜色的方案数量就是在小于等于6种颜色的情况下选两种颜色删除也就是

$C\binom{2}{6}$ ,那么小于等于4种颜色的方案数量就是 $C\binom{2}{6}(k-2)*(k-3)^{6-1}$ 也就是理论上要删去的小于等于4种颜色的方案数量

此时小于等于4种颜色的方案数量就是在小于等于5种颜色的情况下选一种颜色删除也就是

$C\binom{1}{5}$ ,那么小于等于4种颜色的方案数量就是 $C\binom{1}{5}(k-2)*(k-3)^{6-1}$ 也就是理论上要删去的小于等于4种颜色的方案数量但是在这之前小于等于4种颜色在小于等于5种颜色的情况被减去了 $C\binom{1}{6}$ 次也就是 $C\binom{1}{5}*C\binom{1}{6}$ 次，此时加上最前面那个恰好使用k种颜色的情况公式的第三项 $C\binom{2}{6}$ 恰好等于要蓝字中本该减去的 $C\binom{2}{6}$ ,第三项以后同理.

代码实现

#include <map>
#include <queue>
#include <deque>
#include <cmath>
#include <vector>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <unordered_map>
using namespace std;
#define fi first
#define se second
#define pb push_back
#define pp pop_back()
#define int long long
#define laile cout<<"laile"<<endl
#define lowbit(x) ((x)&(-x))
#define double long double
#define sf(x) scanf("%lld",&x)
#define sff(x,y) scanf("%lld %lld",&x,&y)
#define _for(i,n) for(int i=0;i<(n);++i)
#define _rep(i,a,b) for(int i=(a);i<=(b);++i)
#define all(x) (x).begin(), (x).end()
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0)
typedef unsigned long long ULL;
typedef pair<int,int>PII;
const int N=1e6+10,INF=4e18,P=1e9+7;
int A[N],FA[N];
int qmi(int a,int k)
{
	int res=1;
	while(k)
	{
		if(k&1)res=res*a%P;
		a=a*a%P;
		k>>=1;
	}
	return res;
}
void init()
{
	A[0]=1;
	FA[0]=qmi(A[0],P-2);
	for(int i=1;i<N;i++)
		A[i]=A[i-1]*i%P,FA[i]=qmi(A[i],P-2);
}
int C(int a,int b)
{
	return A[a]*FA[b]%P*FA[a-b]%P;
}
int CC(int a,int b)
{
	int res=1;
	for(int i=1,j=a;i<=b;i++,j--)
	{
		res=res*j%P;
		res=res*qmi(i,P-2)%P;
	}
	return res;
}
int n,m,k,cnt;
void solve()
{
	cin>>n>>m>>k;
	int res=0;
	for(int i=0;i<k;i++)
	{
		int now=C(k,i)*(k-i)%P*qmi((k-i-1),n-1)%P;
		if(i%2==0)res+=now;
		else res-=now;
		res=(res+P)%P;
	}
	res=(res+P)%P*CC(m,k)%P;
	cout<<"Case #"<<++cnt<<": "<<res<<endl;
	return ;
}
signed main()
{
	init();
	IOS;
	int T=1;
	cin>>T;
	while(T--)
		solve();
	return 0;
}