标签：Building cnt include int gym dfs num antennas now

Building 5G antennas

dfs 剪枝

要字典序最小，显然第一个点就是 \(1\)，后面考虑走 \(k\) 步后能到达的点集中选一个字典序最小的，重复该过程

考虑 \(set[i][j]\) 表示第 \(i\) 号点当前能走 \(j\) 步所包含的点的集合，我们可以发现对于相同的点，如果 \(j_1 > j_2\)，显然有 \(set[i][j_1]\) 包含 \(set[i][j_2]\) 的情况

因此之前如果访问到一个点，且步数比当前搜到这个点的步数还大，则说明当前走下去想要访问的点之前已经被访问过了，直接剪枝

这样每个点最多被访问 \(k\) 次

时间复杂度是 \(O(nk)\)

#include <iostream>
#include <cstdio>
#include <vector>
#include <algorithm>
#include <queue>
#include <functional>
using namespace std;
const int maxn = 1e5 + 10;
vector<int>gra[maxn];
priority_queue<int, vector<int>, greater<int> >q;
int num[maxn];

void dfs(int now, int pre, int cnt)
{
    if(cnt == -1) return;
    if(num[now] >= cnt) return;
    if(num[now] == -1) q.push(now);
    num[now] = cnt;
    for(int nex : gra[now])
    {
        if(nex == pre) continue;
        dfs(nex, now, cnt - 1);
    }
}

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    int n, k;
    cin >> n >> k;
    for(int i=1; i<n; i++)
    {
        int a, b;
        cin >> a >> b;
        gra[a].push_back(b);
        gra[b].push_back(a);
    }
    for(int i=0; i<=n; i++) num[i] = -1;
    q.push(1);
    vector<int>ans;
    while(q.size())
    {
        int now = q.top();
        q.pop();
        ans.push_back(now);
        num[now] = 0;
        dfs(now, now, k);
    }
    for(int i=0; i<ans.size(); i++)
    {
        if(i) cout << " ";
        cout << ans[i];
    }
    cout << endl;
    return 0;
}

标签：Building,cnt,include,int,gym,dfs,num,antennas,now
来源： https://www.cnblogs.com/dgsvygd/p/16633445.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。