Codeforces Round #408 (Div. 2) D. Police Stations（多起点bfs）

题目链接


思路：要想知道那条路是多余的，我们可以把所有的警察局的点进行bfs，如果该那条边没有被访问过的，就说明它是多余的。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=3e5+1;
int n,m,d,t,vis[maxn],ans[maxn],sum;
vector<pair<int,int>>g[maxn];
queue<int>q;
void bfs()
{
	while(!q.empty())
	{
		int top=q.front();
		q.pop();
		for(auto it:g[top])
		{
			if(!vis[it.first]) ans[it.second]=1,q.push(it.first),vis[it.first]=1;
		}
	}
}
int main()
{
	scanf("%d%d%d",&n,&m,&d);
	for(int i=1;i<=m;++i)
	{
		scanf("%d",&t);
		vis[t]=1;
		q.push(t);
	}
	for(int i=1,u,v;i<n;++i)
	{
		scanf("%d%d",&u,&v);
		g[u].push_back({v,i});
		g[v].push_back({u,i});
	}
	bfs();
	for(int i=1;i<n;++i)
	if(!ans[i]) sum++;
	printf("%d\n",sum);
	for(int i=1;i<n;++i)
	if(!ans[i]) printf("%d ",i);
 }