hdu3605(网络流+状态压缩)

链接：点击打开链接

题意：判断二分图匹配左部的点个数为n，右部的点的个数为m，判断左部是否都能匹配(1<=n<=100000,1<=m<=10)

代码：

#include <queue>
#include <vector>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <iostream>
#include <algorithm>
const int INF=0x3f3f3f3f;
using namespace std;
struct node{
    int u,v,cap;
    node(){}
    node(int u,int v,int cap):u(u),v(v),cap(cap){}
}es[1000005];
int R,S,T;
int dis[200005],vis[200005],iter[200005];
vector<int> tab[200005];
void addedge(int u, int v, int cap){
    tab[u].push_back(R);
    es[R++]=node(u,v,cap);
    tab[v].push_back(R);
    es[R++]=node(v,u,0);
}
int bfs(){
    int i,h;
    queue<int> q;
    q.push(S);
    memset(dis,INF,sizeof(dis));
    dis[S]=0;
    while(q.size()){
        h=q.front();
        q.pop();
        for(i=0;i<tab[h].size();i++){
            node &e=es[tab[h][i]];
            if(e.cap>0&&dis[e.v]==INF){
                dis[e.v]=dis[h]+1;
                q.push(e.v);
            }
        }
    }
    return dis[T]<INF;
}
int dfs(int x,int maxflow){
    int flow;
    if(x==T)
    return maxflow;
    for(int &i=iter[x];i<tab[x].size();i++){
        node &e=es[tab[x][i]];
        if(dis[e.v]==dis[x]+1&&e.cap>0){
            flow=dfs(e.v,min(maxflow,e.cap));
            if(flow){
                e.cap-=flow;
                es[tab[x][i]^1].cap+=flow;
                return flow;
            }
        }
    }
    return 0;
}
int dinic(){
    int ans,flow;
    ans=0;
    while(bfs()){
        memset(iter,0,sizeof(iter));
        while(flow=dfs(S,INF))
        ans+=flow;
    }
    return ans;
}                                               //网络流模板
vector<int> state[1500];                        //因为左边有100000个点，右边只有
int main(){                                     //10个点，因此边可能达到一百万，因此
    int i,j,n,m,s,ans,num,tmp,siz;              //直接跑网络流一定会超时，可以将左边
    while(scanf("%d%d",&n,&m)!=EOF){            //进行缩点，每个人喜欢的可能只有2^10
        R=S=0,T=n+m+1;                          //因此用状态压缩进行缩点
        for(i=0;i<=T;i++)
        tab[i].clear();
        for(i=0;i<=1024;i++)
        state[i].clear();
        for(i=1;i<=n;i++){
            s=0;
            for(j=0;j<m;j++){
                scanf("%d",&num);
                if(num==1)
                s|=(1<<j);
            }
            state[s].push_back(i);
        }                                       //缩点
        for(i=0;i<(1<<m);i++){
            if(state[i].size()==0)
            continue;
            tmp=state[i][0];
            siz=state[i].size();
            addedge(S,tmp,siz);
            for(j=0;j<m;j++)                    //由原来容量为1改为缩点的个数
            if(i&(1<<j))
            addedge(tmp,n+j+1,siz);
        }
        for(i=1;i<=m;i++){
        scanf("%d",&num);
        addedge(i+n,T,num);
        }
        if(dinic()==n)
        puts("YES");
        else
        puts("NO");
    }
    return 0;
}