牛客挑战赛28 C.msc的宠物

题目链接：https://ac.nowcoder.com/acm/contest/217/C

题意：

msc有n个小宠物，这些宠物的家是连在一起的，更有趣的是，这些宠物的家之间的连接关系形成了一个树的形态。
每个小宠物的习性是不太一样的，比如说有的可能吃素，有的可能吃荤。
作为直觉系女生，msc凭着自己突出的直觉对每个小宠物都有一个关于习性的评估值，对于宠物i，它的评估值是a[i]。
考虑到习性差异过大会很难伺候，甚至宠物间会发生冲突，所以msc希望去掉至多k个连接关系使得任意两个可以互相到达的宠物间的评估值的差的最大值尽量小。
两个宠物(x,y)可以互相到达当且仅当存在一个序列s1,s2,...,sk-1,sk使得对于1 ≤ i < k，si与si+1之间相连且s1=x,sk=y
现在msc给出了n个小宠物的评估值a[1..n]和这棵树的形态，msc希望你帮她求出去掉至多k个连接关系之后，任意两个可以互相到达的宠物间的评估值的差的最大值最小是多少。

|a[i]|≤1000000000

1≤k+1≤n≤1000

题解：

二分答案dfs check(mid)，f[x][y]表示x子树里与x相连的联通块每个点都在[a[y],a[y]+mid]需要切几条边，f[x]表示只考虑处理x子树需要切几条边

#include<bits/stdc++.h>
#define pb push_back
#define sc(x) scanf("%d",&x)
#define sll(x) scanf("%lld",&x)
using namespace std;
typedef long long LL;
const int INF = 0x3f3f3f3f;
const int maxn = 1e3+10;
vector<int> to[maxn];
int n, k, a[maxn], f[maxn][maxn], g[maxn];

void dfs(int u, int fa, LL x){
	for(int i=1;i<=n;i++) f[u][i]=(a[u]>=a[i]&&a[u]<=a[i]+x)?0:INF;
	for(int v: to[u]) if(v!=fa){
		dfs(v,u,x);
		for(int i=1;i<=n;i++) f[u][i]+=min(f[v][i],g[v]+1);
	}
	for(int i=1;i<=n;i++)g[u]=min(g[u],f[u][i]);
}
int check(LL x){
	memset(g,INF,sizeof(g));
	dfs(1,1,x);
	return g[1];
}

int main(){
	sc(n); sc(k);
	for(int i=1;i<=n;i++) sc(a[i]);
	for(int i=1;i<n;i++){
		int u,v; sc(u); sc(v);
		to[u].pb(v); to[v].pb(u);
	}
	LL L=-1,R=INF*2;
	while(L+1<R){
		LL mid=(L+R)>>1;
		if(check(mid)>k) L=mid;
		else R=mid;
	}
	printf("%lld\n",R);
	return 0;
}