HDU3605 Escape（缩点+二进制+最大流）

题目链接

http://acm.hdu.edu.cn/showproblem.php?pid=3605

读题

给定n个人和m个星球。
通过一个n*m矩阵来描述某人是否适合生存在某星球。
通过一行m个数组来标是每个星球的最大容纳量。
求能否满足每个人都被安排到其适合居住的星球。

解题

人与星球进行匹配，星球有最大容纳量。明显的二分图多重匹配模型。
用匈牙利算法可解。
而用最大流解的话，设立一个超级源点S，源点到每个人连一条容量为1的有向弧，人到星球连一条容量为1的有向弧，星球到超级汇点T连一条容量为星球最大容纳量的有向弧。

但是dinic算法的时间复杂度上界为O（n^n*m).而n的数量级到达了1e5级别。所以需要优化。
我记得有一个数据级别到达1e5的最大流题目给的时限是10秒，通过当前弧优化9秒多卡过。这里只给了2秒，卡的应该不是当前弧优化策略。

注意到m的值非常小。这么反常，解决方案肯定在m上。
如果两个不同的人对星球的要求是一致的，那么可以将这两个人看作一个人，将容量合并。对每个人根据与星球关系重新编号，即用n*m矩阵中第i行的数的二进制值作为第i个人的新编号。m最多为10，对人合并后最多有2^10个“人”。如此，复杂度大大降低。

AC代码

//1544ms 2068kB
#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
#include <map>
#include <vector>
#include <string>
#define INF 0x3f3f3f3f
using namespace std;

int S,T;//源点和汇点
const int maxe=1e5+1000;
const int maxv=1e4+100;
struct edge
{
    int to,w,next;
}e[maxe<<1];
int cur[maxv<<1];
int n,m,sum;
int head[maxv<<1],depth[maxv],cnt;
void init()
{
    memset(head,-1,sizeof(head));
    cnt=-1;
}
void add_edge(int u,int v,int w)
{
    e[++cnt].to=v;
    e[cnt].w=w;
    e[cnt].next=head[u];
    head[u]=cnt;
}
void _add(int u,int v,int w)
{
    add_edge(u,v,w);
    add_edge(v,u,0);
}

bool bfs()
{
    queue<int> q;
    while(!q.empty()) q.pop();
    memset(depth,0,sizeof(depth));
    depth[S]=1;
    q.push(S);

    while(!q.empty())
    {
        int u=q.front();q.pop();
        for(int i=head[u];i!=-1;i=e[i].next)
        {
            if(!depth[e[i].to] && e[i].w>0)
            {
                depth[e[i].to]=depth[u]+1;
                q.push(e[i].to);
            }
        }
    }
    if(!depth[T]) return false;
    return true;
}
int dfs(int u,int flow)
{
    if(u==T) return flow;
    int ret=0;
    for(int i=head[u];i!=-1 && flow;i=e[i].next)
    {
        if(depth[u]+1==depth[e[i].to] && e[i].w!=0)
        {
            int tmp=dfs(e[i].to,min(e[i].w,flow));
            if(tmp>0)
            {
                flow-=tmp;
                ret+=tmp;
                e[i].w-=tmp;
                e[i^1].w+=tmp;
            }
        }
    }
    return ret;
}
int Dinic()
{
    int ans=0;
    while(bfs())
    {
        ans+=dfs(S,INF);
    }
    return ans;
}

int b[1200];//人对应新编号结点的容量
int main()
{
    while(~scanf("%d%d",&n,&m))
    {
        init();
        memset(b,0,sizeof(b));
        sum=n;
        S=0,T=2000;
        int tmp,x;
        for(int i=0;i<n;i++)
        {
            tmp=0;
            for(int j=0;j<m;j++)
            {
                scanf("%d",&x);
                if(x) tmp|=(1<<j);
            }
            b[tmp]++;
        }
        for(int i=1;i<=1024;i++)//直接枚举新结点编号，枚举原编号再映射会超时
        {
            if(!b[i]) continue;
            _add(S,i,b[i]);
            for(int j=0;j<m;j++)
            {
                if(i&(1<<j))//通过二进制比对来连边
                    _add(i,j+1500,b[i]);
            }
        }
        for(int i=0;i<m;i++)
        {
            scanf("%d",&x);
            _add(i+1500,T,x);
        }
        int ans=Dinic();
        printf("%s\n",ans==sum?"YES":"NO");
    }
    return 0;
}