LA 3902 Network 网络

最新推荐文章于 2022-06-01 22:03:59 发布

FirstLucker

最新推荐文章于 2022-06-01 22:03:59 发布

阅读量633

点赞数

分类专栏：树的应用

本文链接：https://blog.csdn.net/FirstLucker/article/details/46752915

版权

树的应用专栏收录该内容

1 篇文章 0 订阅

订阅专栏

题意：n台机器连成一个树状网络，其中叶节点是客户端，其他结点是服务器。目前有一台服务器正在提供VOD服务。现在要求在一些其他服务器上也安装VOD服务，使得每台客户端到最近服务器的距离不超过k。求最少还需安装VOD服务的服务器台数。

将无根树转化为有树，题目给出了一个已安装服务的服务器，那么就选它作为根来构造树，这样深度不超过k的结点都不用考虑（已经被根结点所代表的服务器覆盖到了）。我们首先满足深度最大的叶节点，我们选取它的 k 级父节点作为要新增服务的服务器，这样一定是最优的。然后增加一个服务器后，就对这台服务器进行DFS，覆盖与这台服务器距离不超过 k 的所有叶节点，然后再找下一个深度最大且未被覆盖的叶节点进行增加服务器，直到所有叶节点均被覆盖为止。

#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstring>
#include <string>
#include <algorithm>
#include <stack>
#include <queue>
#include <vector>
#include <map>
#include <set>
using namespace std;

const int MAX = 1005;
int n, s, k;
vector <int> node[MAX], deep[MAX]; //相连节点数组和深度结点数组
int father[MAX]; //父节点数组
bool flag[MAX]; //标记是否已覆盖

void DFS_Father(int x, int pre, int d) //计算父节点
{
    father[x] = pre;
    if(node[x].size() == 1) //叶节点
        deep[d].push_back(x);
    for(int i = 0; i < node[x].size(); i++)
    {
        if(node[x][i] != pre)
            DFS_Father(node[x][i], x, d + 1);
    }
}

void DFS_Cover(int x, int pre, int d) //覆盖与当前所选服务器距离不超过k的叶节点
{
    flag[x] = true;
    if(d == k)
        return;
    for(int i = 0; i < node[x].size(); i++)
    {
        if(node[x][i] != pre)
            DFS_Cover(node[x][i], x, d + 1);
    }
}

void input()
{
    scanf("%d%d%d", &n, &s, &k);
    for(int i = 0; i < MAX; i++)
        node[i].clear();
    int x, y;
    for(int i = 1; i < n; i++)
    {
        scanf("%d%d", &x, &y);
        node[x].push_back(y);
        node[y].push_back(x);
    }
}

void solve()
{
    int ans = 0;
    for(int i = 0; i < MAX; i++)
        deep[i].clear();
    DFS_Father(s, 0, 0);
    memset(flag, false, sizeof(flag));
    for(int d = n - 1; d > k; d--) //只需考虑深度大于k的结点
    {
        for(int i = 0; i < deep[d].size(); i++)
        {
            int temp = deep[d][i];
            if(flag[temp]) //已覆盖的结点不用考虑
                continue;
            for(int j = 0; j < k; j++) //寻找k级祖先
                temp = father[temp];
            DFS_Cover(temp, 0, 0);
            ans++;
        }
    }
    printf("%d\n", ans);
}

int main()
{
    int T;
    scanf("%d", &T);
    while(T--)
    {
        input();
        solve();
    }
    return 0;
}