Codeforces - Police Stations

题目链接：Codeforces - Police Stations

---

我们首先找到必须边，然后问题就解决了。

所以怎么找必须边呢？

每个点肯定是到最近的警察局，又因为两点之间距离都是1，所以我们从警局开始bfs，找到必须边即可。

---

AC代码：

#pragma GCC optimize("-Ofast","-funroll-all-loops")
#include<bits/stdc++.h>
//#define int long long
using namespace std;
const int N=3e5+10,M=N<<1;
int n,k,d,vis[N],p[N],mark[N],cnt;
int head[N],nex[M],to[M],id[M],tot;
inline void add(int a,int b,int c){
	to[++tot]=b; nex[tot]=head[a]; id[tot]=c; head[a]=tot;
}
void bfs(){
	queue<int> q;	for(int i=1;i<=k;i++)	vis[p[i]]=1,q.push(p[i]);
	while(q.size()){
		int u=q.front();	q.pop();
		for(int i=head[u];i;i=nex[i]){
			if(!vis[to[i]])	q.push(to[i]),vis[to[i]]=1,mark[id[i]]=1;
		}
	}
}
signed main(){
	cin>>n>>k>>d;
	for(int i=1;i<=k;i++)	scanf("%d",&p[i]);
	for(int i=1,a,b;i<n;i++)	scanf("%d %d",&a,&b),add(a,b,i),add(b,a,i);
	bfs();
	for(int i=1;i<n;i++)	cnt+=mark[i]==0;
	cout<<cnt<<endl;
	for(int i=1;i<n;i++)	if(!mark[i])	printf("%d ",i);
	return 0;
}