[2018HN省队集训D8T3] 水果拼盘

题意

给定 \(n\) 个集合, 每个集合包含 \([1,m]\) 中的一些整数, 在这些集合中随机选取 \(k\) 个集合, 求这 \(k\) 个集合的并集的权值的期望.

一个集合的权值定义为, 对于所有 \([1,m]\) 的整数, 若集合中含有 \(i\) 则产生 \(a_i\) 的贡献, 否则产生 \(b_i\) 的贡献.

\(n\le 1\times 10^5, m\le 18,k\le 25\)

题解

好像只有我一个写了一些玄学FWT操作...别人都是组合数直接碾的qaq

显然我们可以通过求所有最终集合的生成概率来计算出最终期望. 而这个概率显然就是个或卷积的形式.

于是我们可以FWT一发.

接着我们发现直接FWT卷 \(k\) 次可能会有重复的方案. (就像[BZOJ 3771] Triple那题). 于是我们需要考虑容斥.

然而这次是广义容斥, 普通二项式反演出来是假的.

stdcall&栋栋说过广义容斥瞎换一波系数就过了, 于是思考一些奇怪的东西来凑容斥系数.

FWT卷 \(k\) 次后得到的方案数是 \(n^k\), 而我们实际上需要的不重复的方案数应该是 \(n^{\underline k}\) (卷积出来的方案有序, 要自带一个全排列), 那么我们需要用一些玄学系数用 \(n^k\) 凑出 \(n^{\underline k}\).

注意到其实 \(n^{\underline k}\) 就是一个普通多项式, 那么我们可以直接算出这个多项式的每一项系数把它作为容斥系数.

实际上就是带符号第一类斯特林数. 用这个系数容斥一下就好了.

FWT一次后的点值可以重复使用, 所以总时间复杂度是 \(O(\sum|S|+(k+m)2^m)\).

参考代码

#include <bits/stdc++.h>

const int MAXK=27;
const int MAXL=1e6+10;
const int MOD=998244353; int n;
int m;
int k;
int a[MAXL];
int pw[MAXK];
int ans[MAXL];
int cof[MAXK];
int c[MAXK][2]; void FWT(int*,int);
void IFWT(int*,int);
inline int ReadInt();
inline int Pow(int,int,int); int main(){
scanf("%d%d%d",&n,&m,&k);
for(int i=0;i<m;i++)
scanf("%d",c[i]+1);
for(int i=0;i<m;i++)
scanf("%d",c[i]);
cof[0]=1;
for(int i=0;i<k;i++){
for(int j=i+1;j>0;j--)
cof[j]=(cof[j-1]-1ll*cof[j]*i%MOD+MOD)%MOD;
cof[0]=(MOD-1ll*cof[0]*i%MOD)%MOD;
}
for(int i=0;i<n;i++){
int cnt=ReadInt(),s=0;
while(cnt--)
s|=(1<<(ReadInt()-1));
++a[s];
}
int maxs=1<<m;
FWT(a,maxs);
pw[0]=1;
for(int s=0;s<maxs;s++){
for(int i=1;i<=k;i++)
pw[i]=1ll*pw[i-1]*a[s]%MOD;
for(int i=0;i<k;i++)
ans[s]=(ans[s]+1ll*pw[k-i]*cof[k-i])%MOD;
}
IFWT(ans,maxs);
int cnt=0;
int sum=0;
for(int s=0;s<maxs;s++){
(cnt+=ans[s])%=MOD;
for(int i=0;i<m;i++)
sum=(sum+1ll*c[i][(s>>i)&1]*ans[s])%MOD;
}
printf("%lld\n",1ll*sum*Pow(cnt,MOD-2,MOD)%MOD);
return 0;
} inline void FWT(int* a,int len){
for(int i=1;i<len;i<<=1)
for(int j=0;j<len;j+=(i<<1))
for(int k=0;k<i;k++){
a[j+k+i]+=a[j+k];
a[j+k+i]=(a[j+k+i]>=MOD?a[j+k+i]-MOD:a[j+k+i]);
}
} inline void IFWT(int* a,int len){
for(int i=1;i<len;i<<=1)
for(int j=0;j<len;j+=(i<<1))
for(int k=0;k<i;k++){
a[j+k+i]-=a[j+k];
a[j+k+i]=(a[j+k+i]<0?a[j+k+i]+MOD:a[j+k+i]);
}
} inline int ReadInt(){
int x=0;
register char ch=getchar();
while(!isdigit(ch))
ch=getchar();
while(isdigit(ch)){
x=x*10+ch-'0';
ch=getchar();
}
return x;
} inline int Pow(int a,int n,int p){
int ans=1;
while(n>0){
if(n&1)
ans=1ll*a*ans%p;
a=1ll*a*a%p;
n>>=1;
}
return ans;
}

