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;
}