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 + ;
 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 = ;
     scanf("%d%d%d", &n, &k, &d);
     for(int i = ; i < k; i++){
         scanf("%d", &x);
         q.push(x);
         vis[x] = ;
     }
     for(int i = ; i <= n - ; 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 = ; i < vt[p].size(); i++){
             if(!vis[vt[p][i]]){
                 vis[vt[p][i]] = ;
                 q.push(vt[p][i]);
                 tag[mp[{p, vt[p][i]}]] = ;
                 ans++;
             }
         }
     }
     cout << n - ans -  << endl;
     for(int i = ; i <= n - ; i++){
         if(!tag[i]) cout << i << " ";
     }
     cout << endl;
     return ;
 }