题解 poj3057 Evacuation

Posted 2022-09-19 white-star

算法：匈牙利算法

复杂度：不好算，懒得算

这道题我调了两个半小时，代码长，细节多，现总结一下栽过的坑

1. 此题数组传参的方式很好，能够降低代码难度，值得学习
2. 如何使人和时间、门的二元组互不冲突很重要
3. 如果点无冲突用链表，否则用vector
4. 点是从0开始建的，初始化match数组不能用0，我虽然注意到了这一点却没调出来

丑陋的代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
#include<queue>
#include<algorithm>
using namespace std;

#define rep(i,a,b) for(int i=a;i<b;++i)
#define go(i,a,b) for(int i=a;i<=b;++i)
#define mem(a,b) memset(a,b,sizeof(a))

const int N=1000010;

int n,m,d[15][15][15][15],match[N],cnt=0,tot=0,head[N],p,door,ans=0;
int match2[N];
bool vis[N];
char g[15][15];
int nd[4][2]= 1,0,-1,0,0,1,0,-1,V;
vector<int> dx,dy,px,py,e[250];
/*struct edge
    int nxt,v;
e[N];

void add(int u,int v)
    e[cnt]=(edge)head[u],v;
    head[u]=cnt++;
*/

void add(int u,int v)
    e[u].push_back(v);


void bfs(int x,int y,int d[15][15])
    queue<int>qx,qy;
    d[x][y]=0;
    qx.push(x),qy.push(y);
    while(!qx.empty())
        x=qx.front();qx.pop();
        y=qy.front();qy.pop();
        go(k,0,3)
            int x2=x+nd[k][0],y2=y+nd[k][1];
            if(g[x2][y2]=='.'&&d[x2][y2]<0)
                d[x2][y2]=d[x][y]+1;
                qx.push(x2),qy.push(y2);
            
        
    


bool dfs(int u)
    /*for(int i=head[u];i+1;i=e[i].nxt)
        int v=e[i].v;
        if(vis[v]) continue;
        vis[v]=1;
        if(!match[v]||dfs(match[v]))
            match[v]=u;match[u]=v;
            return 1;
        
    */
    for(int i=0;i<e[u].size();++i)
        int v=e[u][i];
        if(vis[v]) continue;
        vis[v]=1;
        if(match[v]==-1||dfs(match[v]))
            match[v]=u;match2[u]=v;
            return 1;
        
    
    return 0;


void work(int k)
    for(int i=0;i<p;++i)
        if(match2[i]==-1)
            for(int j=0;j<door*k;j++) vis[j]=false;
            ans+=dfs(i);
        
    


void solve()
    tot=n*m;
    dx.clear(),dy.clear(),px.clear(),py.clear();
    go(i,0,p-1)
        e[i].clear();
    mem(d,-1);
    rep(x,0,n)
        rep(y,0,m)
            if(g[x][y]=='D')
                dx.push_back(x),dy.push_back(y);
                bfs(x,y,d[x][y]);
            
            else if(g[x][y]=='.')
                px.push_back(x),py.push_back(y);
            
        
    door=dx.size(),p=px.size();
    if(!p)
        puts("0");
        return;
    
    mem(match,-1);mem(match2,-1); 
    mem(head,-1);cnt=0;
    ans=0;
    int k;
    for(k=1;k<tot;++k)
        
        //for(int i=0;i<=30;++i) printf("%d %d ",match[i],match2[i]);
        //puts("");
        
        rep(i,0,p)
            rep(j,0,door)
                
                //printf("%d\n",d[dx[j]][dy[j]][px[i]][py[i]]);
                
                if(d[dx[j]][dy[j]][px[i]][py[i]]!=-1&&d[dx[j]][dy[j]][px[i]][py[i]]<=k)
                    add(i,j+door*(k-1));
            
            
        work(k);
        
    //  printf("%d\n",ans);
        
        if(ans==p)
            printf("%d\n",k);
            return;
        
    
    if(k==tot) puts("impossible");
    return;


int main()
    //freopen("input.txt","r",stdin);
    //freopen("wrong.txt","w",stdout);
    int T;
    scanf("%d",&T);
    while(T--)
        cin>>n>>m;
        go(i,0,n-1) scanf("%s",g[i]);
        //printf("%d\n",T);
        solve();
    
    //int x=5;
    //printf("%d",x*x);
    return 0;