47. 参加科学大会
思路
使用Dijkstra 算法 计算从起点(节点 1)到终点(节点 n)的最短路径。用优先队列(小顶堆) 维护当前未访问的最短路径节点,每次选择距离最短的未访问节点进行更新,直到遍历所有可能路径。如果终点不可达,输出 -1,否则输出最短路径长度。
代码
#include <iostream>
#include <vector>
#include <list>
#include <queue>
#include <climits>
using namespace std;
class mycomparison {
public:
bool operator()(const pair<int, int>& lhs, const pair<int, int>& rhs) {
return lhs.second > rhs.second;
}
};
struct Edge {
int to;
int val;
Edge(int t, int w): to(t), val(w) {}
};
int main() {
int n, m, p1, p2, val;
cin >> n >> m;
vector<list<Edge>> grid(n + 1);
for(int i = 0; i < m; i++){
cin >> p1 >> p2 >> val;
grid[p1].push_back(Edge(p2, val));
}
int start = 1;
int end = n;
vector<int> minDist(n + 1, INT_MAX);
vector<bool> visited(n + 1, false);
priority_queue<pair<int, int>, vector<pair<int, int>>, mycomparison> pq;
pq.push(pair<int, int>(start, 0));
minDist[start] = 0;
while (!pq.empty()) {
pair<int, int> cur = pq.top(); pq.pop();
if (visited[cur.first]) continue;
visited[cur.first] = true;
for (Edge edge : grid[cur.first]) {
if (!visited[edge.to] && minDist[cur.first] + edge.val < minDist[edge.to]) {
minDist[edge.to] = minDist[cur.first] + edge.val;
pq.push(pair<int, int>(edge.to, minDist[edge.to]));
}
}
}
if (minDist[end] == INT_MAX) cout << -1 << endl;
else cout << minDist[end] << endl;
}