NetWork(LA 3902)深度优先搜索

来自《算法竞赛入门经典训练指南》

1.题目原文

https://icpcarchive.ecs.baylor.edu/index.php?option=com_onlinejudge&Itemid=8&page=show_problem&problem=1903

2.解题思路

通常来说，把无根树转化成有根树会有助于解题。更何况，本题已经有一个天然的根，原始服务器。

首先对于那些已经满足条件的(到原始服务器距离不超过k)的客户端，直接忽略即可。

其次，考虑深度最大的结点。对于样例，比如结点8，应该在哪里放置一个服务器来覆盖它呢？(覆盖一个叶子值到该叶子的距离不超过k)只有点4和点5符合条件，而结点5能覆盖的叶子都能被结点4覆盖，因此选4会更好。一般来说，对于深度最大的结点u，选择u的k级祖先会更划算(父亲是1级祖先，父亲的父亲是2级祖先，……).

下面给出上述算法的一种实现方法:

每放置一个新服务器，进行一次DFS，覆盖与它距离不超过k的所有结点。本题只需要考虑深度大于k的叶子结点即可。为了让程序更简单，我们用nodes表避开对“按深度排序”的操作。

3.AC代码

#include <algorithm>
#include <cctype>
#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iomanip>
#include <iostream>
#include <map>
#include <queue>
#include <string>
#include <set>
#include <vector>
#include<cmath>
#include<bitset>
#include<sstream>
#include<stack>
using namespace std;
#define INF 0x7fffffff
typedef long long ll;
typedef pair<int,int> P;
const int maxn=1000+10;
int n,s,k;
vector<int> gr[maxn],nodes[maxn];
int fa[maxn];
bool coverd[maxn];

void dfs(int u,int f,int d)
{
    fa[u]=f;
    int nc=gr[u].size();
    //nc==1表示u是叶子结点
    if(nc==1&&d>k) nodes[d].push_back(u);
    for(int i=0;i<nc;i++){
        int v=gr[u][i];
        if(v!=f) dfs(v,u,d+1);
    }
}

void dfs2(int u,int f,int d)
{
    coverd[u]=true;
    int nc=gr[u].size();
    for(int i=0;i<nc;i++){
        int v=gr[u][i];
        if(v!=f&&d<k) dfs2(v,u,d+1);//只覆盖到新服务器距离不超过k的结点
    }
}

void solve()
{
    int ans=0;
    memset(coverd,0,sizeof(coverd));
    for(int d=n-1;d>k;d--){
        for(int i=0;i<nodes[d].size();i++){
            int u=nodes[d][i];
            if(coverd[u]) continue;//不考虑已覆盖的结点

            int v=u;
            for(int j=0;j<k;j++) v=fa[v];//v是u的k级祖先
            dfs2(v,-1,0);//在结点v出放服务器
            ans++;
        }
    }
    printf("%d\n",ans);
}

int main()
{
    int t;
    scanf("%d",&t);
    while(t--){
        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);
        solve();
    }
    return 0;
}