code3731 寻找道路

将图反向,从终点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=;
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]=;
vis[st]=true;
} while(!q.empty()){
int k=q.front(); q.pop();
vis[k]=false;
for(int i=head[k];i!=-;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,-,sizeof(head)); cin>>n>>m;
int a,b,w;
for(int i=;i<=m;i++){
cin>>a>>b;
add(b,a,);
}
cin>>s>>t;
spfa(t);
for(int i=;i<=n;i++){
if(dist[i]>=INF){
//cout<<"Can't Use:"<<i<<endl;
canUse[i]=false;
for(int j=head[i];j!=-;j=eg[j].next){
canUse[eg[j].to]=false;
}
}
}
spfa(t);
if(dist[s]>=INF)cout<<-<<endl;
else cout<<dist[s]<<endl;
return ;
}
上一篇:循环赛日常表算法(N可为奇数和偶数)


下一篇:Python MySQLdb 插入数据