题意:给一张图,1为起点,n为终点 求有权图单源最短路。
解题思路:dijkstra+堆优化。用优先队列来实现最小堆。新学到了优先队列和pair的用法
优先队列:
priority_queue < P, vector < P >, greater<P > >que;
这样是定义了升序队列。包含在头文件<queue>中。
pair:
定义: 1.定义时初始化:pair<int,int>t(1,2)
2.先定义,后初始化:
pair<int,int>t;
t=make_pair(1,2);
排序时,默认用的first。
代码如下:
#include<cstdlib>
#include<cstdio>
#include<algorithm>
#include<queue>
#include<vector>
#define N 1010
#define inf 0x3f3f3f3f
using namespace std;
int dis[N];
int t, n;
struct edge
{
int to;
int cost;
};
vector<edge>G[N];
typedef pair<int, int> P;
void dijkstra(int s)
{
priority_queue < P, vector < P >, greater<P > >que;
fill(dis, dis + n + 1, inf);
dis[s] = 0;
que.push(P(0, s));
while (!que.empty())
{
P p = que.top();
que.pop();
int v = p.second;
if (dis[v] < p.first)continue;//防止负值圈进入死循环
for (int i = 0;i < G[v].size();i++)
{
edge e = G[v][i];
if (dis[e.to] > dis[v] + e.cost)
{
dis[e.to] = dis[v] + e.cost;
que.push(P(dis[e.to], e.to));
}
}
}
}
int main()
{
int a, b, l;
scanf("%d %d", &t, &n);
for (int i = 0;i < t;i++)
{
scanf("%d%d%d", &a, &b, &l);
G[a].push_back({ b,l });
G[b].push_back({ a,l });
}
dijkstra(1);
printf("%d\n", dis[n]);
return 0;
}