【ARC106E】 Medals 【网络流】【最小割】【FWT】

题意

有$N$个人($N\le 18$)，第$i$个人出现$A_i$天然后消失$A_i$天($A_i\le 10^5$)，循环下去。每天可以给某一个出现的人发一块奖牌，问最少要多少天才能让任意一个人都有至少$K$块奖牌($K\le 10^5$)。

分析

首先二分答案，这样就转化为了一个判定性问题。
考虑一个简单的网络流建模，$S$向每个人连一条容量为$K$的边，每个人向他出现的一天连一条容量为$1$的边，每一天向$T$连一条容量为$1$的边。这样只要最大流满流就有解了。
考虑到人数很小，可以把所有天分成$2^N$类，代表每个人是否在这一天出现。这样我们就可以把同一类的若干天合并成一个点，向$T$连一条容量为这一类天的总数的边。
考虑答案的范围。显然答案$\ge NK$。按顺序给每一个人发$K$块奖牌，每个人至多花费$2K$天，因此答案的一个上界是$2NK$，只有$3.6\times 10^6$级别。因此我们可以预处理出每一天是属于哪一类的。
考虑怎样快速计算这个网络流。把网络流转化为最小割来考虑，相当于在左边割掉一些人，右边割掉一些天使得$S$和$T$不联通。如果一个人没有被左边割掉，那右边包含这个人的所有天必须被割掉。因此我们可以枚举左边割掉了哪些人，这样右边割掉的天就已经确定下来了，不难发现右边割掉的边的权值和是一个子集和的形式，可以通过$FWT$求高维前缀和在$2^{N}N$的时间复杂度内预处理出来。判断最小割是否为$NK$即可。
时间复杂度：$2^{N}N\log (NK)$

代码

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
const int N=20,M=1050005;
int n,m,k,l,r,mid,ans,a[N],b[M],c[M],sta[N*M];
bool check(){
	for(int i=0;i<m;i++){
		b[i]=0;
		c[i]=c[i>>1]+(i&1);
	}
	for(int i=0;i<mid;i++){	
		b[sta[i]]++;
	}
	for(int i=1;i<m;i<<=1){
		for(int j=0;j<m;j+=(i<<1)){
			for(int k=j;k<j+i;k++){
				b[k+i]+=b[k];
			}
		}
	}
	int res=n*k;
	for(int i=0;i<m;i++){
		if(b[m-1]-b[i]+c[i]*k<res){
			return false;
		}
	}
	return true;
}
int main(){
	scanf("%d%d",&n,&k);
	m=1<<n;
	for(int i=0;i<n;i++){
		scanf("%d",&a[i]);
	}
	l=n*k,r=2*n*k,ans=r;
	for(int i=0;i<r;i++){
		for(int j=0;j<n;j++){
			if(~(i/a[j])&1){
				sta[i]|=1<<j;
			}
		}
	}
	while(l<=r){
		mid=(l+r)/2;
		if(check()){
			ans=mid;
			r=mid-1;
		}else{
			l=mid+1;
		}
	}
	printf("%d\n",ans);
	return 0;
}