HDU 3605 Escape 最大流

题意：

如果这是2012年世界末日怎么办?我不知道该怎么做。但是现在科学家们已经发现，有些星球上的人可以生存，但有些人却不适合居住。现在科学家们需要你的帮助，就是确定所有人都能在这些行星上生活。输入多组测试数据，每组数据的开头是n (1 <= n <= 100000)， m (1 <= m <= 10) n表示地球上有n个人，m代表m星球，星球和人的标签都是从0开始的。这里有n行，每一行代表一个合适的居住条件的人，每一行有m个数字，第i个数字是1，表示一个人适合居住在第i个星球上，或者是0表示这个人不适合居住在第i个星球上。最后一行有m个数字，第i个数字ai表示第i个星球最能容纳ai人。0 <= ai <= 100000输出确定是否所有人都能达到这些标准如果可以输出YES，则输出NO。

代码：

//这道题一看见是一个最大流模板，但是这道题就是为了卡时间。。。
//因为题目上面n特别大，这个时候因为m只有10，此时最大也就只有2^10个选择，所以有好多人的选择是一样的
//那么我们要按照他们的选择来建图，这样的话边的数量就会大大减少
#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#include<queue>
#include<math.h>
#include<stdlib.h>
#include<vector>
using namespace std;
const int maxn=200005;
const int INF=0x3f3f3f3f;
int head[maxn],cnt,st,en,dis[maxn],cur[maxn],v[maxn];
struct edge
{
    int v,next,c,flow;
} e[1000005];
vector<int> state[maxn];
void add_edge(int x,int y,int z)
{
    e[cnt].v=y;
    e[cnt].c=z;
    e[cnt].flow=0;
    e[cnt].next=head[x];
    head[x]=cnt++;

    e[cnt].v=x;
    e[cnt].c=z;
    e[cnt].flow=0;
    e[cnt].next=head[y];
    head[y]=cnt++;
}
bool bfs()
{
    memset(dis,0,sizeof(dis));
    dis[st]=1;
    queue<int>r;
    r.push(st);
    while(!r.empty())
    {
        int x=r.front();
        r.pop();
        for(int i=head[x]; i!=-1; i=e[i].next)
        {
            int v=e[i].v;
            if(!dis[v] && e[i].c>e[i].flow)
            {
                dis[v]=dis[x]+1;
                r.push(v);
            }
        }
    }
    return dis[en];
}
int dinic(int s,int limit)
{
    if(s==en || !limit) return limit;
    int ans=0;
    for(int &i=cur[s]; i!=-1; i=e[i].next)
    {
        int v=e[i].v,feed;
        if(dis[v]!=dis[s]+1) continue;
        feed=dinic(v,min(limit,e[i].c-e[i].flow));
        if(feed)
        {
            e[i].flow+=feed;
            e[i^1].flow-=feed;
            limit-=feed;
            ans+=feed;
            if(limit==0) break;
        }
    }
    if(!ans) dis[s]=-1;
    return ans;
}
int main()
{
    int n,m;
    char s[25][25];
    int w[25][25];
    while(~scanf("%d%d",&n,&m))
    {
        memset(head,-1,sizeof(head));
        cnt=0;
        int x;
        st=0;
        en=1024+2*m+1;
        memset(v,0,sizeof(v));
//        for(int i=1;i<=n;++i)   //这种方法建图的时候它的终止点en要大于1024，这样的话跑的话太慢了
//        {   //所以这个时候en的大小要改变成n+m+1,这个时候就要用到n了，正确建图方式见下面
//            int y=1,sum=0;
//            for(int j=1;j<=m;++j)
//            {
//                scanf("%d",&x);
//                if(x)
//                {
//                    sum+=y;
//                }
//                y*=2;
//            }
//            v[sum]++;
//        }
//        if(v[0])
//        {
//            printf("NO
");
//            continue;
//        }
//        for(int i=1;i<=1024;++i)
//        {
//            if(v[i])
//            {
//                add_edge(st,i,v[i]);
//                add_edge(i,st,0);
//            }
//            for(int j=0;j<m;++j)
//            {
//                if((1<<j)&i)
//                {
//                    add_edge(i,1024+1+j,INF);
//                    add_edge(1024+1+j,i,0);
//                }
//            }
//        }
//        for(int i=1;i<=m;++i)
//        {
//            scanf("%d",&x);
//            add_edge(1024+i,en,x);
//            add_edge(en,1024+i,0);
//        }
        st=0,en=n+m+1;
        int s,i,j,num;                          //因此用状态压缩进行缩点
        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);  //C语言中的 |= 意思为：按位或后赋值
            }
            state[s].push_back(i);
        }                                       //缩点
        int tmp,siz;
        for(i=0; i<(1<<m); i++)  //这点就比我原来的代码优化了
        {
            if(state[i].size()==0)
                continue;
            tmp=state[i][0];
            siz=state[i].size();
            add_edge(st,tmp,siz);
            for(j=0; j<m; j++)                  //由原来容量为1改为缩点的个数
                if(i&(1<<j))
                    add_edge(tmp,n+j+1,siz);
        }
        for(i=1; i<=m; i++)
        {
            scanf("%d",&num);
            add_edge(i+n,en,num);
        }

        int ans=0;
        while(bfs())
        {
            for(int i=0; i<=en; i++)
                cur[i]=head[i];
            ans+=dinic(st,INF);
        }
        if(ans==n)
        {
            printf("YES
");
        }
        else printf("NO
");
    }
    return 0;
}