题意
一张连通有向图,另一条边反向,是否缩短了1到n的最短路?
保证开始给定的图从城市1可以到达城市n,若边反向后城市1不能到达城市n,我们视为最短路径长度没有变短。
思路
数据范围较大,最原始的暴力每次djikstra一遍肯定是TLE的,于是思考预处理,预先djiksra一遍。
每次反向一条边和没反向有什么区别呢?想到直接在原图上加一条反向边,原来的边不管他不影响结果。
这时候画个图
当我们反转2←4这条路,使得其变为2→4,新的最短路路径的算法为 1到2的距离 + 4到2的距离 + 4到5的距离。
我们将这个模型一般化,就是 和原先的最短路比较即可。
问题是我们知道了1到i的最短路还不知道j到n的最短路呢,难道我们要求所有两点间的最短路吗?那样一定会TLE,我们的终点已经确定了的情况,我们只需要反向建图,再将n作为起点在进行一次djikstra即可。
#include<bits/stdc++.h>
#define fi first
#define se second
using namespace std;
typedef long long ll;
typedef pair<ll,ll>P;
const double eps = 1e-8;
const int NINF = 0xc0c0c0c0;
const ll INF = 0x3f3f3f3f3f3f3f3f;
const ll mod = 1e9 + 7;
const ll maxn = 1e6 + 5;
const int N = 1e5 + 5;
int u[2*N],v[2*N],c[2*N];
struct edge{int to,cost;};
ll n,m,d[N],e[N];
vector<edge>G[N];
vector<edge>F[N];
void Dji1(const int& s,const int& V){
priority_queue<P,vector<P>,greater<P> >q;
memset(d,0x3f,sizeof(d));
d[s]=0;
q.push({0,s});
while(!q.empty()){
P t=q.top();q.pop();
ll v=t.se;
if(d[v]<t.fi) continue;
for(int i=0;i<G[v].size();i++){
edge e=G[v][i];
if(d[e.to]>d[v]+e.cost){
d[e.to]=d[v]+e.cost;
q.push({d[e.to],e.to});
}
}
}
}
void Dji2(const int& s,const int& V){
priority_queue<P,vector<P>,greater<P> >q;
memset(e,0x3f,sizeof(e));
e[s]=0;
q.push({0,s});
while(!q.empty()){
P t=q.top();q.pop();
ll v=t.se;
if(e[v]<t.fi) continue;
for(int i=0;i<F[v].size();i++){
edge k=F[v][i];
if(e[k.to]>e[v]+k.cost){
e[k.to]=e[v]+k.cost;
q.push({e[k.to],k.to});
}
}
}
}
int main(){
ios::sync_with_stdio(false);
cin.tie(0);
cin>>n>>m;
for(int i=1;i<=m;i++){
cin>>u[i]>>v[i]>>c[i];
G[u[i]].push_back({v[i],c[i]});
F[v[i]].push_back({u[i],c[i]});
}
Dji1(1,n);
Dji2(n,n);
int T;cin>>T;
while(T--){
int x;
cin>>x;
int i=u[x],j=v[x],p=c[x];
ll temp=d[j]+p+e[i];
if(temp<d[n]) cout<<"YES\n";
else cout<<"NO\n";
}
return 0;
}
京公网安备 11010502036488号