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