[HAOI 2018]染色

传送门

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?

#include<bits/stdc++.h>
#define reg register
#define ll long long
#define db double
using namespace std;
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=(x<<3)+(x<<1)+ch-'0';ch=getchar();}
    return x*f;
}
const int P=1004535809,G[2]={3,334845270},NN=5e5+5;
int Mul(int x,int y){return 1ll*x*y%P;}
int Add(int x,int y){return (x+y)%P;}
const int MN=1e7+5,MM=1e5+5,MS=155;
int N,M,S,W[MM],f[MM],fac[MN],inv[MN];
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;}
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]));}
int a[NN],b[NN],pos[NN];
void NTT(int *a,bool ty,int L)
{
    reg int i,j,k,w,wn,x,y;
    for(i=0;i<L;++i) if(pos[i]<i) swap(a[i],a[pos[i]]);
    for(i=1;i<L;i<<=1)
    {
        wn=fpow(G[ty],(P-1)/(i<<1));
        for(j=0;j<L;j+=(i<<1))
            for(w=1,k=0;k<i;++k,w=Mul(w,wn))
            {
                x=a[j+k],y=Mul(a[j+i+k],w);
                a[j+k]=Add(x,y);a[j+i+k]=Add(x,P-y);
            }
    }
    if(ty)for(j=fpow(L,P-2),i=0;i<L;++i)a[i]=Mul(a[i],j);
}
int ans;
int main()
{
    N=read(),M=read(),S=read();
    reg int i,lim;
    for(i=0;i<=M;++i) W[i]=read();
    lim=max(N,M);
    for(fac[0]=i=1;i<=lim;++i) fac[i]=Mul(fac[i-1],i);
    for(inv[0]=inv[1]=1,i=2;i<=lim;++i) inv[i]=Mul(inv[P%i],(P-P/i));
    for(i=1;i<=lim;++i) inv[i]=Mul(inv[i],inv[i-1]);
    lim=min(M,N/S);
    for(i=0;i<=lim;++i)
        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)));
    for(i=0;i<=lim;++i) b[lim-i+1]=Mul(f[i],fac[i]);
    for(i=0;i<=lim;++i) a[i]=(i&1)?(P-inv[i]):inv[i];
    int MA;
    for(MA=1;MA<=(lim<<1);MA<<=1);
    for(i=0;i<MA;++i) pos[i]=(pos[i>>1]>>1)|((i&1)*(MA>>1));
    NTT(a,0,MA);NTT(b,0,MA);
    for(i=0;i<MA;++i) a[i]=Mul(a[i],b[i]);
    NTT(a,1,MA);
    for(i=0;i<=lim;++i) ans=Add(ans,Mul(W[i],Mul(a[lim-i+1],inv[i])));
    return 0*printf("%d\n",ans);
}


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

原文地址:https://www.cnblogs.com/PaperCloud/p/10925578.html

时间: 2024-11-02 23:57:39

[HAOI 2018]染色的相关文章

HAOI 2018 染色(容斥+NTT)

题意 https://loj.ac/problem/2527 思路 设 \(f(k)\) 为强制选择 \(k\) 个颜色出现 \(s\) 种,其余任取的方案数. 则有 \[ f(k)={m\choose k}{n\choose sk}{(sk)!\over(s!)^k}(m-k)^{n-sk} \] 不难看出,这个方案可能包括了超过 \(k\) 种颜色,也有重复的方案,所以恰有 \(k\) 个颜色出现 \(s\) 种的方案 \(ans_k\) 满足 \[ ans_k=\sum_{i=k}^{\m

12.19 省选预备班

区间dp 1.石子合并 区间dp  n3方 随 i 增加 k 单调不降  优化成 n2 2. cf1025D gcd 预处理出来 n4 f[i][j][k]  区间 i j  以 k 为跟的二叉树是否存在 n3 f[i][j][0/1]表示区间 i j 根节点在左端还是右端  转移变成 o1 g[i][j][0/1]  区间 i j  根节点为 i/j 的二叉树是否存在   枚举k 与 j 互质 ,则可以作为根     转移复杂度  O(n) 二叉树的一个中序遍历对应一个区间 3. 多重背包 O

树形dp小结

