Educational Codeforces Round 54 (Rated for Div. 2) D. Edge Deletion（最短路+bfs）（好题）

题目链接


题意：给定n点m条边的有向图，现在要求你必须删边，使得删边之和一个点的最短路不变，这样的点要尽可能地多，同时最多保留k条边，问你删边方案。
思路：很明显，我们先求最短路，然后遍历每条边，对于不是最短路地边直接删，可是这个时候如果边地个数依然大于k怎么办？根据贪心思想，我们尽可以删一个点有多条相同最短路地边，如果每一个都只有唯一一条最短路地话，显然我们删谁都无所谓，可是本人在这个地方有点卡住了，不知道怎么删有多条最短路地点地边，这里学到了大佬们地方法，可以再次bfs遍历一下图，把是最短路上地边保存下去，由于每个点只会bfs一次（bfs可以自动略过大于等于最短路的其他路径），这样就我们保留地边是最少的了。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll; 
const ll inf=1e18;
const int maxn=3e5+1; 
struct cxk{
	int v;
	ll d;
	int id;
};
vector<cxk>g[maxn]; 
vector<int>ans;
ll cnt,w,d[maxn];
bool vis[maxn]={false};
void spfa(int x)
{
	memset(vis,false,sizeof(vis));
	fill(d,d+maxn,inf);
	d[x]=0;
	queue<int>q;
	q.push(x);
	vis[x]=true;
	while(!q.empty())
	{
		int top=q.front();
		q.pop();
		vis[top]=false;
		for(auto to:g[top])
		{
			if(d[top]+to.d<d[to.v])
			{
				d[to.v]=d[top]+to.d;
				if(!vis[to.v]) q.push(to.v),vis[to.v]=true;
			}
		}
	}
}
void bfs(int x)
{
	memset(vis,false,sizeof(vis));
	vis[x]=true;
	queue<int>q;
	q.push(x);
	while(!q.empty())
	{
		int top=q.front();
		q.pop();
		for(auto to:g[top])
		{
			if(!vis[to.v]&&d[top]+to.d==d[to.v])
			{
				ans.push_back(to.id);
				q.push(to.v);
				vis[to.v]=true;
			}
		}
	}
}
int main()
{
	int n,m,k,cnt;
	scanf("%d%d%d",&n,&m,&k);
	for(int i=1,u,v;i<=m;++i)
	{
		scanf("%d%d%lld",&u,&v,&w);
		g[u].push_back({v,w,i});
		g[v].push_back({u,w,i});
	}
	spfa(1);
	bfs(1);
	int size=ans.size();
	if(k<size) size=k;
	printf("%d\n",size);
	for(int i=0;i<size;++i) printf("%d ",ans[i]);
}