【最大流+缩点】HDU-3605-Escape

题目链接<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;
}