1076D Edge Deletion 【最短路】

题目：戳这里

题意：求出1到所有点的最短路径后，把边减到小于等于k条，问保留哪些边可以使仍存在的最短路径最多。

解题思路：这题就是考求最短路的原理。比如dijkstra，用优先队列优化后存在队列中的前k条边就是答案。因为可以优先队列维护最小值，因此放进队列中的边一定是最短的。借这道题又复习了下dijkstra。

附本人代码：

#include <cstdio>
#include <cstring>
#include <queue>
#include <vector>
typedef long long ll;
const int Max = 3e5+10;
const ll inf = 1e18;
using namespace std;
struct edge{int to,id; ll cost;};
vector<edge>G[Max];
ll d[Max];
struct nod {
    ll x;
    int y, id;
    nod() {}
    nod(ll xx, int yy, int Id) {
        x = xx,y = yy, id = Id;
    }
    bool friend operator < (nod a, nod b) {
        return a.x > b.x;
    }
};
int n,m, k;
int anslen = 0;
int ans[Max];
void dijkstra(){
    priority_queue<nod >que;
    d[0]=0;
    que.push(nod(0,0,0));
    while(!que.empty()){
        nod p=que.top();que.pop();
        int v=p.y;
        if(d[v]<p.x) continue;
        if(anslen == k) break;
        if(p.id != 0)
        ans[++anslen] = p.id;
        for(int i=0;i<G[v].size();i++){
            edge e=G[v][i];
            if(d[e.to]>d[v]+e.cost){
                d[e.to]=d[v]+e.cost;
                que.push(nod(d[e.to],e.to,e.id));
            }
        }
    }
}
int main(){

    scanf("%d %d %d",&n,&m, &k);
        fill(d,d +Max,inf);
        int a,b;
        ll c;
        for(int i=0;i<m;i++){
            scanf("%d %d %lld",&a,&b,&c);
            a-=1;
            b-=1;
            edge e;
            e.to=b;e.cost=c;
            e.id = i+1;
            G[a].push_back(e);
            e.to=a;
            G[b].push_back(e);
        }
        dijkstra();
        printf("%d\n", anslen);
        for(int i = 1; i <= anslen; ++i) {
            printf("%d ", ans[i]);
        }

    return 0;
}