版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/monochrome00/article/details/81515584
题目链接<http://acm.hdu.edu.cn/showproblem.php?pid=3605>
题意:
有n个人,m个星球。每个人都对不同的星球有自己的喜好,每个星球都有自己的容量。问能否让所有的人都呆在自己喜欢的星球里。
(1 <= n <= 100000), m (1 <= m <= 10)
题解:
以为是套模板的题,一直TLE,MLE。看了大佬的代码才明白这题需要缩点。因为最多有10个星球,所以最多有种可能的情况。也就是说可以把100000个人缩成个人。这样就能过了。
#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<cmath>
using namespace std;
const int N=3e5+7;
const int inf=1<<26;
struct Edge{
int u,v,w,nxt;
Edge(int u=0,int v=0,int w=0,int nxt=0):u(u),v(v),w(w),nxt(nxt){}
}edge[10*N];
int n,m,edn,sp,tp;
int p[N],d[N],c[N];
void add(int u,int v,int w){
edge[edn]=Edge(u,v,w,p[u]);p[u]=edn++;
edge[edn]=Edge(v,u,0,p[v]);p[v]=edn++;
}
bool bfs(){
memset(d,-1,sizeof(d));d[sp]=0;
queue<int>q;q.push(sp);
while(!q.empty()){
int u=q.front();
q.pop();
for(int i=p[u];~i;i=edge[i].nxt){
int v=edge[i].v;
if(d[v]==-1&&edge[i].w){
d[v]=d[u]+1;
q.push(v);
if(v==tp) return true;
}
}
}
return ~d[tp];
}
int dfs(int u,int b){
if(u==tp) return b;
int r=0;
for(int i=c[u];~i;i=edge[i].nxt){
int v=edge[i].v;
if(edge[i].w&&d[v]==d[u]+1){
int x=min(edge[i].w,b-r);
c[u]=i;
x=dfs(v,x);
r+=x;
edge[i].w-=x;
edge[i^1].w+=x;
if(r==b) break;
}
}
if(!r)d[u]=-2;
return r;
}
int dinic(){
int total=0,t;
while(bfs()){
memcpy(c,p,sizeof(p));
while(t=dfs(sp,inf))
total+=t;
}
return total;
}
int num[2000],cap[20];
int main()
{
int x;
while(~scanf("%d%d",&n,&m)){
memset(p,-1,sizeof(p));edn=0;
memset(num,0,sizeof(num));
sp=2000;tp=sp+1;
for(int i=1;i<=n;i++){
int tmp=0;
for(int j=0;j<m;j++){
scanf("%d",&x);
if(x) tmp+=(1<<j);
}
num[tmp]++;
}
for(int i=0;i<=(1<<m)-1;i++){
if(num[i]){
add(sp,i,num[i]);
for(int j=0;j<m;j++){
if(i&(1<<j)){
add(i,j+1025,inf);
}
}
}
}
for(int i=1;i<=m;i++){
scanf("%d",&x);
add(i+1024,tp,x);
}
if(dinic()==n) printf("YES\n");
else printf("NO\n");
}
return 0;
}