Wannafly挑战赛14D.codeJan和树（dfs+线段树）

链接：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;
}