题目链接
思路:要想知道那条路是多余的,我们可以把所有的警察局的点进行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);
}