链接:https://ac.nowcoder.com/acm/contest/81/D
来源:牛客网
codeJan和树
时间限制:C/C++ 1秒,其他语言2秒
空间限制:C/C++ 262144K,其他语言524288K
64bit IO Format: %lld
题目描述
codeJan有一天脑洞大开,想到一个有趣的问题。给一个固定根为1号结点的树,定义一个子树的beauty是这个子树的根节点到所有这棵树上其他节点的距离和,叶子节点的beauty是0。定义一个子树的sub-beauty是这个子树的beauty值减去这个子树的某一个子树(不包括自身)的beauty值。显然一个子树的beauty值是唯一的,而sub-beauty值可以有很多个。codeJan想要知道所有子树的所有sub-beauty中不超过m的最大值。
输入描述:
第一行是一个T≤20代表测试组数。每组测试的第一行包含两个正整数是n,m(n≤105,m≤108),接下来n−1
行每行包含三个正整数a b d,分别表示a结点和b结点之间的距离是d,a,b∈[1,n],1≤d≤103。请注意每棵树的根节点都是1号结点,并且保证输入合法。
输出描述:
对于每组测试样例输出一个整数表示所有子树sub-beauty中不超过m的最大值。如果所有子树的sub-beauty都大于m,输出-1。
示例1
输入
复制
3
3 4
1 2 1
1 3 2
3 4
1 2 3
1 3 2
4 6
1 2 2
2 3 5
3 4 2
输出
复制
3
-1
6
这个题一开始思维没错,对于每个点,我们可以得到这个子树的权值val[u],然后我们可以在他的子树中寻找一个子节点v,使得val[u]-val[v]<=m且最大,更新当前的答案。
然后我就陷入了如何用数据结构快速的查询上面那个式子的最大值23333,结果就错了。
然后学弟用的也是差不多的思路,但并没有什么dfs序,直接一遍dfs下去,对于每个子树u,我们要在1到m+val[v]中找到一个当前已经存在的最大值,这个最大值就是当前的答案,那么直接对这些权值离散化然后建立线段树,下去一遍挨个查询,回来的时候将这个权值删除就好了。
/**
* author: MySakure
* created: 06.11.2019 11:17:38
**/
#include<bits/stdc++.h>
using namespace std;
#define pb push_back
#define mp make_pair
#define all(x) (x).begin(),(x).end()
#define int long long
const int maxn = 1e5 + 10;
int T, n, m;
vector<pair<int, int>>g[maxn];
int s, dep[maxn], ans, sz[maxn];
vector<int>v;
int getid(int u) {
return lower_bound(v.begin(), v.end(), u) - v.begin() + 1;
}
int dfs(int u, int fa) {
for (auto v : g[u]) {
if (v.first == fa)continue;
dep[u] += dfs(v.first, u) + v.second * sz[v.first];
sz[u] += sz[v.first];
}
sz[u]++;
v.push_back(dep[u]);
//cout<<v.size()<<" size"<<endl;
return dep[u];
}
struct node {
int l, r, Max;
}t[maxn << 2];
void build(int u, int l, int r) {
t[u].l = l, t[u].r = r;
t[u].Max = 0;
if (l == r)return;
int mid = (l + r) / 2;
build(u << 1, l, mid);
build(u << 1 | 1, mid + 1, r);
}
int quare(int u, int L, int R) {
int l = t[u].l, r = t[u].r;
if (l == L && R == r) {
return t[u].Max;
}
int mid = (l + r) >> 1;
if (L > mid)return quare(u << 1 | 1, L, R);
else if (R <= mid)return quare(u << 1, L, R);
else return max(quare(u << 1, L, mid), quare(u << 1 | 1, mid + 1, R));
}
void upd(int u, int inx, int v) {
int l = t[u].l, r = t[u].r;
if (l == r) {
t[u].Max = v;
return;
}
int mid = (l + r) >> 1;
if (inx > mid)upd(u << 1 | 1, inx, v);
else upd(u << 1, inx, v);
t[u].Max = max(t[u << 1].Max, t[u << 1 | 1].Max);
}
void dfs2(int u, int fa) {
//cout<<"debug "<<u<<" "<<getid(dep[u])<<" "<<dep[u]<<endl;
if (getid(m + dep[u]) != 1) {
//cout<<"debug "<<u<<" "<<getid(dep[u])<<" "<<dep[u]<<endl;
int pos = getid(m + dep[u]) - 1;
//cout<<"pos "<<pos<<endl;
if (pos != n && v[pos] == m + dep[u])++pos;
int cur = quare(1, 1, pos);
if (cur != 0) {
ans = max(ans, cur - dep[u]);
}
}
upd(1, getid(dep[u]), dep[u]);
for (auto v : g[u]) {
if (v.first == fa)continue;
dfs2(v.first, u);
}
upd(1, getid(dep[u]), 0);
}
signed main() {
scanf("%lld", &T);
while (T--) {
scanf("%lld%lld", &n, &m);
for (int i = 1; i <= n; ++i) {
g[i].clear();
dep[i] = sz[i] = 0;
}
for (int i = 1; i < n; ++i) {
int a, b, c; scanf("%lld%lld%lld", &a, &b, &c);
g[a].push_back({ b,c });
g[b].push_back({ a,c });
}
v.clear();
ans = -1;
dfs(1, -1);
sort(all(v));
v.erase(unique(all(v)), v.end());
build(1, 1, n + 1);
dfs2(1, -1);
printf("%lld\n", ans);
}
return 0;
}