传送门

Description

一个长度为\(N\)的序列, 每个位置都可以被染成 \(M\)种颜色中的某一种.

出现次数恰好为 \(S\)的颜色种数有\(i\)种, 会产生\(w_i\)的愉悦度.

对于所有染色方案, 能获得的愉悦度的和对\(1004535809\)取模的结果.

Solution 

\[ans=\sum_{i=0}^{lim} w_i\cdot num_i
\]

how to get \(num_i\)?

\(f_i\) : the number of occurrences of at least i colors is exactly the number of S

so \(f_i=\binom{m}{i}\cdot \frac{n!}{(s!)^i(n-iS)!}\cdot(m-i)^{n-iS}\)

According to the binomial inversion, we can know that:

\[num_i=\sum_{j=i}^{lim}(-1)^{j-i}\binom{j}{i}f[j]
\]

so

\[num_i=\frac{1}{i!}\sum_{j=i}^{lim} \frac{(-1)^{j-i}}{(j-i)!}\cdot(f[j]\cdot j!)
\]

we can use NTT.

Code 

  1. #include<bits/stdc++.h>
  2. #define reg register
  3. #define ll long long
  4. #define db double
  5. using namespace std;
  6. inline int read()
  7. {
  8. int x=0,f=1;char ch=getchar();
  9. while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
  10. while(ch>='0'&&ch<='9'){x=(x<<3)+(x<<1)+ch-'0';ch=getchar();}
  11. return x*f;
  12. }
  13. const int P=1004535809,G[2]={3,334845270},NN=5e5+5;
  14. int Mul(int x,int y){return 1ll*x*y%P;}
  15. int Add(int x,int y){return (x+y)%P;}
  16. const int MN=1e7+5,MM=1e5+5,MS=155;
  17. int N,M,S,W[MM],f[MM],fac[MN],inv[MN];
  18. int fpow(int x,int y){int r=1;for(;y;y>>=1,x=Mul(x,x))if(y&1)r=Mul(r,x);return r;}
  19. int C(int x,int y){if(x<0||y<0||x<y)return 0;return Mul(fac[x],Mul(inv[y],inv[x-y]));}
  20. int a[NN],b[NN],pos[NN];
  21. void NTT(int *a,bool ty,int L)
  22. {
  23. reg int i,j,k,w,wn,x,y;
  24. for(i=0;i<L;++i) if(pos[i]<i) swap(a[i],a[pos[i]]);
  25. for(i=1;i<L;i<<=1)
  26. {
  27. wn=fpow(G[ty],(P-1)/(i<<1));
  28. for(j=0;j<L;j+=(i<<1))
  29. for(w=1,k=0;k<i;++k,w=Mul(w,wn))
  30. {
  31. x=a[j+k],y=Mul(a[j+i+k],w);
  32. a[j+k]=Add(x,y);a[j+i+k]=Add(x,P-y);
  33. }
  34. }
  35. if(ty)for(j=fpow(L,P-2),i=0;i<L;++i)a[i]=Mul(a[i],j);
  36. }
  37. int ans;
  38. int main()
  39. {
  40. N=read(),M=read(),S=read();
  41. reg int i,lim;
  42. for(i=0;i<=M;++i) W[i]=read();
  43. lim=max(N,M);
  44. for(fac[0]=i=1;i<=lim;++i) fac[i]=Mul(fac[i-1],i);
  45. for(inv[0]=inv[1]=1,i=2;i<=lim;++i) inv[i]=Mul(inv[P%i],(P-P/i));
  46. for(i=1;i<=lim;++i) inv[i]=Mul(inv[i],inv[i-1]);
  47. lim=min(M,N/S);
  48. for(i=0;i<=lim;++i)
  49. f[i]=Mul(Mul(C(M,i),Mul(fac[N],fpow(inv[S],i))),Mul(inv[N-i*S],fpow(M-i,N-i*S)));
  50. for(i=0;i<=lim;++i) b[lim-i+1]=Mul(f[i],fac[i]);
  51. for(i=0;i<=lim;++i) a[i]=(i&1)?(P-inv[i]):inv[i];
  52. int MA;
  53. for(MA=1;MA<=(lim<<1);MA<<=1);
  54. for(i=0;i<MA;++i) pos[i]=(pos[i>>1]>>1)|((i&1)*(MA>>1));
  55. NTT(a,0,MA);NTT(b,0,MA);
  56. for(i=0;i<MA;++i) a[i]=Mul(a[i],b[i]);
  57. NTT(a,1,MA);
  58. for(i=0;i<=lim;++i) ans=Add(ans,Mul(W[i],Mul(a[lim-i+1],inv[i])));
  59. return 0*printf("%d\n",ans);
  60. }

Blog来自PaperCloud,未经允许,请勿转载,TKS!

