题目链接:
http://codeforces.com/problemset/problem/796/D
题意:
有一棵树,n个结点,编号1到n。有m个特殊点,保证树上每个点到其中一个特殊点的距离小于等于d. 问最多能删除多少条边,删完之后每个点到其中一个特殊点的距离仍然小于等于d。
输出最多能删除边的条数,然后输出这几条边的编号。
题解:
首先把m个特殊点都放到队列里,然后尽心bfs,如果当前结点第一次走到,那么这条边就是必须走,如果这个点之前已经走过,那么这条边就可以删除。 相当于是只保留了每个点到特殊点的最短路径。
代码:
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=6e5+5;
int head[maxn],nex[maxn],to[maxn],cnt;
int n,m,d;
bool flag[maxn]; //标记i是否走过,true表示走过
bool mov[maxn]; //标记这条边是否必须走,值为true表示必须走,false表示这条边可以删去
vector<int> ans;
queue<int> q;
void add(int i,int j){
cnt++;
to[cnt]=j;nex[cnt]=head[i];head[i]=cnt;
}
void bfs(){
while(q.size()){
int x=q.front();q.pop();
for(int i=head[x];i;i=nex[i]){
int y=to[i];
if(!flag[y]){
flag[y]=true;
mov[i]=mov[i^1]=true;
q.push(y);
}
}
}
}
int main(){
scanf("%d%d%d",&n,&m,&d);
for(int i=1;i<=m;i++){
int x;
scanf("%d",&x);
flag[x]=true;
}
for(int i=1;i<=n;i++){
if(flag[i]) q.push(i);
}
cnt=1;
for(int i=1;i<n;i++){
int a,b;
scanf("%d%d",&a,&b);
add(a,b);
add(b,a);
}
bfs();
for(int i=2;i<=(n-1)*2;i+=2){
if(!mov[i]){
ans.push_back(i/2);
}
}
printf("%d\n",ans.size());
for(int i=0;i<ans.size();i++) printf("%d ",ans[i]);
return 0;
}