codeforces 100548F (西安现场赛F题):容斥原理

时间:2021-06-14 15:39:37

题目大意:

对n个排成一排的物品涂色,有m种颜色可选。

要求相邻的物品颜色不相同,且总共恰好有K种颜色,问所有可行的方案数

分析:

从m种颜色中选出k种,有c(m,k)种方法,那么我们只用考虑 k种颜色的涂法即可

显然第一个物品有k种涂法,后面的因为不能跟前面的相同都只有k-1种涂法

因此容易想到一个公式:k*(k-1)^(n-1)

但是这个公式算的是 不超过k种颜色的涂法,题目要求必须k种,怎么办呢?

先考虑一个简化版的问题:

用而且用完5种颜色涂不相关的五个物品的方案数

用阶乘的方法可以算出 ans=120,换一种思路呢想一想这个问题,容易想到

ans(取五种颜色)=5^5(取不大于5种颜色)-c(5,4)*4^5(取不大于4种颜色)

可是一算发现ans竟然小于0了,这是怎么回事呢?容易发现其实取小于四种颜色的方案被减重复了

于是想到需要容斥

ans=c(5,5)*5^5-c(5,4)*4^5+c(5,3)*3^5-c(5,2)*2^5+c(5,1)*1^5 =120

这个问题解决了。原问题也就差不多了。。

代码:

#include <iostream>
#include <stdio.h>
#include<string.h>
#include<algorithm>
#include<string>
#include<ctype.h>
using namespace std;
const long long mod=;
const long long ny=;
long long n,m,k;
long long cm[];
long long cn[];
long long ck[];
long long inv[];
long long mo(long long x)
{
while(x<)
x+=mod;
return x%mod;
}
long long exgcd(long long a,long long b,long long &x,long long &y)
{
if(a==&&b==) return -;
if(b==){x=;y=;return a;}
long long d=exgcd(b,a%b,y,x);
y-=a/b*x;
return d;
}
long long Inv(long long a,long long n)
{
long long x,y;
long long d=exgcd(a,n,x,y);
if(d==) return (x%n+n)%n;
else return -;
}
long long quickmod(long long a,long long b,long long m) //a^b%m
{
long long res=;
while(b)
{
if(b&)
res=res*a%mod;
b>>=;
a=a*a%mod;
}
return res;
}
void ini()
{
cn[]=cm[]=;
memset(cm,,sizeof(cm));
cm[]=;
int tmp=min(m/,k);
for(int i=;i<=tmp;i++)
{
cm[i]=(cm[i-]*(m+-i)%mod*inv[i])%mod;
}
if(cm[k]==)
cm[k]=cm[m-k];
ck[]=ck[k]=;
for(int i=;i<=k/;i++)
{
ck[i]=(ck[i-]*(k+-i)%mod*inv[i])%mod;
ck[k-i]=ck[i];
}
}
int main()
{
//freopen("in.txt","r",stdin);
//freopen("out.txt","w",stdout);
inv[]=;
for(int i=;i<=;i++)
{
inv[i]=Inv(i,mod);
}
int t;
scanf("%d",&t);
int cas=;
while(t--)
{
cas++;
scanf("%I64d%I64d %I64d",&n,&m,&k);
ini();
long long ans=;
long long p=;
for(int i=k;i>=;i--)
{
ans=(ans+p*((ck[k-i])*i%mod*quickmod(i-,n-,mod)%mod)+mod)%mod;
p=-p;
}
ans=(ans*cm[k])%mod;
printf("Case #%d:%c%I64d\n",cas,' ',ans);
}
return ;
}