【学习笔记】「CEOI2014」 The Wall

这题有点困难😅

发现可以用从左上角到每个关键点的左上角的 最短路径 来代替边界，那么问题转化为从左上角出发不穿过边界的最短闭合回路，这样可以把每个城市都包住。

$\text{remark}$ 这个结论完全想不到😪

正解是把一个点拆成左上，右上，右下，左下四个点，然后跑最短路。没错，这就是经典的 网格图转平面图！

复杂度$O(nm\log nm)$。

#include<bits/stdc++.h>
#define ll long long
#define pb push_back
#define fi first
#define se second
using namespace std;
const int N=405;
const int M=405*405*4;
int id[N][N],id2[N][N][4],cnt,cnt2;
int n,m,a[N][N],b[N][N],s[N][N],ban[M],pre[M];
vector<pair<int,int>>G[M];
priority_queue<pair<ll,int>,vector<pair<ll,int>>,greater<pair<ll,int>>>Q;
ll dis[M];
int vs[M];
void add(int x,int y,int z){
    if(!x||!y)return;
    G[x].pb({y,z});
    G[y].pb({x,z});
}
void dijkstra(int s){
    memset(dis,0x3f,sizeof dis),memset(vs,0,sizeof vs);
    Q.push({0,s}),dis[s]=0;
    while(Q.size()){
        int u=Q.top().se;Q.pop();
        if(vs[u])continue;vs[u]=1;
        for(auto v:G[u]){
            if(dis[u]+v.se<dis[v.fi])dis[v.fi]=dis[u]+v.se,pre[v.fi]=u,Q.push({dis[v.fi],v.fi});
        }
    }
}
queue<int>que;
int state[M];
int check(int x,int y){
    return state[x]&&state[y]&&(pre[x]==y||pre[y]==x);
}
int main(){
    ios::sync_with_stdio(false);
    cin.tie(0),cout.tie(0);
    cin>>n>>m;
    for(int i=0;i<n;i++){
        for(int j=0;j<m;j++){
            cin>>s[i][j];
        }
    }
    for(int i=0;i<n;i++){
        for(int j=0;j<=m;j++){
            cin>>a[i][j];
        }
    }
    for(int i=0;i<=n;i++){
        for(int j=0;j<m;j++){
            cin>>b[i][j];
        }
    }
    for(int i=0;i<=n;i++){
        for(int j=0;j<=m;j++){
            id[i][j]=++cnt;
        }
    }
    for(int i=0;i<n;i++){
        for(int j=0;j<=m;j++){
            add(id[i][j],id[i+1][j],a[i][j]);
        }
    }
    for(int i=0;i<=n;i++){
        for(int j=0;j<m;j++){
            add(id[i][j],id[i][j+1],b[i][j]);
        }
    }
    dijkstra(id[0][0]);
    for(int i=1;i<=cnt;i++)G[i].clear();
    for(int i=0;i<n;i++){
        for(int j=0;j<m;j++){
            if(s[i][j]){
                ban[id[i][j]]=ban[id[i][j+1]]=ban[id[i+1][j]]=ban[id[i+1][j+1]]=1;
                que.push(id[i][j]);state[id[i][j]]=1;
            }
        }
    }
    while(que.size()){
        int x=que.front();que.pop();
        if(pre[x]&&!state[pre[x]])ban[pre[x]]=state[pre[x]]=1,que.push(pre[x]);
    }
    for(int i=0;i<=n;i++){
        for(int j=0;j<=m;j++){
            for(int k=0;k<4;k++){
                id2[i][j][k]=++cnt2;
            }
        }
    }
    for(int i=0;i<n;i++){
        for(int j=0;j<=m;j++){
            add(id2[i][j][2],id2[i+1][j][1],a[i][j]);
            add(id2[i][j][3],id2[i+1][j][0],a[i][j]);
        }
    }
    for(int i=0;i<=n;i++){
        for(int j=0;j<m;j++){
            add(id2[i][j][1],id2[i][j+1][0],b[i][j]);
            add(id2[i][j][2],id2[i][j+1][3],b[i][j]);
        }
    }
    for(int i=0;i<=n;i++){
        for(int j=0;j<=m;j++){
            if(i==0&&j==0)continue;
            if(!i||!check(id[i][j],id[i-1][j])&&(j==0||!s[i-1][j-1])&&!s[i-1][j])add(id2[i][j][0],id2[i][j][1],0);
            if(j==m||!check(id[i][j],id[i][j+1])&&(i==0||!s[i-1][j])&&!s[i][j])add(id2[i][j][1],id2[i][j][2],0);
            if(i==n||!check(id[i][j],id[i+1][j])&&(j==0||!s[i][j-1])&&!s[i][j])add(id2[i][j][2],id2[i][j][3],0);
            if(!j||!check(id[i][j],id[i][j-1])&&(i==0||!s[i-1][j-1])&&!s[i][j-1])add(id2[i][j][3],id2[i][j][0],0);
        }
    }
    dijkstra(id2[0][0][1]);
    cout<<dis[id2[0][0][3]];
}