Tree Gym - 101964C（二分+搜索）

借鉴了一位大佬的思路 https://www.cnblogs.com/Chen-Jr/p/9879980.html

题目链接 https://cn.vjudge.net/problem/Gym-101964C

题意给你一颗有n个节点的最小生成树，每个节点都有一个颜色，1代表黑色，0代表白色，让你从n个节点中选择至少m个黑色节点。 求这m个黑色节点中任意两个黑色节点的最远距离的最小值。

分析 ：首先，题目中的”使得最大值最小“这句话就非常符合二分的条件了。因此我们考虑对答案（两个黑点的最远的距离）进行二分。

现在就要考虑如何进行check。

对于每一个二分值k，我们考虑先用bfs遍历整颗树，然后利用bfs的性质（距离当前点的所有已bfs过且小于等于k的那些点，他们之间两两距离也小于等于k），我们可以用dfs去遍历这些点集，判断点集中的黑点的个数与需要选取的黑点个数m之间的关系即可（如果数量>=m，则右指针左移动；反之左指针右移）。

#include <iostream>
#include <string.h>
#include <math.h>
#include <stdlib.h>
#include <algorithm>
#include <vector>
#include <map>
#include <queue>
#include <stdio.h>

using namespace std;

const int maxn =1000+10;

int n,m;
int a[maxn],vis[maxn];
vector<int>G[maxn];

int dfs(int now,int dis,int fa,int k)
{
    int sum=a[now];
    if(dis>=k)
        return sum;
    for(int i=0; i<G[now].size(); i++)
    {
        int v=G[now][i];
        if(!vis[v]||v==fa)
            continue;
        sum+=dfs(v,dis+1,now,k);
    }
    return sum;
}

int check(int k)///本质为一个bfs
{
    memset(vis,0,sizeof(vis));
    queue<int>Q;
    Q.push(1);
    vis[1]=1;
    while(!Q.empty())
    {
        int u=Q.front();
        Q.pop();
          vis[u]=1;///表示已经访问过

        if(dfs(u,0,0,k)>=m)///获取已访问点的集合中的黑点个数
          return 1;

        for(int i=0; i<G[u].size(); i++)
        {
            int v=G[u][i];
            if(!vis[v])
            {
                Q.push(v);
            }
        }
    }
    return 0;
}

int main()
{
    ///freopen("in.txt","r",stdin);
    while(scanf("%d %d",&n,&m)!=EOF)
    {
        for(int i=0; i<=n; i++)
            G[i].clear();

        for(int i=1; i<=n; i++)
            scanf("%d",&a[i]);

        for(int i=0; i<n-1; i++)
        {
            int u,v;
            scanf("%d %d",&u,&v);
            G[u].push_back(v);
            G[v].push_back(u);
        }

        int l=0,r=n,mid;

        while(l<r)
        {
            mid=(l+r)/2;

            if(check(mid))
                r=mid;
            else
                l=mid+1;

        }
       printf("%d\n",r);
    }
    return 0;
}