uvalive3902（树上的最优化 贪心）

题目链接：https://vjudge.net/problem/UVALive-3902

#include<cstdio>
#include<cstring>
#include<vector>
#include<algorithm>
using namespace std;

const int maxn=1010;
vector<int> gr[maxn],nodes[maxn];
int n,s,k,fa[maxn];
bool covered[maxn];
void dfs(int u,int f,int d)
{
    fa[u]=f;  //计算fa数组
    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 dfs2(int u,int f,int d)
{
    covered[u]=1;
    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的点
    }
}

int solve()
{
    int ans=0;
    memset(covered,0,sizeof(covered));
    for(int d=n-1;d>k;d--)
        for(int i=0;i<nodes[d].size();i++)
    {
        int u=nodes[d][i];
        if(covered[u]) continue; //已经覆盖的不考虑
        int v=u;
        for(int j=0;j<k;j++) v=fa[v];  //v是u的k级祖先
        dfs2(v,-1,0);  //在节点v放置服务器
        ans++;
    }
    return ans;
}
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d%d",&n,&s,&k);
        for(int i=0;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);
        printf("%d
",solve());
    }
    return 0;
}