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

最新推荐文章于 2022-07-04 08:25:42 发布

滑稽大佬

最新推荐文章于 2022-07-04 08:25:42 发布

阅读量231

点赞数

分类专栏：点分治文章标签：点分治

本文链接：https://blog.csdn.net/qq_35786326/article/details/95580733

版权

点分治专栏收录该内容

2 篇文章 0 订阅

订阅专栏

233

题目：
题意：
分析：
代码：

题目：

传送门

题意：

点分治！！！

分析：

点分治 $. .$
滚，虽然也没有可以讲的~~扣就完事了~~理解好就行
这里讲一下自己计算答案时的一个部分，我们设 $tt_i$ 表示到当前重心的距离为 $i$ 的路径是否存在
对于每个询问 $q_i$ ，因为我们便利完前面的 $i - 1$ 棵子树，所以用 $tt_{q_i-i}$ 来判断在之前的子树中，有没有能与当前这个距离为 $i$ 的路径拼出长度为 $q_i$ 的路径，如果有，那么答案就是 $A Y E$ ，否则就是 $N A Y$

代码：

#include<cstdio>
#include<string>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>
#define LL long long 
#define LZX WCIMU
using namespace std;
inline LL read() {
    LL d=0,f=1;char s=getchar();
    while(s<'0'||s>'9'){if(s=='-')f=-1;s=getchar();}
    while(s>='0'&&s<='9'){d=d*10+s-'0';s=getchar();}
    return d*f;
}  
struct node{
	LL to,next,w;
}e[20005];
LL ls[20005],cnt=0,n=read(),m=read();
void add(LL x,LL y,LL w)
{
	e[cnt]=(node){y,ls[x],w};
	ls[x]=cnt++;
	return;
}
LL qu[105],maxp[10005],size[10005],sum,tf[10005],root=0;
void find(LL u,LL fa)
{
	size[u]=1;maxp[u]=0;
	for(LL i=ls[u];~i;i=e[i].next)
	{
		LL v=e[i].to;
		if(v==fa||tf[v]) continue;
		find(v,u);
		size[u]+=size[v];
		maxp[u]=max(maxp[u],size[v]);
	}
	if(maxp[u]<maxp[root]) root=u;
	return;
}
LL dis[10005],l1,l2,tt[10000005],rem[10005],bc[10005],ff[105];
void get(LL u,LL fa)
{
	rem[++l1]=dis[u];
	for(LL i=ls[u];~i;i=e[i].next)
	{
		LL v=e[i].to;
		if(v==fa||tf[v]) continue;
		dis[v]=dis[u]+e[i].w;
		get(v,u);
	}
	return;
}
void calc(LL u)
{
	l2=0;
	for(LL i=ls[u];~i;i=e[i].next)
	{
		LL v=e[i].to;
		if(tf[v]) continue;
		l1=0;dis[v]=e[i].w;
		get(v,u);
		for(LL j=l1;j;j--)
		  for(LL k=1;k<=m;k++)
		    if(qu[k]>=rem[j]) ff[k]|=tt[qu[k]-rem[j]];
		for(LL j=1;j<=l1;j++) bc[++l2]=rem[j],tt[rem[j]]=1;
	}
	for(LL i=1;i<=l2;i++) tt[bc[i]]=0;
	return;
}
void work(LL u)
{
	tf[u]=tt[0]=1; 
	calc(u);
	for(LL i=ls[u];~i;i=e[i].next)
	{
		LL v=e[i].to;
		if(tf[v]) continue;
		sum=size[v];maxp[root=0]=23333333;
		find(v,0);work(v);
	}
	return;
}
int main()
{
	memset(ls,-1,sizeof(ls));
	for(LL i=1;i<n;i++)
	{
		LL x=read(),y=read(),w=read();
		add(x,y,w);add(y,x,w);
	}
	for(LL i=1;i<=m;i++) qu[i]=read();
	maxp[0]=sum=n;
	find(1,0);
	work(root);
	for(LL i=1;i<=m;i++) if(ff[i]) printf("AYE\n"); else printf("NAY\n");
	return 0;
}