Codeforces - Police Stations

最新推荐文章于 2020-05-13 22:19:44 发布

原创最新推荐文章于 2020-05-13 22:19:44 发布 · 248 阅读

0 ·

CC 4.0 BY-SA版权

图论同时被 3 个专栏收录

421 篇文章

订阅专栏

Codeforces

373 篇文章

订阅专栏

bfs

21 篇文章

订阅专栏

本文详细解析了Codeforces平台上的PoliceStations题目，介绍了如何通过广度优先搜索（BFS）找到必经边，从而解决警察局位置优化问题。代码使用C++实现，展示了完整的算法流程和数据结构设计。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目链接：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;
}