CF1076D Edge Deletion

洛谷传送门
cf传送门
这道题作为div.2的D题，被我一眼秒了我觉得十分荣幸，然后就开始写，然后就写了好久。
AC之后看网上的题解，发现好多最短路树的，猛然发现我写的好复杂啊，结果还看到了直接一遍dijkstra+贪心的标算。。。
只有我建的最短路图，再慢慢删边的吗。。。
说白了也是贪心啦，
策略十分显然：
1、如果有多条最短路到一个点，显然可以删到只剩一条，这也是最先需要做的（最短路树的不用管这个）
2、然后就只剩下一颗树了，topsort即可
代码：

#include<cstdio>
#include<queue>
#include<cstring>
#include<vector>
#include<algorithm>
using namespace std;
#define rg register
vector<int>s[300001];
long long dis[300001];int n,m,k,cnt,sum,x[300001],used[300001],y[300001],in[300001],out[300001],z[300001],pre[600001],nxt[600001],h[300001],v[600001],op[600001];
struct oo{int id;long long dis;};bool vis[300001];
bool operator<(oo a,oo b){return a.dis>b.dis;}
priority_queue<oo>q;
void add(int x,int y,int z,int now)
{
    pre[++cnt]=y,nxt[cnt]=h[x];h[x]=cnt,v[cnt]=z;op[cnt]=now;
    pre[++cnt]=x,nxt[cnt]=h[y];h[y]=cnt,v[cnt]=z;op[cnt]=now;
}
void ins(int x,int y,int z,int now){pre[++cnt]=y,nxt[cnt]=h[x];h[x]=cnt,v[cnt]=z;op[cnt]=now;}
void dijkstra()
{
    memset(dis,63,sizeof dis);
    q.push((oo){1,dis[1]=0});
    while(!q.empty())
    {
        int x=q.top().id;q.pop();
        if(vis[x])continue;vis[x]=1;
        for(int i=h[x];i;i=nxt[i])
            if(dis[pre[i]]>dis[x]+v[i])dis[pre[i]]=dis[x]+v[i],q.push((oo){pre[i],dis[pre[i]]});
    }
}
bool cmp(vector<int>a,vector<int>b){return a.size()>b.size();}
void topsort()
{
    queue<int>q;
    for(int i=1;i<=n;i++)if(!in[i])q.push(i);
    while(!q.empty())
    {
        int x=q.front();q.pop();
        for(int i=h[x];i;i=nxt[i])
        {
            used[op[i]]=0,sum--;if(sum<=k)return ;
            if(!(--in[pre[i]]))q.push(pre[i]);
        }
    }
}
int main()
{
    scanf("%d%d%d",&n,&m,&k);
    for(rg int i=1;i<=m;i++)scanf("%d%d%d",&x[i],&y[i],&z[i]),add(x[i],y[i],z[i],i);
    dijkstra();
    for(int i=1;i<=n;i++)
        for(int j=h[i];j;j=nxt[j])
            if(dis[pre[j]]==dis[i]+v[j])used[op[j]]=i;
    memset(h,0,sizeof h),cnt=0;
    for(int i=1;i<=m;i++)
        if(used[i])
        {
            sum++;
            if(used[i]==x[i])s[y[i]].push_back(i),in[y[i]]++,out[x[i]]++;
            else s[x[i]].push_back(i),in[x[i]]++,out[y[i]]++;
        }
    if(sum<=k)
    {
        printf("%d\n",sum);
        for(int i=1;i<=m;i++)if(used[i])printf("%d ",i);
        return 0;
    }
    int now=1;
    while(now<=n)
    {
        if(s[now].size()>1)
        {
            int w=s[now].size(),t;
            if(w-1>=sum-k)t=sum-k;
            else t=w-1;sum-=t;
            for(int i=0;i<t;i++)used[s[now][i]]=0;
            if(sum<=k)break;
        }
        if(sum<=k)break;
        now++;
    }
    if(sum<=k)
    {
        printf("%d\n",sum);
        for(int i=1;i<=m;i++)if(used[i])printf("%d ",i);
        return 0;
    }
    memset(in,0,sizeof in),sum=cnt=0,memset(h,0,sizeof h);
    for(int i=1;i<=m;i++)
        if(used[i])
        {
            sum++;
            if(used[i]==x[i])ins(y[i],x[i],z[i],i),in[x[i]]++;
            else ins(x[i],y[i],z[i],i),in[y[i]]++;
        }
    topsort();
    printf("%d\n",sum);
    for(int i=1;i<=m;i++)if(used[i])printf("%d ",i);
}

转载于:https://www.cnblogs.com/lcxer/p/10028222.html