[HAOI 2018]染色的更多相关文章

  1. HAOI 2018 染色(容斥+NTT)

    题意 https://loj.ac/problem/2527 思路 设 \(f(k)\) 为强制选择 \(k\) 个颜色出现 \(s\) 种,其余任取的方案数. 则有 \[ f(k)={m\choos ...

  2. Solution -「HAOI 2018」「洛谷 P4491」染色

    \(\mathcal{Description}\)   Link.   用 \(m\) 种颜色为长为 \(n\) 的序列染色,每个位置一种颜色.对于一种染色方案,其价值为 \(w(\text{出现恰 ...

  3. 「HAOI 2018」染色

    题目链接 戳我 \(Solution\) 观察题目发现恰好出现了\(s\)次的颜色有\(k\)种,不太好弄. 所以我们设\(a[i]\)表示为恰好出现了\(s\)次的颜色有至少\(i\)种的方案数,然 ...

  4. HAOI树上染色

    Description : 有一棵点数为 N 的树,树边有边权.给你一个在 0~ N 之内的正整数 K ,你要在这棵树中选择 K个点,将其染成黑色,并将其他 的N-K个点染成白色 . 将所有点染色后, ...

  5. [BJOI 2018]染色

    题意:求01成立. 并查集维护,记录一个变量判断决策. #include<bits/stdc++.h> using namespace std; #define int long long ...

  6. HAOI 2018 Round 1 题解

    无聊了开一套省选题刷刷--u1s1 感觉三个题都不错,难度也挺有梯度,是一道标准的省选难度的题(话说 CSP 前你刷省选题干嘛/ts/ts) 小 C 珂海星 T1:P4495 [HAOI2018]奇怪 ...

  7. luogu 4429 染色

    bjoi 2018 染色 推了个错误结论得了60分? 题目大意: 一个无重边和自环的无向图,并且对每个点分别给了一个大小为2的颜色集合,只能从这个集合中选一种颜色给这个点染色 求一个染色方案使得没有两 ...

  8. [HAOI 2015]树上染色

    Description 题库链接 给出一棵 \(n\) 个节点的树,边有权值.让你将树上 \(k\) 个点染黑,剩余 \(n-k\) 个点染白.染色后记一种染色方案的价值为黑点间两两距离和以及白点间两 ...

  9. 洛谷 P3177 树上染色 解题报告

    P3177 [HAOI2015]树上染色 题目描述 有一棵点数为\(N\)的树,树边有边权.给你一个在\(0\) ~ \(N\)之内的正整数\(K\),你要在这棵树中选择\(K\)个点,将其染成黑色, ...

随机推荐

  1. 拓展 - Webrtc 的回声抵消(aec、aecm)算法简介

    webrtc 的回声抵消(aec.aecm)算法简介 原文链接:丢失.不好意思        webrtc 的回声抵消(aec.aecm)算法主要包括以下几个重要模块:1.回声时延估计 2.NLMS( ...

  2. loadrunner通过字符串左右边界提取字符串

    /****** *函数名称:strcut *函数说明:通过左边界.右边界,从字符串中截取子字符串 *注意事项:会申请新的内存,需要手动释放 ******/ void strcut(char *strS ...

  3. 设置vue-quill-editor禁止输入或编辑

    <quill-editor class="ql-editor" v-model="form.content" ref="myQuillEdito ...

  4. ansible 中 JAVA_HOME不生效问题

    解决方案 ~/.bash_profile 是交互式.login 方式进入 bash 运行的,意思是只有用户登录时才会生效. ~/.bashrc 是交互式 non-login 方式进入 bash 运行的 ...

  5. maven设定项目编码

    今天在DOS下执行mvn compile命令时报错说缺少必要符号,事实上根本就没有缺少,但何以如此呢,为啥eclipse在编译时就没有这问题呢? 原因是编码的问题造成的! eclipse在编译的使用使 ...

  6. SSH SSL TELNET的比较(转)

    转载链接  https://blog.csdn.net/baidu_39486224/article/details/81295701 SSL(Secure Sockets Layer (SSL) a ...

  7. vue2 design 手记

    Ant Design of Vue github地址:https://github.com/vueComponent/ant-design-vue Ant Design of Vue文档:https: ...

  8. SpringCloud2.0 Ribbon 服务发现 基础教程(四)

    1.启动[服务中心]集群,即 Eureka Server 参考 SpringCloud2.0 Eureka Server 服务中心 基础教程(二) 2.启动[服务提供者]集群,即 Eureka Cli ...

  9. 【Selenium-WebDriver实战篇】java测试使用HttpClient debug日志关闭

    在上一篇设置完Tess4J之后,引用jar包之前,我的日志体系一直是只出现info级别的,但是引用之后出现很多httpClient的请求. 于是网上调查了下,可以通过代码实现,就在入口程序增加该部分代 ...

  10. C#编写简单的聊天程序(转)

    这是一篇基于Socket进行网络编程的入门文章,我对于网络编程的学习并不够深入,这篇文章是对于自己知识的一个巩固,同时希望能为初学的朋友提供一点参考.文章大体分为四个部分:程序的分析与设计.C#网络编 ...