思路:
套dijkstra模板,记录每个点是由哪条边(设为X)更新过来的,放在pre数组里,因为是无向图,由idx的含义可知 x = i / 2 + 1,注意要用long long ……
C++ 代码
#include <iostream>
#include <queue>
#include <cstring>
using namespace std;
typedef long long ll;
typedef pair<ll, int> PII;
const int N = 2e5 + 10;
// 对于每个点k,开一个单链表,存储k所有可以走到的点。h[k]存储这个单链表的头结点
int h[N], e[N], ne[N], w[N], idx;
int a, b, c;
// 添加一条边a->b
void add(int a, int b, int c)
{
w[idx] = c, e[idx] = b, ne[idx] = h[a], h[a] = idx ++ ;
}
int n, m, k; // 点的数量
long long dist[N]; // 存储所有点到1号点的距离
bool st[N]; // 存储每个点的最短距离是否已确定
int pre[N];
vector<int> res;
// 求1号点到n号点的最短距离,如果不存在,则返回-1
void dijkstra()
{
memset(dist, 0x3f, sizeof dist);
dist[1] = 0;
priority_queue<PII, vector<PII>, greater<PII>> heap;
heap.push({0, 1}); // first存储距离,second存储节点编号
int cnt = 0;
while (heap.size())
{
auto t = heap.top();
heap.pop();
int ver = t.second;
ll distance = t.first;
if (st[ver]) continue;
if (pre[ver] != -1 && cnt < k) {
res.push_back(pre[ver]);
cnt++;
}
st[ver] = true;
for (int i = h[ver]; i != -1; i = ne[i])
{
int j = e[i];
if (dist[j] > distance + w[i])
{
pre[j] = i / 2 + 1;
dist[j] = distance + w[i];
heap.push({dist[j], j});
}
}
}
}
int main () {
// 初始化
idx = 0;
memset(pre, -1, sizeof pre);
memset(h, -1, sizeof h);
scanf("%d%d%d", &n, &m, &k);
for (int i = 0; i < m; i ++) {
scanf("%d%d%d", &a, &b, &c);
add(a, b, c);
add(b, a, c);
}
dijkstra();
printf("%d\n", res.size());
for(auto x : res) {
printf("%d ", x);
}
return 0;
}