首页 > 代码库 > UVA1267 - Network

UVA1267 - Network

题意:n台及其连成一个树状网络,其中叶结点是客户端,其他结点是服务器。已经有一台服务器提供VOD服务,要求出最少在几个记得点放置服务器,使得所有客户端到服务器的距离都不超过k。


思路:先将无根树状化为有根树。之后就利用贪心,找到深度最大的结点u,每次放置服务器的地方都要放在u的k级祖先,这样是最划算的。


#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>

using namespace std;

const int MAXN = 1005;
vector<int> gr[MAXN], nodes[MAXN];
int n, s, k, fa[MAXN];
bool vis[MAXN];

void dfs(int u, int f, int d) {
    fa[u] = f;
    int nc = gr[u].size();
    if (nc == 1 && d > k)
        nodes[d].push_back(u);  //利用nodes表可以省去对于叶子结点深度的搜索
    for (int i = 0; i < nc; i++)  {
        int v = gr[u][i]; 
        if (v != f) 
            dfs(v, u, d + 1);
    }
}

void dfs_cover(int u, int f, int d) {
    vis[u] = true;
    int nc = gr[u].size();
    for (int i = 0; i < nc; i++) {
        int v = gr[u][i]; 
        if (v != f && d < k)
            dfs_cover(v, u, d + 1);
    }
}

int solve() {
    int ans = 0;
    memset(vis, 0, sizeof(vis));
    for (int d = n - 1; d > k; d--)
        for (int i = 0; i < nodes[d].size(); i++) {
            int u = nodes[d][i];
            if (vis[u])
                continue;
            int v = u;
            for (int j = 0; j < k; j++)
                v = fa[v];
            dfs_cover(v, -1, 0);
            ans++;
        }
    return ans;
}

int main() {
    int cas;
    scanf("%d", &cas);
    while (cas--) {
        scanf("%d%d%d", &n, &s, &k); 
        for (int i = 1; i <= n; i++) {
            gr[i].clear();
            nodes[i].clear();
        }
        for (int i = 0; i < n - 1; i++) {
            int a, b; 
            scanf("%d%d", &a, &b);
            gr[a].push_back(b);
            gr[b].push_back(a);
        }
        dfs(s, -1, 0);
        int ans = solve();
        printf("%d\n", ans);
    }
    return 0;
}