[2018HN省队集训D8T3] 水果拼盘的更多相关文章

  1. [2018HN省队集训D9T1] circle

    [2018HN省队集训D9T1] circle 题意 给定一个 \(n\) 个点的竞赛图并在其中钦定了 \(k\) 个点, 数据保证删去钦定的 \(k\) 个点后这个图没有环. 问在不删去钦定的这 \ ...

  2. [2018HN省队集训D8T1] 杀毒软件

    [2018HN省队集训D8T1] 杀毒软件 题意 给定一个 \(m\) 个01串的字典以及一个长度为 \(n\) 的 01? 序列. 对这个序列进行 \(q\) 次操作, 修改某个位置的字符情况以及查 ...

  3. [2018HN省队集训D6T2] girls

    [2018HN省队集训D6T2] girls 题意 给定一张 \(n\) 个点 \(m\) 条边的无向图, 求选三个不同结点并使它们两两不邻接的所有方案的权值和 \(\bmod 2^{64}\) 的值 ...

  4. [Luogu P4143] 采集矿石 [2018HN省队集训D5T3] 望乡台platform

    [Luogu P4143] 采集矿石 [2018HN省队集训D5T3] 望乡台platform 题意 给定一个小写字母构成的字符串, 每个字符有一个非负权值. 输出所有满足权值和等于这个子串在所有本质 ...

  5. [2018HN省队集训D5T2] party

    [2018HN省队集训D5T2] party 题意 给定一棵 \(n\) 个点以 \(1\) 为根的有根树, 每个点有一个 \([1,m]\) 的权值. 有 \(q\) 个查询, 每次给定一个大小为 ...

  6. [2018HN省队集训D5T1] 沼泽地marshland

    [2018HN省队集训D5T1] 沼泽地marshland 题意 给定一张 \(n\times n\) 的棋盘, 对于位置 \((x,y)\), 若 \(x+y\) 为奇数则可能有一个正权值. 你可以 ...

  7. [Codeforces 321D][2018HN省队集训D4T2] Ciel and Flipboard

    [Codeforces 321D][2018HN省队集训D4T2] Ciel and Flipboard 题意 给定一个 \(n\times n\) 的矩阵 \(A\), (\(n\) 为奇数) , ...

  8. [2018HN省队集训D1T3] Or

    [2018HN省队集训D1T3] Or 题意 给定 \(n\) 和 \(k\), 求长度为 \(n\) 的满足下列条件的数列的数量模 \(998244353\) 的值: 所有值在 \([1,2^k)\ ...

  9. [2018HN省队集训D1T1] Tree

    [2018HN省队集训D1T1] Tree 题意 给定一棵带点权树, 要求支持下面三种操作: 1 root 将 root 设为根. 2 u v d 将以 \(\operatorname{LCA} (u ...

随机推荐

  1. Mac配置多个版本JDK

    2016年mac上已经安装有jdk1.6的版本  目录在/Library/Java/JavaVirtualMachines/1.6.0.jdk 有时候mac版本跟新会自动删除jdk1.6 所以要去ma ...

  2. java中线程同步问题

    先不多说,直接上个例子,著名的生产者消费者问题. public class ProducerConsumer { public static void main(String[] args) { Sy ...

  3. java计算机二级笔记

    java.applet.AppletAppletHTMLAppletextends Appletextends AppletprintinitcalendarCalendarCalendar 日历类J ...

  4. win32FTP程序设计

    掌握socket基于事件机制的网络程序设计,掌握多线程技术的FTP Server端设计方法,掌握FTP标准基本协议及其程序的实现,掌握文件内容的网络传输设计方法. 利用CFtpServer类接收和解析 ...

  5. 使用postman测试接口时需要先登录怎么办

    原文 1.在浏览器上先登录,登录成功后获取cookie: 2.接着打开postman:

  6. Yii中文乱码 解决

    需要将config/main.php改成utf-8编码即可

  7. mootools vs jquery

    大部分最近才剛接觸JavaScript的人會面臨到的困難是該選擇哪個套件(library)或是該先學哪個套件.如果你在一間公司裡上班,那麼可能公司已經有一套固定使用的套件,若是在這種情況下,問題就沒那 ...

  8. AIX6.1配置etherchannel

    如果系统之前网卡为up状态,需要将网卡变更为detach chdev -l en2 -a state=detach 绑定完成后,配置IP地址 #smitty mktcpip

  9. ios 九宫格

    #define kViewW 40 //宽度 #define kViewH 61 //高度 #define kColCount 4 //共几列 CGFloat marginX = (self.view ...

  10. asp.net mvc +easyui 实现权限管理(二)

    一写完后,好久没有继续写了.最近公司又在重新开发权限系统了,但是由于我人微言轻,无法阻止他们设计一个太监版的权限系统.想想确实是官大一级压死人啊, 没办法我只好不参与了 让他们去折腾. 我就大概说一下 ...