7-10 Dijkstra算法(模板) (30 分)
给一个n(1 ≤ n ≤ 2500) 个点 m(1 ≤ m ≤ 6200) 条边的无向图,求 s 到 t 的最短路。
输入格式:
第一行四个由空格隔开的整数 n、m、s、t。
之后的 m 行,每行三个正整数 si、ti、wi(1≤wi≤10^9),表示一条从si到 ti 长度为 wi 的边。
输出格式:
一个整数,表示从s 到t 的最短路径长度。数据保证至少存在一条道路。
输入样例:
7 11 5 4
2 4 2
1 4 3
7 2 2
3 4 3
5 7 5
7 3 3
6 1 1
6 3 4
2 4 3
5 6 3
7 2 1
输出样例:
7
#include<iostream>
#include<queue>
#define inf 0x3f3f3f3f
using namespace std;
const int maxn = 1e5+5;
priority_queue< pair<long long,int> > q; //优先队列
struct node{
int nxt,to,dis;
}edge[maxn<<1];
int n,m,s,t,R,dis[maxn],head[maxn<<1];
void add(int from,int to,int dis){ //链式前向星建图
edge[++R].nxt=head[from];
edge[R].to=to;
edge[R].dis=dis;
head[from]=R;
}
void mem(){ //初始化设每个点距离和未被访问
for(int i=1;i<=n;++i) dis[i]=inf;
}
void dijkstra(){ //堆优化dijkstra
dis[s]=0; vis[s]=1; q.push(make_pair(0,s));
while(!q.empty()){
int tmp=q.top().second;
int d=-q.top().first;
if(d!=dis[tmp]) continue;
q.pop();
for(int i=head[tmp];i;i=edge[i].nxt){
int to=edge[i].to;
if(dis[to]>dis[tmp]+edge[i].dis){
dis[to]=dis[tmp]+edge[i].dis;
q.push(make_pair(-dis[to],to));
}
}
}
cout<<dis[t];
}
int main(){
cin>>n>>m>>s>>t;
for(int i=0,u,v,w;i<m;++i){
cin>>u>>v>>w; add(u,v,w); add(v,u,w);
}mem();
dijkstra();
return 0;
}
qwq累了,摸了。