796D(bfs)

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

题意: 给出一颗 n 个节点树, 树枝连接的两个定点距离为 1, 树中有 k 个特殊点, 问最多可以删除哪些树枝, 使得树中其他顶点到特殊点的最小距离不大于 d.

注意: 题目说明了一定有解.

思路: bfs

可以直接 bfs 求其他点到特殊点的最短距离, 因为题目说明给出的数据都是有解的, 即所有顶点到特殊点的距离都是不大于 d 的. 那么搜完所有顶点后剩余的边就是不必要的.  即答案中的可删除边.

代码:

#include <iostream>
#include <stdio.h>
#include <vector>
#include <queue>
#include <map>
using namespace std;

const int MAXN = 3e5 + 10;
int vis[MAXN], tag[MAXN];
map<pair<int, int>, int> mp;
vector<int> vt[MAXN];
queue<int> q;

int main(void){
    int n, k, d, x, y, ans = 0;
    scanf("%d%d%d", &n, &k, &d);
    for(int i = 0; i < k; i++){
        scanf("%d", &x);
        q.push(x);
        vis[x] = 1;
    }
    for(int i = 1; i <= n - 1; i++){
        scanf("%d%d", &x, &y);
        vt[x].push_back(y);
        vt[y].push_back(x);
        mp[{x, y}] = mp[{y, x}] = i;
    }
    while(!q.empty()){
        int p = q.front();
        q.pop();
        for(int i = 0; i < vt[p].size(); i++){
            if(!vis[vt[p][i]]){
                vis[vt[p][i]] = 1;
                q.push(vt[p][i]);
                tag[mp[{p, vt[p][i]}]] = 1;
                ans++;
            }
        }
    }
    cout << n - ans - 1 << endl;
    for(int i = 1; i <= n - 1; i++){
        if(!tag[i]) cout << i << " ";
    }
    cout << endl;
    return 0;
}