LA 3702 network(greedy + dfs)

这个题也是好久以前看过的。就是用贪心的思想去dfs

题目要求是 给出一棵树 然后给出vod服务器的标号 和这种服务所能到达的层数k。

所有叶子节点都是客户端 非叶节点是服务器 现在要选出尽量少的服务器，使得所有客户端享受到vos服务。o

把无根数转换为有根数有利于解题。。。现在有一个现成的根 就是最原始的vod服务器编号

然后 读入的时候 把每个节点相邻的节点保存到vector 如果size是1  说明这个点是叶节点

首先第一次dfs 这次的目的主要是找出每个节点的fa节点 然后保存到fa数组里

然后就是把叶节点按照层次 放入node数组里 因为后面贪心的过程 dfs要从最深层开始

这样第一次dfs完成后 fa数组和node数组就搞定了

开始solve函数

在dfs中记录了maxd 就是最深的层次数 我们从最深层开始 一直到k+1曾 因为k曾之内的 被跟节点就搞定了

我们贪心的思想是 从最后一层 直接网上找k级父亲 这样肯定是最贪心的思路了

所以 找到一个没有vis的点就先找k级父亲 这个服务器添加服务 也就是ans++ 然后把这个服务器能服务到的点全部标记vis 也就是dfs2了

这样最后就找全了 直接输出ans就好了

#include<iostream>
#include<cstdio>
#include<vector>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=1010;

vector<int> tree[maxn],node[maxn];
int n,s,k,maxd;
int fa[maxn];
bool vis[maxn];
//dfs函数完成初始化fa数组，然后把节点按照层数放到node里
void dfs(int u,int f,int d)
{
    fa[u]=f;
    int size=tree[u].size();
    if(size==1 && d>k)//size等于1 的话 这个点就是叶节点，因为层数在k之内的话 跟节点就覆盖到了 所以不用管
    node[d].push_back(u);
    for(int i=0;i<size;++i)
    {
        if(tree[u][i]!=f) dfs(tree[u][i],u,d+1);
    }
    maxd=d>maxd? d:maxd;//maxd记录树最深的层数，从0开始
}
//dfs2 就是 再标记了u添加服务后，把它能服务到的叶子节点都标记vis
void dfs2(int u,int f,int d)
{
    vis[u]=1;
    int size=tree[u].size();
    for(int i=0;i<size;++i)
    {
        int v=tree[u][i];
        if(v!=f && d<k)
         dfs2(v,u,d+1);
    }
}

int solve()
{
    int ans=0;
    memset(vis,0,sizeof(vis));
    for(int i=maxd;i>k;i--)
    {
        int si=node[i].size();
        for(int j=0;j<si;++j)
        {
            int t=node[i][j];
            if(vis[t]) continue;
            int v=t,tmp=k;
            while(tmp--) v=fa[v];
            dfs2(v,-1,0);
            ans++;
        }
    }
    return ans;
}

int main()
{
    //freopen("/home/gl/in.txt","r",stdin);
    int T,i,a,b;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d %d %d",&n, &s,&k);
        for(i=1;i<=n;++i)
        {
            tree[i].clear();
            node[i].clear();
        }
        for(i=0;i<n-1;++i)
        {
            scanf("%d %d ",&a,&b);
            tree[a].push_back(b);
            tree[b].push_back(a);
        }
        maxd=0;

        dfs(s,-1,0);
        int res=solve();
        printf("%d\n",res);
    }
    return 0;
}