题目描述
A \mathrm{A} A 国有 N N N 个城市,编号为 1 … N 1 \ldots N 1…N 小明是编号为 1 1 1 的城市中一家公司的员工,今天突然接到了上级通知需要去编号为 N N N 的城市出差。
由于疫情原因,很多直达的交通方式暂时关闭,小明无法乘坐飞机直接从城市 1 1 1 到达城市 N N N,需要通过其他城市进行陆路交通中转。小明通过交通信息网,查询到了 M M M 条城市之间仍然还开通的路线信息以及每一条路线需要花费的时间。
同样由于疫情原因,小明到达一个城市后需要隔离观察一段时间才能离开该城市前往其他城市。通过网络,小明也查询到了各个城市的隔离信息。(由于小明之前在城市 1 1 1,因此可以直接离开城市 1 1 1,不需要隔离)
由于上级要求,小明希望能够尽快赶到城市 N \mathrm{N} N, 因此他求助于你,希望你能帮他规划一条路线,能够在最短时间内到达城市 N N N 。
输入格式
第 1 1 1 行:两个正整数 N , M N, M N,M 表示 A 国的城市数量, M M M 表示末关闭的路线数量。
第 2 2 2 行: N N N 个正整数,第 i i i 个整数 C i C_{i} Ci 表示到达编号为 i \mathrm{i} i 的城市后需要隔离的时间。
第 3 … M + 2 3 \ldots M+2 3…M+2 行: 每行 3 3 3 个正整数, u , v , c u, v, c u,v,c, 表示有一条城市 u u u 到城市 v v v 的双向路线仍然开通着,通过该路线的时间为 c c c。
输出格式
第 1 1 1 行: 1 1 1 个正整数,表示小明从城市 1 1 1 出发到达城市 N N N 的最短时间。(到达城市 N N N,不需要计算城市 N N N 的隔离时间)
样例 #1
样例输入 #1
4 4
5 7 3 4
1 2 4
1 3 5
2 4 3
3 4 5
样例输出 #1
13
提示
【样例说明】
【评测用例规模与约定】
对于 100 % 100 \% 100% 的数据, 1 ≤ N ≤ 1000 , 1 ≤ M ≤ 10000 , 1 ≤ C i ≤ 200 , 1 ≤ u , v ≤ 1 \leq N \leq 1000,1 \leq M \leq 10000,1 \leq C_{i} \leq 200,1 \leq u, v \leq 1≤N≤1000,1≤M≤10000,1≤Ci≤200,1≤u,v≤ N , 1 ≤ c ≤ 1000 N, 1 \leq c \leq 1000 N,1≤c≤1000
解析
将目标点的权值加到该边的权值上,跑一遍Dijkstra即可
#include<bits/stdc++.h>
using namespace std;
#define int long long
#define endl '\n'
#define INF 0x3f3f3f3f
const int N=2e5+5;
int n,m;
int a[N],dis[N],done[N];
struct edge{
int to,w;
};
struct node{
int id,dis;
bool operator<(const node& a)const{
return dis>a.dis;
}
};
vector<edge>e[N];
void Dijkstra(){
int s=1;
for(int i=1;i<=n;i++) dis[i]=INF;
priority_queue<node>q;
q.push({1,0});
dis[1]=0;
while(q.size()){
node u=q.top();
q.pop();
if(done[u.id]) continue;
done[u.id]=1;
for(int i=0;i<e[u.id].size();i++){
edge x=e[u.id][i];
if(done[x.to]) continue;
if(dis[x.to]>x.w+u.dis){
dis[x.to]=x.w+u.dis;
q.push({x.to,dis[x.to]});
}
}
}
cout<<dis[n]-a[n]<<endl;
}
void solve(){
cin>>n>>m;
for(int i=1;i<=n;i++){
cin>>a[i];
}
for(int i=1;i<=m;i++){
int x,y,w;
cin>>x>>y>>w;
e[x].push_back({y,w+a[y]});
e[y].push_back({x,w+a[x]});
}
Dijkstra();
}
signed main(){
ios::sync_with_stdio(false);
cin.tie(nullptr);
cout.tie(nullptr);
int t=1;
// cin>>t;
while(t--) solve();
return 0;
}