如果给你一个查询 那就是裸的点分治 但是给你m个查询 你该咋办呢
不妨在查询之前先处理好所有可能出现的边 然后在线查询就行
总根进行++操作 消除子树的影响 进行--操作
具体可以看代码吧
/*
luogu 3806
*/
#include <cstdio>
#include <cstring>
#include <queue>
#include <cmath>
#include <iostream>
#include <stack>
#include <set>
#include <map>
#include <sstream>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long ll;
const int MAX_N = 100024;
int f[MAX_N],siz[MAX_N];
struct edge{
int next,v,dis;
}e[MAX_N<<1];
int eid,p[MAX_N],use[MAX_N],Siz,rt,cnt,k,ask[MAX_N],ANS[MAX_N],m,sum[MAX_N];
int d[10000005];
inline int read()
{
int date = 0,m = 1;
char ch = 0;
while(ch!='-'&&(ch<'0'||ch>'9'))
ch = getchar();
if(ch=='-')
{
m = -1;
ch = getchar();
}
while(ch>='0' && ch<='9')
{
date = date*10+ch-'0';
ch = getchar();
}
return date*m;
}
inline void write(ll qw)
{
if(qw<0)
{
putchar('-');
qw = -qw;
}
if(qw>9)
write(qw/10);
putchar(qw%10+'0');
}
void init(){
memset(p,-1,sizeof(p));
memset(use,0,sizeof(use));
rt = 0;
eid = 0;
}
void Insert(int u,int v,int dis){
e[eid].v = v;
e[eid].dis = dis;
e[eid].next = p[u];
p[u] = eid++;
}
void get_rt(int u,int fa){//u为当前点,fa为父亲节点
f[u] = 0, siz[u] = 1;//f表示这个点最大子树的大小,siz是这个点子树大小的和
for(int i = p[u];i!=-1;i = e[i].next){//枚举儿子
int y = e[i].v;
if(use[y]||y==fa) continue;//use表示之前遍历过了,这里没啥用
get_rt(y,u);//往下遍历
f[u] = max(f[u],siz[y]);//更新f
siz[u] += siz[y];
}
f[u] = max(f[u],Siz - siz[u]);//Siz表示在现在这棵子树中点的总数,开始时Siz=n,除了枚举的儿子所在的子树外,还有一棵子树是上面的那一堆,容斥原理
if(f[u]<f[rt]) rt = u;//更新root
}
void query(int u,int fa,int dis){
d[++cnt]=dis;
for(int i = p[u];i!=-1;i=e[i].next){
int y = e[i].v;
if(use[y]||y==fa) continue;
query(y,u,dis+e[i].dis);
}
return ;
}
void solve(int u,int dis,int flag){
cnt = 0;
query(u,0,dis);
if(flag==1){
for(int i = 1;i<cnt;++i)
for(int j = i+1;j<=cnt;j++){
sum[d[i]+d[j]]++;
}
}
else {
for(int i = 1;i<cnt;++i)
for(int j = i+1;j<=cnt;j++)
sum[d[i]+d[j]]--;
}
return;
}
void dfs(int u){//Divide
use[u] = 1,solve(u,0,1);
for(int i = p[u];i!=-1;i=e[i].next){
int y = e[i].v;
if(use[y]) continue;
solve(y,e[i].dis,0);
Siz = siz[y],rt = 0;
get_rt(y,u),dfs(rt);
}
return ;
}
int main(){
int n;
n = read(),m = read();
init();
for(int i = 1;i<n;++i){
int a,b,dis;
a = read(),b = read(),dis = read();
Insert(a,b,dis);
Insert(b,a,dis);
}
f[0] = Siz = n;
get_rt(1,0);
dfs(rt);
int opt;
for(int i = 1;i<=m;++i){
opt = read();
if(sum[opt]) printf("AYE\n");
else printf("NAY\n");
}
return 0;
}