洛谷 2296 寻找道路

【题解】

　　先建反向图，dfs求出哪些点可以到达终点。

　　再建正向图，dfs求出哪些点可以作为路径上的点。

　　最后在合法的点之间连边，跑dijkstra.

#include<cstdio>
#include<cstring>
#include<algorithm>
#define LL long long
#define rg register
#define N 10010
#define M 200010
using namespace std;
int n,m,tot,st,ed,fa,son,last[N],dis[N],pos[N];
bool arr[N],ok[N],vis[N];
struct edge{int to,pre;}e[M];
struct heap{int p,d;}h[N];
struct rec{int u,v;}r[M];
inline int read(){
    int k=0,f=1; char c=getchar();
    while(c<'0'||c>'9')c=='-'&&(f=-1),c=getchar();
    while('0'<=c&&c<='9')k=k*10+c-'0',c=getchar();
    return k*f;
}
void dfs(int x){
    arr[x]=1;
    for(rg int i=last[x],to;i;i=e[i].pre)if(!arr[to=e[i].to]) dfs(to);
}
void dfs2(int x){
    vis[x]=1; bool can=1;
    for(rg int i=last[x],to;i;i=e[i].pre){
        if(!arr[to=e[i].to]) can=0;
        if(!vis[to]) dfs2(to);
    }
    ok[x]=can&arr[x];
}
inline void up(int x){
    while((fa=x>>1)&&h[fa].d>h[x].d) swap(h[x],h[fa]),swap(pos[h[x].p],pos[h[fa].p]),x=fa;
}
inline void down(int x){
    while((son=x<<1)<=tot){
        if(h[son+1].d<h[son].d&&son<tot) son++;
        if(h[son].d<h[x].d) swap(h[son],h[x]),swap(pos[h[son].p],pos[h[x].p]),x=son;
        else return;
    }
}
void dijkstra(int x){
    for(rg int i=1;i<=n;i++) dis[i]=1e9;
    h[tot=pos[x]=1]=(heap){x,dis[x]=0};
    while(tot){
        int now=h[1].p; pos[h[tot].p]=1; h[1]=h[tot--]; if(tot) down(1);
        for(rg int i=last[now],to;i;i=e[i].pre)if(dis[to=e[i].to]>dis[now]+1){
            dis[to]=dis[now]+1;
            if(!pos[to]) h[pos[to]=++tot]=(heap){to,dis[to]};
            else h[pos[to]].d=dis[to];
            up(pos[to]);
        }
    }
}
int main(){
    n=read(); m=read();
    for(rg int i=1,u,v;i<=m;i++){
         u=r[i].u=read(),v=r[i].v=read();
         e[++tot]=(edge){u,last[v]}; last[v]=tot;
    }
    st=read(); ed=read();
    dfs(ed);
//    for(rg int i=1;i<=n;i++) printf("%d ",arr[i]); puts("arr");
    
    memset(last,0,sizeof(last)); tot=0;
    for(rg int i=1;i<=m;i++){
        int u=r[i].u,v=r[i].v;
        e[++tot]=(edge){v,last[u]}; last[u]=tot;
    }
    dfs2(st); ok[ed]=1;
//    for(rg int i=1;i<=n;i++) printf("%d ",ok[i]); puts("ok");
    if(!ok[st]){puts("-1"); return 0;}
    
    memset(last,0,sizeof(last)); tot=0;
    for(rg int i=1,u,v;i<=m;i++)if(ok[u=r[i].u]&&ok[v=r[i].v])
        e[++tot]=(edge){v,last[u]},last[u]=tot;
    dijkstra(st);
    if(dis[ed]!=1e9) printf("%d
",dis[ed]);
    else puts("-1");
    return 0; 
}