做了好久的树形DP(大雾) ,(noip)csp-s考了好多树形DP 树形DP基本分为这几种 1.最大独立集(没有上司的舞会) 经典树形DP \(dp[i][0/1]\) 表示i这个点选与不选. \(dp[u][0] += dp[v][1];\) \(dp[u][1] += max(dp[v][0],dp[v][1]);\) 2.最小点覆盖(战略游戏) 状态与上面一样 \(dp[u][0] += dp[v][1];\) \(dp[u][1] += min(dp[v][0],dp[v][1]);\

如何将C/C++程序转译成Delphi(十四)

众所周知,数据科学是这几年才火起来的概念,而应运而生的数据科学家(data scientist)明显缺乏清晰的录取标准和工作内容.此次课程以<星际争霸II>回放文件分析为例,集中在IBM Cloud相关数据分析服务的应用.面对星际游戏爱好者希望提升技能的要求,我们使用IBM Data Science Experience中的jJupyter Notebooks来实现数据的可视化以及对数据进行深度分析,并最终存储到IBM Cloudant中.这是个介绍+动手实践的教程,参会者不仅将和讲师一起在线

世界著名设计小组nrg推荐的75个FLASH酷站

众所周知,数据科学是这几年才火起来的概念,而应运而生的数据科学家(data scientist)明显缺乏清晰的录取标准和工作内容.此次课程以<星际争霸II>回放文件分析为例,集中在IBM Cloud相关数据分析服务的应用.面对星际游戏爱好者希望提升技能的要求,我们使用IBM Data Science Experience中的jJupyter Notebooks来实现数据的可视化以及对数据进行深度分析,并最终存储到IBM Cloudant中.这是个介绍+动手实践的教程,参会者不仅将和讲师一起在线

铺揭竟冒剐惹蒲掷咕堑接veld

IEEE Spectrum 杂志发布了一年一度的编程语言排行榜,这也是他们发布的第四届编程语言 Top 榜. 据介绍,IEEE Spectrum 的排序是来自 10 个重要线上数据源的综合,例如 Stack Overflow.Twitter.Reddit.IEEE Xplore.GitHub.CareerBuilder 等,对 48 种语言进行排行. 与其他排行榜不同的是,IEEE Spectrum 可以让读者自己选择参数组合时的权重,得到不同的排序结果.考虑到典型的 Spectrum 读者需求

支酸权我据好张近拉其深就率sJFlzqgrA

社保划到税务征收,将大大提升社保费的征管效率.税务的征管能力是目前而言最强的,以后税务征收社保不是代收,属于本职了. 之前税局要把社保信息和交个税的工资比对起来有困难!现在好了,个税是自己的,社保也是自己的,比对困难?不存在的! 这一变革,会给那些不给员工上社保.不全额上社保的企业致命一击! 最新案例 前段时间的发改委关于限制特定严重失信人乘坐民航的一则意见--发改财金[2018]385号,其中还有税务总局的联合署名. http://weibo.com/20180408PP/2309279811

此每究具条如书中深花等业从次BaefoZ16

社保划到税务征收,将大大提升社保费的征管效率.税务的征管能力是目前而言最强的,以后税务征收社保不是代收,属于本职了. 之前税局要把社保信息和交个税的工资比对起来有困难!现在好了,个税是自己的,社保也是自己的,比对困难?不存在的! 这一变革,会给那些不给员工上社保.不全额上社保的企业致命一击! 最新案例 前段时间的发改委关于限制特定严重失信人乘坐民航的一则意见--发改财金[2018]385号,其中还有税务总局的联合署名. http://weibo.com/20180408PP/2309279811

$[\ HAOI\ 2015\ ]\ $树上染色

\(\\\) \(Description\) 一棵\(N\)个节点的树,树边有边权.在这棵树中选择\(K\)个点,将其染成黑色,并将其他的\(N-K\)个点染成白色 . 定义树的价值为,黑点两两之间的最短距离的和加上白点两两之间的最短距离距离的和.问价值最大值是多少. \(N,K\in [1,2000],K\le N\) \(\\\) \(Solution\) 将问题转化为对每条边的贡献统计.考虑一条边会将树分成两个连通块,则该边的贡献等于 边权\(\times(\)由这条边分开的两个连通块内黑