codeforces 796D Police Stations

题目链接：http://codeforces.com/contest/796/problem/D

题意：有n个城市，坐标1到n，n-1条边，长度都为1。然后有k个警局，下标pi。现在要求拆掉其中的一部分边，并且还要保证所有城市到警局距离小于等于d。一开始保证所有城市到金具距离小于等于d。输出删除的路径总数，以及删除的路径的id。

分析：将k个警局求入一个queue，然后bfs，vis数组标记是否到达某一个城市，初始话为0，如果是第一次到达（vis[i]=0），那么这条路是必须的，vis[i]=1,找到所有必须路径，剩下的就是可以删除的，完全不用考虑d。

AC代码：

#include<bits/stdc++.h>
using namespace std;

struct Node{
    int v;
    int id;
    Node(int _v,int _id){
        v=_v;
        id=_id;
    }
};
int a[300005];
vector<Node>g[300005];
int vis[300005];
int vis2[300005];
queue<int >q;
int c[300005];
int length[300005];
int main() {
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    int n,k,d;
    int u,v;
    memset(vis,0,sizeof(vis));
    memset(vis2,0,sizeof(vis2));
    cin>>n>>k>>d;
    for(int i=1;i<=k;i++){
        cin>>a[i];
        vis[a[i]]=1;
        q.push(a[i]);
    }
    sort(a+1,a+1+k);
    for(int i=1;i<n;i++){
        cin>>u>>v;
        g[u].push_back(Node(v,i));
        g[v].push_back(Node(u,i));
    }
    int result=0;
    while(!q.empty()){
        int x=q.front();
        int sz=g[x].size();
        q.pop();
        for(int i=0;i<sz;i++){
            if(vis[g[x][i].v]==0){
                vis[g[x][i].v]=1;
                vis2[g[x][i].id]=1;
                q.push(g[x][i].v);
            }
        }
    }
    for(int i=1;i<n;i++){
        if(vis2[i]==0){
            result++;
            c[result]=i;
        }
    }
    cout<<result<<endl;
    for(int i=1;i<=result;i++){
        cout<<c[i]<<" ";
    }
    cout<<endl;
return 0;
}