题意:N个人去m个星球,给出n个人可以去哪些星球的01矩阵。求是否能满足所有人都去。(n到10万,m<=10)
一看,起先一瞬间就建图,准备秒了,人向星球连边,直接最大流判断是否为n,提交超时。。。是啊,10W*10=100W条边,铁定超时。。
后来经牛提示:注意,m<10! 人的可以去星球,一共最多有10个,那只有 2^10次种情况,就是说x部与Y部连线情况很多点是一样的(所给的01矩阵,最多10W行,10列,必然有很多行是一样的)。所以X部只留1024个点,这些点中,点i含j个人的状态,者源点向其连边j,该点向可以去的连边,流量inf.
#include<iostream>
#include<queue>
#include<cstdio>
#include<cstring>
#include<string>
using namespace std;
const int inf=0x3f3f3f3f;
const int maxv=2000,maxe=200000;
int nume=0;int head[maxv];int e[maxe][3];
void inline adde(int i,int j,int c)
{
e[nume][0]=j;e[nume][1]=head[i];head[i]=nume;
e[nume++][2]=c;
e[nume][0]=i;e[nume][1]=head[j];head[j]=nume;
e[nume++][2]=0;
}
int ss,tt,n,m;
int vis[maxv];int lev[maxv];
bool bfs()
{
for(int i=0;i<tt+1;i++)
vis[i]=lev[i]=0;
queue<int>q;
q.push(ss);
vis[ss]=1;
while(!q.empty())
{
int cur=q.front();
q.pop();
for(int i=head[cur];i!=-1;i=e[i][1])
{
int v=e[i][0];
if(!vis[v]&&e[i][2]>0)
{
lev[v]=lev[cur]+1;
vis[v]=1;
q.push(v);
}
}
}
return vis[tt];
}
int dfs(int u,int minf)
{
if(u==tt||minf==0)return minf;
int sumf=0,f;
for(int i=head[u];i!=-1&&minf;i=e[i][1])
{
int v=e[i][0];
if(lev[v]==lev[u]+1&&e[i][2]>0)
{
f=dfs(v,minf<e[i][2]?minf:e[i][2]);
e[i][2]-=f;e[i^1][2]+=f;
sumf+=f;minf-=f;
}
}
if(!sumf) lev[u]=-1;
return sumf;
}
int dinic()
{
int sum=0;
while(bfs())sum+=dfs(ss,inf);
return sum;
}
int contain[15];int mark[1025];
void read_build()
{
int tx;
for(int i=0;i<n;i++)
{
int tkind=0;
for(int j=0;j<m;j++)
{
scanf("%d",&tx);
if(tx==1)
tkind=tkind|(tx<<(m-j-1)); //二进制压缩状态,或者开10维数组判重也方便。
}
mark[tkind]++;
}
for(int i=0;i<=1024;i++)
{
if(mark[i]!=0)
adde(ss,i,mark[i]);
}
for(int i=0;i<=1024;i++)
{
if(mark[i]!=0)
{
int ti=i;int cwei=0;
while(ti!=0)
{
if(ti%2==1)
adde(i,m-1-cwei+1024,inf);
ti=ti/2;
cwei++;
}
}
}
for(int i=0;i<m;i++) //每个星球人数
{
scanf("%d",&contain[i]);
}
for(int i=0;i<m;i++) //星球
{
adde(i+1024,tt,contain[i]);
}
/* for(int i=0;i<=n*m+1;i++)
for(int j=head[i];j!=-1;j=e[j][1])
{
printf("%d->%d:%d\n",i,e[j][0],e[j][2]);
}*/
}
void init()
{
nume=0; ss=1050;tt=ss+1;
for(int i=0;i<=tt;i++)
{
head[i]=-1;
contain[i]=mark[i]=0;
}
}
int main()
{
while(~scanf("%d%d",&n,&m))
{
init();
read_build();
int ans=dinic();
if(ans==n)printf("YES\n");
else printf("NO\n");
}
return 0;
}