codeforces 1076D Edge Deletion

题目：
传送门

题意：

给出n个点，m条边，要求图中最多有k条边的前提下，生成尽可能多的good点。good点：删除一些边和没删除前从1到good点的最短路径是一样的。

思路：

先求出各个点的最短路径，记录下每个点的前驱，然后走一遍bfs，将符合条件的边的序号加入vector中。

代码如下：
 

#include 
#include 
#include 
#include 
#include 
#include 
using namespace std;
const int maxn=3*1e5+5;
typedef long long ll;
const ll INF=1e15;
int n,m,k;
int head[maxn];
ll d[maxn];
int pre[maxn];
vectorans;
struct Node
{
    int v;
    ll dis;
    bool operator < (const Node x) const
    {
        return  dis > x.dis;
    }
};
struct edge
{
    int to;
    ll w;
    int next;
}edge[maxn<<1];
void addedge (int u,int v,ll w,int id)
{
    edge[id].to=v;
    edge[id].w=w;
    edge[id].next=head[u];
    head[u]=id;
}
void djst ()
{
    int vis[maxn];
    for (int i=1;i<=n;i++)
    {
        vis[i]=0;
        d[i]=INF;
    }
    d[1]=0;
    priority_queueq;
    Node t;
    t.v=1; t.dis=0;
    q.push(t);
    while (!q.empty())
    {
        Node t=q.top();
        q.pop();
        int u=t.v; ll dis=t.dis;
        if(vis[u])
            continue;
        vis[u]=1;
        for (int i=head[u];i!=-1;i=edge[i].next)
        {
            int v=edge[i].to;
            if(!vis[v]&&d[u]+edge[i].wq;
    q.push(1);
    while (!q.empty())
    {
        int t=q.front();
        q.pop();
        for (int i=head[t];i!=-1;i=edge[i].next)
        {
            int v=edge[i].to;
            if(pre[v]==t&&ans.size()=k)
            break;
    }
}
int main()
{
    scanf("%d%d%d",&n,&m,&k);
    int id=0;
    memset (pre,-1,sizeof(pre));
    memset (head,-1,sizeof(head));
    for (int i=0;i