Codeforces Round #408 D. Police Stations

题意

有 n 个城市 n-1 条城市间连通的双向路（每条路的长度均为 1 ），有 k 个警察局坐落在 $p_i$ 城市，每个警察局最远可以出警到达距离为 d 的城市。问最多可以删除多少条路，仍使得每个城市均可由至少一个警察局出警到达。

解题思路

对于 $k_{sub}$ 个坐落于不同城市的警察局，可以将弱连通的 n 个城市 $k_{sub}-1$ 个独立集。

利用 BFS 以 $k_{sub}$ 个警局为起点不断向外围扩展，直到两个警局的出警范围相冲突的两城市见的连通路即可视作是可删除的路。

代码

#include<bits/stdc++.h>
using namespace std;
const int V = 300010;
int n, k, d, p[V];
vector<int> shut;
vector< pair<int, int> > g[V];
bool vis[V];
void bfs()
{
    queue< pair<int, int> > que;
    sort(p+1, p+k+1);
    for(int i=1;i<=k;i++)
        if(p[i] == p[i-1])  continue;
        else
            que.push(make_pair(p[i], -1)), vis[ p[i] ] = 1;
    while(!que.empty())
    {
        int q = que.front().first;
        int pre = que.front().second;
        que.pop();
        for(int i=0, to;i<g[q].size();i++)
        {
            to = g[q][i].first;
            if(to == pre)   continue;
            if(vis[to]) {
                shut.push_back(g[q][i].second);
                continue;
            }
            vis[to] = 1;
            que.push(make_pair(to, q));
        }
    }
}
int main()
{
    scanf("%d %d %d",&n,&k,&d);
    for(int i=1, police;i<=k;i++)
        scanf("%d",&p[i]);
    for(int i=1, u, v;i<n;i++)
    {
        scanf("%d %d",&u,&v);
        g[u].push_back(make_pair(v, i));
        g[v].push_back(make_pair(u, i));
    }
    bfs();
    printf("%d\n", shut.size()/2);
    sort(shut.begin(), shut.end());
    for(int i=0;i<shut.size();i+=2)
        printf("%d ",shut[i]);
}