codeforces 796D bfs

题目链接：

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;
}