洛谷P3806 【模板】点分治1 【点分治】

题意：

给出一棵树，每条边都有一个权值$w$，$m$次询问，每次询问树上是否有一条长度为$k$的路径

题解：

点分治模板题，但是处理起来麻烦

#include<iostream>
#include<sstream>
#include<string>
#include<queue>
#include<map>
#include<unordered_map>
#include<set>
#include<vector>
#include<stack>
#include <utility>
#include<algorithm>
#include<cstdio>
#include<list>
#include<cmath>
#include<cstdlib>
#include<cstring>
#include<iomanip>
#include<time.h>
//#define int long long
#define PI acos(-1.0)
#define eps 1e-9
#define lowbit(a) ((a)&-(a))
const int mod = 1e9+7;
using namespace std;
inline int read(){
	char c=getchar();int x=0,s=1;
	while(c<'0'||c>'9'){if(c=='-')s=-1;c=getchar();}
	while(c>='0'&&c<='9'){x=x*10+c-'0';c=getchar();}
	return x*s;
}
int qpow(int a,int b){
	int ans=1;
	while(b){
		if(b&1)ans=(ans*a)%mod;
		a=(a*a)%mod;
		b>>=1;
	}
	return ans;
}
#define endl '\n'
const int INF = 0x3f3f3f3f;
const int N = 1e4+100;
const int maxn = 1e3+10;


struct edge{
	int w,to,next;
}e[N];
int head[N],cnt,n,qt,ans;
int vis[N],son[N],sz[N],q[N],ck[N];
int dis[N],root[N],id[N];
int alls,MX,rt,m;

void add(int u,int v,int w){
	e[++cnt].w=w;
	e[cnt].to=v;
	e[cnt].next=head[u];
	head[u]=cnt;
}

void init(){
	memset(head,0,sizeof head);
	memset(vis,0,sizeof vis);
	cnt=ans=0;
	MX=INF,alls=n;
}

void get_rt(int u,int fa){
  	sz[u]=1,son[u]=0;
  	for(int i=head[u];i;i=e[i].next){
      	int v=e[i].to;
      	if(v==fa||vis[v])continue;
      	get_rt(v,u);
      	sz[u]+=sz[v];
      	son[u]=max(sz[v],son[u]);
    }
  	son[u]=max(son[u],alls-sz[u]);
  	if(son[u]<MX)rt=u,MX=son[u];
}

void get_dis(int u,int fa,int dist,int rt){
	id[++m]=u;
	root[u]=rt;
	dis[u]=dist;
  	for(int i=head[u];i;i=e[i].next){
      	int v=e[i].to;
     	if(v==fa||vis[v])continue;
    	get_dis(v,u,dist+e[i].w,rt);
    }
  	return;
}

bool cmp(int a,int b){ return dis[a]<dis[b]; }
void calc(int rt){
	m=0;
	id[++m]=rt;
	root[rt]=rt;
	dis[rt]=0;
	for(int i=head[rt];i;i=e[i].next){
		int v=e[i].to;
		if(vis[v])continue;
		get_dis(v,rt,e[i].w,v);
	}
  	sort(id+1,id+1+m,cmp);
  	for(int i=1;i<=qt;i++){
		int l=1,r=m;
		if(ck[i])continue;
		while(l<r){
			if(dis[id[l]]+dis[id[r]]>q[i]) r--;
			
			else if(dis[id[l]]+dis[id[r]]<q[i]) l++;
			
			else if(root[id[l]]==root[id[r]]){
				if(dis[id[l]]==dis[id[l+1]]) l++;
				else r--;
			}
			else{
				ck[i]=1; break;
			}
		}
	}
}

void div(int tr){
  	calc(tr);
  	vis[tr]=1;
  	for(int i=head[tr];i;i=e[i].next){
      	int v=e[i].to;
      	if(vis[v])continue;
      	alls=son[v],rt=0,MX=INF;
		get_rt(v,v);
      	div(rt);
    }
}

signed main(){
	std::ios::sync_with_stdio(false);
	cin.tie(0),cout.tie(0);
	cin>>n>>qt;
	init();
	for(int i=1;i<n;i++){
		int u,v,w; cin>>u>>v>>w;
		add(u,v,w); add(v,u,w);
	}
	for(int i=1;i<=qt;i++)cin>>q[i],ck[i]=0;
	get_rt(1,1);
	div(rt); 
	for(int i=1;i<=qt;i++)cout<<(ck[i]?"AYE":"NAY")<<endl;
}