【bzoj 2834】回家的路（分层图最短路）

传送门biu~
把一个点拆成两个点，一个代表横向连接的点，一个代表纵向连接的点，用权为1的边连接。首先对于每个横向连接的点，同一行按顺序连边，权值为距离的二倍，纵向同理。跑一遍最短路就可以了。

#include<bits/stdc++.h>
using namespace std;
int n,m;
int head[200005],nex[1000005],to[1000005],v[1000005],tp;
int dis[200005];
bool in[200005];
inline void add(int x,int y,int z){
    nex[++tp]=head[x];
    head[x]=tp;
    to[tp]=y;
    v[tp]=z;
}
const int INF=1e9;
struct point{int x,y,num;}a[100005];
bool cmp1(point a,point b){return a.y<b.y || (a.y==b.y && a.x<b.x);}
bool cmp2(point a,point b){return a.x<b.x || (a.x==b.x && a.y<b.y);}
inline void spfa(){
    queue<int>q;dis[m+1]=0;
    q.push(m+1);
    while(!q.empty()){
        int x=q.front();q.pop();in[x]=0;
        for(int i=head[x];i;i=nex[i]){
            if(dis[x]+v[i]<dis[to[i]]){
                dis[to[i]]=dis[x]+v[i];
                if(!in[to[i]]){
                    in[to[i]]=1;
                    q.push(to[i]);
                }
            }
        }
    }
}
int main(){
    memset(dis,127,sizeof(dis));
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m+2;++i) scanf("%d%d",&a[i].x,&a[i].y),a[i].num=i;
    sort(a+1,a+m+3,cmp1);
    for(int i=1;i<m+2;++i)
        if(a[i].y==a[i+1].y)    add(a[i].num,a[i+1].num,(a[i+1].x-a[i].x)*2),add(a[i+1].num,a[i].num,(a[i+1].x-a[i].x)*2);
    sort(a+1,a+m+3,cmp2);
    for(int i=1;i<m+2;++i)
        if(a[i].x==a[i+1].x)    add(a[i].num+m+2,a[i+1].num+m+2,(a[i+1].y-a[i].y)*2),add(a[i+1].num+m+2,a[i].num+m+2,(a[i+1].y-a[i].y)*2);
    for(int i=1;i<=m;++i)       add(i,i+m+2,1),add(i+m+2,i,1);
    for(int i=m+1;i<=m+2;++i)   add(i,i+m+2,0),add(i+m+2,i,0);
    spfa();
    printf("%d",dis[m+2]);
    return 0;
}