P5838 [USACO19DEC]Milk Visits G（树剖+主席树

题意就是问你：给定一棵1e5大小的树，有点权，1e5次询问（u，v，w）询问从u到v的路径是否存在w的点权——很自然想到对一个序列操作的话就是主席树，只不过上了个树，1e5的复杂度支持树剖，那就剖成一条条序列暴力即可

#include<bits/stdc++.h>
using namespace std;
const int N = 1e5+15;
const int M = 1e5;
#define pb push_back
int n,m;
vector<int>E[N];
int dep[N],fa[N],siz[N],son[N];
void dfs1(int u,int f){
	siz[u]=1;
	for(auto v:E[u])if(v!=f){
		dep[v]=dep[u]+1;
		fa[v]=u;
		dfs1(v,u);
		siz[u]+=siz[v];
		if(siz[son[u]]<siz[v])
			son[u]=v;
	}
}
int rev[N],dfn[N],top[N],tot;
void dfs2(int u,int t){
	top[u]=t;
	rev[++tot]=u;
	dfn[u]=tot;
	if(!son[u]) return;
	dfs2(son[u],t);
	for(auto v:E[u]){
		if(v!=fa[u] && v!=son[u])
			dfs2(v,v);
	}
}

int tr[200005],lc[200005*20],rc[200005*20],sm[200005*20],cnt;
void update(int &rt,int pre,int l,int r,int pos){
	rt=++cnt;
	lc[rt]=lc[pre];
	rc[rt]=rc[pre];
	sm[rt]=sm[pre]+1;
	if(l==r){
		return;
	}
	int mid=l+r>>1;
	if(pos<=mid)
		update(lc[rt],lc[pre],l,mid,pos);
	else
		update(rc[rt],rc[pre],mid+1,r,pos);
}	
int query(int now,int pre,int l,int r,int pos){
	if(l==r)
		return sm[now]-sm[pre];
	int mid=l+r>>1;
	int ans=0;
	if(pos<=mid)
		ans+=query(lc[now],lc[pre],l,mid,pos);
	else
		ans+=query(rc[now],rc[pre],mid+1,r,pos);
	return ans;
}
int query(int u,int v,int w){
	int ans=0;
	while(top[u]!=top[v]){
		if(dep[top[u]]<dep[top[v]])
			swap(u,v);
		ans+=query(tr[dfn[top[u]]-1],tr[dfn[u]],1,M,w);
		u=fa[top[u]];
	}
	if(dep[u]<dep[v])
		swap(u,v);
	ans+=query(tr[dfn[v]-1],tr[dfn[u]],1,M,w);
	return ans;
}
int w[N];
int main(){
	ios::sync_with_stdio(false);cin.tie(0);
	cin>>n>>m;
	for(int i=1;i<=n;i++)
		cin>>w[i];
	for(int i=1;i<n;i++){
		int u,v;cin>>u>>v;
		E[u].pb(v);E[v].pb(u);
	}
	dfs1(1,1);
	dfs2(1,1);
	for(int i=1;i<=n;i++){
		update(tr[i],tr[i-1],1,M,w[rev[i]]);
	}
	while(m--){
		int u,v,w;cin>>u>>v>>w;
		if(query(u,v,w))
			cout<<1;
		else
			cout<<0;
	}
	cout<<'\n';
}