题目链接
自己假自己,怎么写都只能过一半的测试点,学了别人的题解。
题目描述
给定一棵树,每个结点有一个权值,求
k
+
1
k+1
k+1个连通块中最值之差的最小值是多少。
思路
二分答案很显然,主要是用树形dp计算最少需要的次数。
d
p
[
u
]
[
i
]
dp[u][i]
dp[u][i]:结点
u
u
u的值
a
[
u
]
a[u]
a[u]在范围为
[
a
i
,
a
i
+
m
i
d
]
[a_i,a_i + mid]
[ai,ai+mid]中,且满足其子树所有点的值在该范围中时需要删除的边数。
f
[
u
]
f[u]
f[u]:结点
u
u
u所需要删除的最少的边的数量。
转移方程:
d
p
[
u
]
[
i
]
=
m
i
n
(
d
p
[
v
]
[
i
]
,
f
[
v
]
+
1
)
dp[u][i]=min(dp[v][i],f[v]+1)
dp[u][i]=min(dp[v][i],f[v]+1)
代码
#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int inf = 0x3f3f3f3f;
const int N = 1010;
LL dp[N][N], f[N];
LL a[N];
vector<int> e[N];
int n, k;
void dfs(int u, int fa, LL mid) {
for(int i = 1; i <= n; i++) {
if(a[i] <= a[u] && a[u] <= a[i] + mid) {
dp[u][i] = 0;
} else dp[u][i] = inf;
}
for(int i = 0; i < e[u].size(); i++) {
int v = e[u][i];
if(v == fa) continue;
dfs(v, u, mid);
for(int j = 1; j <= n; j++) {
dp[u][j] += min(dp[v][j], f[v] + 1);
}
}
for(int i = 1; i <= n; i++) {
f[u] = min(f[u], dp[u][i]);
}
}
bool check(LL mid) {
memset(f, 0x3f, sizeof f);
dfs(1, 0, mid);
return f[1] <= k;
}
void solve() {
scanf("%d%d", &n, &k);
for(int i = 1; i <= n; i++) {
scanf("%lld", &a[i]);
}
for(int i = 1; i < n; i++) {
int u, v;
scanf("%d%d", &u, &v);
e[u].push_back(v);
e[v].push_back(u);
}
LL l = 0, r = 2e9;
LL res = -1;
while(l <= r) {
LL mid = l + r >> 1;
// printf("mid = %lld\n", mid);
if(check(mid)) {
res = mid;
r = mid - 1;
} else l = mid + 1;
}
printf("%lld\n", res);
}
int main() {
// freopen("in.txt", "r", stdin);
solve();
return 0;
}