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