1401 有趣的旅行

有N个城市，每个城市编号时从0到(N - 1)，它们靠(N - 1)条双向道路相连。任意两个不同地城市之间都有一条唯一的路径相连。每个城市还有一个吸引力值，吸引力值越大的城市代表它越有趣，就越能吸引人。你要决定一次旅行，满足如下条件：

（1） 计划中最多去K个城市

（2） 必须可以从计划中的某个城市出发只经过计划中的城市达到计划中其他的城市——中途不能经过任何计划外的城市

（3） 选入计划的城市不能比任何计划外的城市的吸引力低。城市的吸引力可能不唯一，所以对于吸引力相等的城市可以只选择其中的一部分列入计划。

你的目标时在同时满足上述条件下，选择列入计划的城市尽可能多。

如图，灰色圆圈的代表城市，圆圈内的数字代表城市的编号，圆圈旁边的数字代表城市的吸引力。

如果K = 2, 你可以选择城市2和城市0， 或者城市2和城市4。

如果K = 5, 你最多只能选4个城市： 2， 0， 4和5。

Input

第一行包含两个正整数N和K。(1<=K<=N<=100000)。
接下来一行包含N个正数，每个数不超过1000000，表示每个城市的吸引力的值。
再接下来(N - 1)行，每行时两个不超过(N - 1)的非负整数，表示道路。

Output

一行包含一个整数，表示满足要求的旅行计划中最多能选择的城市数量。

Input示例

7 5
6 2 7 5 6 5 2
3 1
1 0
0 2
2 4
4 5
4 6

Output示例

4

题解：将所有点按v从大到小排序，每次扩展价值相同的点，c1表示当前连通块的个数，只有当i==0||c1==1时(即上一步扩展的点全部连通时)才进行ans更新.

eg：样例  第一次扩展    2(7)                                  2(7)                                                       ans=1,c1=1;

               第二次扩展    0(6) 4(6)             0(6)---------2(7)--------4(6)                                          ans=3,c1=1;

               第三次扩展    3(5)  5(5)            3(5)      0(6)---------2(7)--------4(6)----------5(5)               ans=4,c1=2;

dfs()的返回值为连接所有大于vmin点的个数.

用到dfs()的情况：k=3;

1(5)-----2(4)-----3(5)----4(6)          第二次扩展后    1(5）    3(5)----4(6)      ans=2<k;

            |                                  第三次扩展后     1(5)-----2(4)-----3(5)----4(6)     dfs()=4>k,所以ans不能更新.若k=4，则ans=k;

           5(4)                                                                  |

                                                                                  5(4)

#include<iostream>
#include<stdio.h>
#include<vector>
#include<algorithm>
using namespace std;
const int maxn=1e5+10;
int n,kk,c1,c2,ans,Max,v[maxn],id[maxn],fa[maxn],sz[maxn];
bool vis[maxn],is[maxn];
vector<int> e[maxn];
bool cmp(int a,int b){
    return v[a]>v[b];
}
int getfa(int x){
    int c,a=x;
    while(x!=fa[x]) x=fa[x];
    while(x!=fa[a]) c=fa[a],fa[a]=x,a=c;
    return x;
}
void mer(int a,int b,int x){
    a=getfa(a),b=getfa(b);
    if(v[a]>v[b]) swap(a,b);
    fa[a]=b;
    sz[b]+=sz[a];
    if(v[a]==x) c2--;
    else        c1--;
}
int dfs(int u,int f){
    int ret=1;
    bool flag=v[u]>Max;
    for(int i=0;i<e[u].size();i++){
        int w=e[u][i];
        if(w==f||!is[w]) continue;
        int s=dfs(w,u);
        if(s) ret+=s,flag=1;
    }
    return flag?ret:0;
}
int main(){
    scanf("%d%d",&n,&kk);
    for(int i=0;i<n;i++){
        scanf("%d",&v[i]);
        id[i]=fa[i]=i;
        sz[i]=1;
    }
    sort(id,id+n,cmp);
    for(int u,v,i=1;i<n;i++){
        scanf("%d%d",&u,&v);
        e[u].push_back(v);
        e[v].push_back(u);
    }
    for(int i=0,j=0;i<n;i=j){
        while(j<n&&v[id[i]]==v[id[j]]) j++;
        for(int k=i;k<j;k++){
            int w=id[k];
            vis[w]=1;
            c2++;
            for(int a=0;a<e[w].size();a++){
                int u=e[w][a];
                if(vis[u]) mer(w,u,v[w]);
            }
        }
        if(!i) for(int k=0;k<j;k++) ans=max(ans,sz[getfa(id[k])]);
        else if(c1==1){
            int s=sz[getfa(id[0])];
            if(s<=kk) ans=max(ans,s);
            else{
                for(int k=0;k<j;k++) is[k]=1;
                Max=v[id[i]];
                int z=dfs(id[0],-1);
                if(z<=kk) ans=kk;
                break;
            }
        }
        c1+=c2;
        c2=0;
    }
    printf("%d\n",ans);
}