UVALive - 3902 Network

上一篇博客题目的普遍情况

这里跟前面那道题有点不同的地方是它的半径变得任意可变，并且给了一个初始的节点已经覆盖，并且最终的目的是让所有的叶子都覆盖。

做法跟上道题是类似的。

无根树还是变成有根树好，既然直接给了我们一个特殊点，那就直接拿它做根。

首先先dfs一下，判断哪些点是叶子，顺便统计出dep数组。

既然根已经添加，直接先半径为\(k\)地覆盖一下。把一些初始化就覆盖好的叶子忽略掉。

然后就是一样的弄一个堆，按照深度大小来排序，深度大的叶子先出队。

这个叶子直接取它的\(k\)级父亲，然后就拿它来覆盖就可以了。直到所有叶子都覆盖完。

代码：

#include<cstdio>
#include<cstring>
#include<queue>
const int maxn = 1005;
struct Edges
{
    int next, to;
} e[maxn << 1];
int head[maxn], tot;
int dep[maxn], fa[maxn];
bool vis[maxn];
bool leaf[maxn];
int n, s, k;

struct cmp
{
    bool operator () (const int &a, const int &b) const
    {
        return dep[a] < dep[b];
    }
};
void link(int u, int v)
{
    e[++tot] = (Edges){head[u], v};
    head[u] = tot;
}
void dfs(int u, int f)
{
    dep[u] = dep[f] + 1; fa[u] = f;
    int cnt = 0;
    for(int i = head[u]; i; i = e[i].next)
    {
        cnt++;
        int v = e[i].to;
        if(v == f) continue;
        dfs(v, u);
    }
    if(cnt == 1) leaf[u] = true;
}
void dfs2(int u, int f, int d)
{
    vis[u] = true;
    if(d == k) return;
    for(int i = head[u]; i; i = e[i].next)
    {
        int v = e[i].to;
        if(v == f) continue;
        dfs2(v, u, d + 1);
    }
}
int main()
{
    int T; scanf("%d", &T);
    while(T--)
    {
        memset(head, 0, sizeof head); tot = 0;
        scanf("%d%d%d", &n, &s, &k);
        for(int i = 1; i < n; i++)
        {
            int u, v; scanf("%d%d", &u, &v);
            link(u, v); link(v, u);
        }
        memset(dep, 0, sizeof dep);
        memset(fa, 0, sizeof fa);
        dfs(s, 0);
        memset(vis, false, sizeof vis);
        dfs2(s, 0, 0);
        std::priority_queue<int, std::vector<int>, cmp> heap;
        for(int i = 1; i <= n; i++)
        {
            if(leaf[i] && !vis[i]) heap.push(i);
        }
        int ans = 0;
        while(!heap.empty())
        {
            int x;
            while(!heap.empty() && vis[x = heap.top()]) heap.pop();
            if(heap.empty()) break;
            int v = x;
            for(int i = 1; i <= k; i++) v = fa[v];
            dfs2(v, 0, 0);
            ans++;
        }
        printf("%d\n", ans);
    }
    return 0;
}