题目入口
题意:给定一个n个点m条边的有向图,图中可能存在重边和自环,所有边权均为非负值。
请你求出1号点到n号点的最短距离,如果无法从1号点走到n号点,则输出-1。
输入格式
第一行包含整数n和m。
接下来m行每行包含三个整数x,y,z,表示存在一条从点x到点y的有向边,边长为z。
输出格式
输出一个整数,表示1号点到n号点的最短距离。
如果路径不存在,则输出-1。
数据范围
1≤n,m≤1.5×10^5,
图中涉及边长均不小于0,且不超过10000。
输入样例:
3 3
1 2 2
2 3 1
1 3 4
输出样例:
3
思路:
其实堆优化的Dijkstra和上一篇博客朴素的Dijkstra基本原理是一样的。
只不过朴素的是找到最小边后去更新n个点,但如果n比较大时O(n^2)的效率就不行了。
这个时候就可以找到最小边后去更新m条边,而且为了提高效率还可以用堆即优先队列来寻找,这样找最小值的效率是差不多logn的, 整个堆优化的Dijkstra效率就变成了O(mlogn)。
代码实现:
#include <cstdio>
#include <cstring>
#include <cmath>
#include <cstdlib>
#include <ctime>
#include <cctype>
#include <cstring>
#include <iostream>
#include <sstream>
#include <string>
#include <list>
#include <vector>
#include <set>
#include <map>
#include <queue>
#include <stack>
#include <algorithm>
#include <functional>
#define lowbit(x) (x &(-x))
#define me(ar) memset(ar, 0, sizeof ar)
#define mem(ar,num) memset(ar, num, sizeof ar)
#define rp(i, n) for(int i = 0, i < n; i ++)
#define rep(i, a, n) for(int i = a; i <= n; i ++)
#define pre(i, n, a) for(int i = n; i >= a; i --)
#define IOS ios::sync_with_stdio(0); cin.tie(0);cout.tie(0);
const int way[4][2] = {{1, 0}, {-1, 0}, {0, 1}, {0, -1}};
using namespace std;
typedef long long ll;
typedef pair<ll,ll> pll;
typedef pair<int, int> pii;
const int INF = 0x3f3f3f3f;
const double PI = acos(-1.0);
const double EXP = 1e-8;
const ll MOD = 1e9 + 7;
const int N = 1e6 + 5;
int n, m;
int h[N], e[N], ne[N], w[N], idx;
int dist[N];
bool st[N];
void add(int a, int b, int c)
{
e[idx] = b;
w[idx] = c;
ne[idx] = h[a];
h[a] = idx ++;
}
int Dijkstra()
{
mem(dist, 0x3f);
dist[1] = 0;
priority_queue<pii, vector<pii>, greater<pii> > heap;
//先把1号点加入队列
heap.push({0, 1});
while(heap.size()){
//直接取出对头min
auto t = heap.top(); heap.pop();
int ver = t.second, dt = t.first;
//如果已经是确定了最短路的点就跳过
if(st[ver]) continue;
st[ver] = 1;
//开始更新以对头t为起点的边
for(int i = h[ver]; ~i; i = ne[i]){
int j = e[i];
if(dist[j] > dist[ver] + w[i]){
dist[j] = dist[ver] + w[i];
heap.push({dist[j], j});
}
}
}
if (dist[n] == 0x3f3f3f3f) return -1;
return dist[n];
}
signed main()
{
IOS;
cin >> n >> m;
mem(h, -1);
while(m --){
int a, b, c;
cin >> a >> b >> c;
add(a, b, c);
}
cout << Dijkstra() << endl;
return 0;
}