博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
2019.02.09 codeforces gym 100548F. Color(容斥原理)
阅读量:5016 次
发布时间:2019-06-12

本文共 1540 字,大约阅读时间需要 5 分钟。

题意简述:对n个排成一排的物品涂色,有m种颜色可选。
要求相邻的物品颜色不相同,且总共恰好有K种颜色,问所有可行的方案数。(n,m≤1e9,k≤1e6n,m\le1e9,k\le1e6n,m1e9,k1e6


思路:

容斥原理套路:
先不考虑是否选全kkk种颜色,方案数为Cmk∗k∗(k−1)n−1C_m^k*k*(k-1)^{n-1}Cmkk(k1)n1
然后枚举剩下的至少有几种颜色没选来容斥掉非法情况:
于是Ans=Cmk∑i=k1(−1)k−iCkii(i−1)n−1Ans=C_m^k\sum_{i=k}^1(-1)^{k-i}C_k^ii(i-1)^{n-1}Ans=Cmki=k1(1)kiCkii(i1)n1
代码:

#include
#define ri register intusing namespace std;const int N=1e6+5,mod=1e9+7;typedef long long ll;inline int read(){
int ans=0; char ch=getchar(); while(!isdigit(ch))ch=getchar(); while(isdigit(ch))ans=(ans<<3)+(ans<<1)+(ch^48),ch=getchar(); return ans;}inline int add(const int&a,const int&b){
return a+b>=mod?a+b-mod:a+b;}inline int dec(const int&a,const int&b){
return a>=b?a-b:a-b+mod;}inline int mul(const int&a,const int&b){
return (ll)a*b%mod;}inline int ksm(int a,int p){
int ret=1;for(;p;p>>=1,a=mul(a,a))if(p&1)ret=mul(ret,a);return ret;}int n,m,k,C[N],inv[N],mult;inline void init(){
inv[1]=1; for(ri i=2;i<=1000000;++i)inv[i]=mul(inv[mod-mod/i*i],mod-mod/i);}inline void Init(){
C[0]=1,mult=1; for(ri i=1;i<=k;++i)C[i]=mul(mul(C[i-1],k-i+1),inv[i]); for(ri i=1;i<=k;++i)mult=mul(mult,mul(m-i+1,inv[i]));}int main(){
init(); for(ri ans,tt=1,up=read();tt<=up;++tt){
n=read(),m=read(),k=read(),ans=0; Init(); for(ri i=k,tmp;i;--i){
tmp=mul(C[i],mul(i,ksm(i-1,n-1))); ans=(k-i)&1?dec(ans,tmp):add(ans,tmp); } cout<<"Case #"<
<<": "<
<<'\n'; } return 0;}

转载于:https://www.cnblogs.com/ldxcaicai/p/10367692.html

你可能感兴趣的文章
WEB安全实战(四)关于 Cookie
查看>>
go输入Hello word
查看>>
ST-1之乱码bug
查看>>
Nopcommerce主要用到的技术及特点
查看>>
Chrome 表单背景是淡黄色的问题
查看>>
Http Status 404 - No result defined for action cn.itcast.web.action.CustomerAction and result error
查看>>
788-旋转数字
查看>>
UVA-10054.The Necklace(欧拉回路)解题报告
查看>>
C++面向对象之内核、实现、反思与救赎
查看>>
LVS实现负载均衡
查看>>
[Linux]第三部分-学习Shell和Shell脚本
查看>>
C#中byte[]与string的转换
查看>>
z-index问题
查看>>
Java设计模式应用——模板方法模式
查看>>
Zookeeper学习(八):Zookeeper的数据发布与订阅模式
查看>>
nginx限制请求之四:目录进行IP限制
查看>>
一步步构建大型网站架构
查看>>
快速掌握ES6语法
查看>>
Python 多进程教程
查看>>
CSS3和javascript中的transform
查看>>