796D(bfs)
Posted ygeloutingyu
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了796D(bfs)相关的知识,希望对你有一定的参考价值。
题目链接: http://codeforces.com/problemset/problem/796/D
题意: 给出一颗 n 个节点树, 树枝连接的两个定点距离为 1, 树中有 k 个特殊点, 问最多可以删除哪些树枝, 使得树中其他顶点到特殊点的最小距离不大于 d.
注意: 题目说明了一定有解.
思路: bfs
可以直接 bfs 求其他点到特殊点的最短距离, 因为题目说明给出的数据都是有解的, 即所有顶点到特殊点的距离都是不大于 d 的. 那么搜完所有顶点后剩余的边就是不必要的. 即答案中的可删除边.
代码:
1 #include <iostream> 2 #include <stdio.h> 3 #include <vector> 4 #include <queue> 5 #include <map> 6 using namespace std; 7 8 const int MAXN = 3e5 + 10; 9 int vis[MAXN], tag[MAXN]; 10 map<pair<int, int>, int> mp; 11 vector<int> vt[MAXN]; 12 queue<int> q; 13 14 int main(void){ 15 int n, k, d, x, y, ans = 0; 16 scanf("%d%d%d", &n, &k, &d); 17 for(int i = 0; i < k; i++){ 18 scanf("%d", &x); 19 q.push(x); 20 vis[x] = 1; 21 } 22 for(int i = 1; i <= n - 1; i++){ 23 scanf("%d%d", &x, &y); 24 vt[x].push_back(y); 25 vt[y].push_back(x); 26 mp[{x, y}] = mp[{y, x}] = i; 27 } 28 while(!q.empty()){ 29 int p = q.front(); 30 q.pop(); 31 for(int i = 0; i < vt[p].size(); i++){ 32 if(!vis[vt[p][i]]){ 33 vis[vt[p][i]] = 1; 34 q.push(vt[p][i]); 35 tag[mp[{p, vt[p][i]}]] = 1; 36 ans++; 37 } 38 } 39 } 40 cout << n - ans - 1 << endl; 41 for(int i = 1; i <= n - 1; i++){ 42 if(!tag[i]) cout << i << " "; 43 } 44 cout << endl; 45 return 0; 46 }
以上是关于796D(bfs)的主要内容,如果未能解决你的问题,请参考以下文章
HDU3247 Resource Archiver(AC自动机+BFS+DP)