题目链接:bzoj1316
题意:
现在给你一个n个点的树,给你q个询问,每次询问,询问树上是否有一条长度为k的路径。
n<=10000,q<=100.
题解:
我们发现q是极小的,那么我们只要每次询问能够做到的复杂度限制就不会太高,我们不妨考虑只有一个询问,要知道树上是否有一条长度为k的路径,我们不妨用点分治的方法来对路径进行分类,这样我们只要能够在较低的复杂度下处理点分治时的查询即可,我们可以点分治时,先将所有的dis(即这一层所有点到重心的距离)处理出来,放进一个set中,我们查询以某个节点为端点,经过重心的路径是否有长度为k的,这可以用set很容易做到,当然我们需要先把与这个节点在同一子层中的点给剔掉。这样此题就可以在nlogn^2的复杂度中得到解决。
#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <ctime>
#include <cmath>
#include <algorithm>
#include <set>
#define lld long long
using namespace std;
const int maxn=10010;
struct Edge{
int to,next,w;
}edge[maxn*2];
int head[maxn],vis[maxn*2],p,n,use[maxn*2],line[maxn],size[maxn],op,fa[maxn],can[maxn],que[maxn],OP;
lld dis[maxn],l[maxn];
void build(int a,int b,int c){
edge[++p]=(Edge){b,head[a],c};
head[a]=p;
}
void dfs(int a){
use[a]=1;
size[a]=1;
line[++op]=a;
for (int i=head[a];i;i=edge[i].next)
if (!vis[i]&&!use[edge[i].to]){
fa[edge[i].to]=a;
dfs(edge[i].to);
size[a]+=size[edge[i].to];
}
}
void dfs2(int a){
use[a]=1;
line[++op]=a;
for (int i=head[a];i;i=edge[i].next)
if (!vis[i]&&!use[edge[i].to]){
dis[edge[i].to]=dis[a]+edge[i].w;
fa[edge[i].to]=a;
dfs2(edge[i].to);
}
}
void dfs3(int a){
que[++OP]=a;
for (int i=head[a];i;i=edge[i].next)
if (!vis[i]&&fa[edge[i].to]==a)
dfs3(edge[i].to);
}
int store[maxn];
int get(int a){
op=0;
fa[a]=0;
dfs(a);
int ans=a;
for (int i=1;i<=op;i++){
store[line[i]]=size[a]-size[line[i]];
for (int j=head[line[i]];j;j=edge[j].next)
if (!vis[j]&&edge[j].to!=fa[line[i]])
store[line[i]]=max(store[line[i]],size[edge[j].to]);
if (store[line[i]]<store[ans])
ans=line[i];
use[line[i]]=0;
}
return ans;
}
set<lld> s;
int q;
void fenzhi(int a){
s.clear();
int point=get(a);
op=0;
fa[point]=0;
dis[point]=0;
dfs2(point);
for (int i=1;i<=op;i++)
s.insert(dis[line[i]]);
for (int i=head[point];i;i=edge[i].next)
if (!vis[i]){
OP=0;
dfs3(edge[i].to);
for (int j=1;j<=OP;j++)
s.erase(dis[que[j]]);
set<lld>::iterator it;
for (int j=1;j<=OP;j++)
for (int k=1;k<=q;k++)
if (!can[k]){
it=s.find(l[k]-dis[que[j]]);
if (it!=s.end())
can[k]=1;
}
for (int j=1;j<=OP;j++)
s.insert(dis[que[j]]);
}
for (int i=1;i<=op;i++)
use[line[i]]=0;
for (int i=head[point];i;i=edge[i].next)
if (!vis[i]){
vis[i]=1;
vis[i^1]=1;
fenzhi(edge[i].to);
}
}
int main(){
freopen("1316.in","r",stdin);
freopen("1316.out","w",stdout);
scanf("%d%d",&n,&q);
p=1;
for (int i=1;i<=n-1;i++){
int x,y,c;
scanf("%d%d%d",&x,&y,&c);
build(x,y,c);
build(y,x,c);
}
for (int i=1;i<=q;i++) scanf("%lld",&l[i]);
fenzhi(1);
for (int i=1;i<=q;i++){
if (!can[i]&&l[i]!=0)
printf("No");
else printf("Yes");
printf("\n");
}
fclose(stdin);
fclose(stdout);
}