code3731 寻找道路

Posted 2020-07-28 FuTaimeng

将图反向，从终点spfa，把和终点没有联系的点找出来，

把它们和它们所连接的所有点删去（反向图）

然后再一遍spfa，输出最短路即可

 

代码：

#include<iostream>
#include<cstring>
#include<queue>
#define MAXM 200005
#define MAXN 10005
#define INF 0x3f3f3f3f
using namespace std;

int n,m;
struct edge{
    int to,next,w;
}eg[MAXM];
int head[MAXN];
int cnt=0;
int s,t;
int dist[MAXN];
bool canUse[MAXN];
bool vis[MAXN];

void add(int a,int b,int w){
    cnt++;
    eg[cnt].to=b;
    eg[cnt].next=head[a];
    eg[cnt].w=w;
    head[a]=cnt;
}

void spfa(int st){
    memset(dist,0x7f,sizeof(dist));
    memset(vis,false,sizeof(vis));
    
    queue<int> q;
    if(canUse[st]){
        q.push(st);
        dist[st]=0;
        vis[st]=true;
    }
    
    while(!q.empty()){
        int k=q.front(); q.pop();
        vis[k]=false;
        for(int i=head[k];i!=-1;i=eg[i].next){
            int u=eg[i].to;
            if(canUse[u]){
                if(dist[u]>dist[k]+eg[i].w){
                    dist[u]=dist[k]+eg[i].w;
                    if(!vis[u]){
                        vis[u]=true;
                        q.push(u);
                    }
                }
            }
        }
    }
}

int main(){
    memset(canUse,true,sizeof(canUse));
    memset(head,-1,sizeof(head));
    
    cin>>n>>m;
    int a,b,w;
    for(int i=1;i<=m;i++){
        cin>>a>>b;
        add(b,a,1);
    }
    cin>>s>>t;
    spfa(t);
    for(int i=1;i<=n;i++){
        if(dist[i]>=INF){
            //cout<<"Can‘t Use:"<<i<<endl;
            canUse[i]=false;
            for(int j=head[i];j!=-1;j=eg[j].next){
                canUse[eg[j].to]=false;
            }
        }
    }
    spfa(t);
    if(dist[s]>=INF)cout<<-1<<endl;
    else cout<<dist[s]<<endl;
    return 0;
}