[AMPPZ2014]Jaskinia

[AMPPZ2014]Jaskinia

题目大意：

一个\(n(n\le3\times10^5)\)的树，\(m(m\le3\times10^5)\)个约束条件\((a_i,b_i,d_i)\)。请你找到一个点\(x\)，满足\(dist(x,a_i)+dist(x,b_i)\le d_i\)。

思路：

对于每个条件，找出尽可能靠近根的合法点\(e_i\)，然后在所有\(e_i\)中找到最深的那个，判断是否合法。

源代码：

#include<cstdio>
#include<cctype>
#include<vector>
inline int getint() {
	register char ch;
	while(!isdigit(ch=getchar()));
	register int x=ch^'0';
	while(isdigit(ch=getchar())) x=(((x<<2)+x)<<1)+(ch^'0');
	return x;
}
const int N=3e5+1;
int a[N],b[N],d[N],par[N],dep[N];
std::vector<int> e[N];
inline void add_edge(const int &u,const int &v) {
	e[u].push_back(v);
	e[v].push_back(u);
}
void dfs(const int &x,const int &par) {
	::par[x]=par;
	dep[x]=dep[par]+1;
	for(unsigned i=0;i<e[x].size();i++) {
		const int &y=e[x][i];
		if(y==par) continue;
		dfs(y,x);
	}
}
inline int jump(int x,const int &t) {
	while(dep[x]>t) x=par[x];
	return x;
}
int main () {
	for(register int T=getint();T;T--) {
		const int n=getint(),m=getint();
		for(register int i=1;i<n;i++) {
			add_edge(getint(),getint());
		}
		dfs(1,0);
		int low=1,id=0;
		for(register int i=1;i<=m;i++) {
			a[i]=getint(),b[i]=getint(),d[i]=getint();
			const int t=std::max((dep[a[i]]+dep[b[i]]-d[i]+1)/2,1);
			if(t>=low) {
				low=t;
				id=i;
			}
		}
		const int ans=jump(a[id],low);
		dfs(ans,0);
		for(register int i=1;i<=m;i++) {
			if(dep[a[i]]+dep[b[i]]-2>d[i]) goto Fail;
		}
		printf("TAK %d\n",ans);
		goto Succ;
		Fail:
			puts("NIE");
		Succ:
			for(register int i=1;i<=n;i++) {
				e[i].clear();
			}
	}
	return 0;
}