CF 1076D Edge Deletion （最短路）

題目傳送

構建出一個最短路樹，與1聯通的每條邊對結果都是1，所以在求最短路的時候，就可以按順序把邊加入結果中。

有幾個要注意的點：
1 存圖的時候一條邊存入了兩次，但要求輸出輸入時的編號，所以輸出時邊的序號要除2。如果邊從1開始編號的話，1,2表示同一條邊，但除2後就不同了，從2開始編號就可以避免。
2 優先隊列默認按pair的first升序排列，但我們需要降序，可以直接存入-dis[v],可以少寫一個重載。（主要還是懶）

詳見代碼：

#include <iostream>
#include <queue>
#include <vector>
#include <algorithm>
#include <cstring>
using namespace std;
#define ll long long
#define pli pair<ll,int>
#define inf (ll)1e15
#define maxn (int)3e5+5

struct edgs {
	int from, to, val, nxt;
	edgs() {}
	edgs(int from, int to, int val, int nxt) :from(from), to(to), val(val), nxt(nxt) {}
}edg[maxn << 1];

int n, m, k, cnt = 1;
int head[maxn], vis[maxn], par[maxn];
ll dis[maxn];
vector<int> ans;

//鏈式前向星存圖，注意cnt要從2開始，因爲後面存答案的時候要除2
//2,3表示的是一條邊，除2後都等於1
//如果從1開始的話，1,2是同一條邊，但除2後就不同了
void add_edg(int from, int to, int val)
{
	edg[++cnt].from = from;
	edg[cnt].to = to;
	edg[cnt].val = val;
	edg[cnt].nxt = head[from];
	head[from] = cnt;
}

//dij模板
void Dij()
{
	for (int i = 1; i <= n; i++) dis[i] = inf;
	dis[1] = 0;
	priority_queue<pli> q;
	q.push(make_pair(0, 1));
	while (q.size() && ans.size() < k) {
		int u = q.top().second;
		q.pop();
		if (vis[u]) continue;
		vis[u] = 1;
		//u^1==1 相當於u!=1
		if (u ^ 1) ans.push_back(par[u] >> 1);
		for (int i = head[u]; i; i = edg[i].nxt) {
			int v = edg[i].to;
			int dist = edg[i].val;
			if (dis[v] > dis[u] + dist) {
				dis[v] = dis[u] + dist;
				par[v] = i;
				//優先隊列默認按pair的first升序排列，但我們需要降序，所以存入-dis[v]，可以少寫一個重載
				q.push(make_pair(-dis[v], v));
			}
		}
	}
}

int main(void)
{
	cin >> n >> m >> k;
	for (int i = 1; i <= m; i++) {
		int a, b, c;
		cin >> a >> b >> c;
		add_edg(a, b, c);
		add_edg(b, a, c);
	}
	Dij();
	cout << ans.size() << endl;
	for (int i = 0; i < ans.size(); i++) cout << ans[i] << " ";
	return 0;
}