Description
一棵n个点的带权有根树,有p个询问,每次询问树中是否存在一条长度为Len的路径,如果是,输出Yes否输出No.
Input
第一行两个整数n, p分别表示点的个数和询问的个数. 接下来n-1行每行三个数x, y, c,表示有一条树边x→y,长度为c. 接下来p行每行一个数Len,表示询问树中是否存在一条长度为Len的路径.
Output
输出有p行,Yes或No.
Sample Input
6 4
1 2 5
1 3 7
1 4 1
3 5 2
3 6 3
1
8
13
14
Sample Output
Yes
Yes
No
Yes
HINT
30%的数据,n≤100.
100%的数据,n≤10000,p≤100,长度≤1000000.
[Submit][Status][Discuss]
分析:
一开始我直接进行了Q次点分治
但是交上去8ms就WA了
怀疑自己的板子写错了,但是交上板子也是WA
实际上,点分治的常数比较大
我们做Q次点分治,不如一次点分治中处理Q个询问
tip
点分治的精髓就是cal
这道题中,我们为了降低时间复杂度而且我们要找的是长度为定值的路径,
我们可以利用乘法原理进行路径统计
int cal(int now,int z)
{
tt=0; gg=0;
getdeep(now,0);
sort(d+1,d+1+tt);
g[0]=0;
for (int i=1;i<=tt;i++)
if (d[i]!=d[i-1]||i==1)
g[++gg]=d[i],num[gg]=1; //g是树链的长度
else num[gg]++; //num是每种长度树链的数量
for (int i=1;i<=Q;i++)
{
int l=1,r=gg;
for (int j=1;j<=gg;j++)
if (num[j]>1&&g[j]*2==m[i])
ans[i]+=z*num[j]*(num[j]-1);
while (l<r)
{
if (g[l]+g[r]>m[i]&&l<r) r--;
else
{
if (g[l]+g[r]==m[i])
ans[i]+=z*num[l]*num[r];
l++;
}
}
}
}
当然,我们还是可以像以前那样进行cal
但是时间稍微慢点
int cal(int now,int z)
{
tt=0;
getdeep(now,0);
sort(d+1,d+1+tt);
for (int i=1;i<=Q;i++)
{
int l=1,r=tt;
while (l<r)
{
while (d[l]+d[r]>m[i]&&l<r) r--;
int rr=r;
if (r<=l) break;
while (d[l]+d[r]==m[i])
{
ans[i]+=z;
r--;
if (r<=l) break;
}
r=rr;
l++;
}
}
}
//这里写代码片
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int N=10010;
int f[N],d[N],sz[N],deep[N],n,m[N];
int st[N],tot=0,tt,root,sum,Q,ans[N],gg,g[N],num[N];
struct node{
int x,y,v,nxt;
};
node way[N<<1];
bool vis[N];
void add(int u,int w,int z)
{
tot++;
way[tot].x=u;way[tot].y=w;way[tot].v=z;way[tot].nxt=st[u];st[u]=tot;
}
void getroot(int now,int fa)
{
f[now]=0;
sz[now]=1;
for (int i=st[now];i;i=way[i].nxt)
if (way[i].y!=fa&&!vis[way[i].y])
{
getroot(way[i].y,now);
sz[now]+=sz[way[i].y];
f[now]=max(f[now],sz[way[i].y]);
}
f[now]=max(f[now],sum-sz[now]);
if (f[now]<f[root]) root=now;
}
void getdeep(int now,int fa)
{
d[++tt]=deep[now];
for (int i=st[now];i;i=way[i].nxt)
if (way[i].y!=fa&&!vis[way[i].y])
{
deep[way[i].y]=deep[now]+way[i].v;
getdeep(way[i].y,now);
}
}
int cal(int now,int z)
{
tt=0; gg=0;
getdeep(now,0);
sort(d+1,d+1+tt);
g[0]=0;
for (int i=1;i<=tt;i++)
if (d[i]!=d[i-1]||i==1)
g[++gg]=d[i],num[gg]=1; //g是树链的长度
else num[gg]++; //num是每种长度树链的数量
for (int i=1;i<=Q;i++)
{
int l=1,r=gg;
for (int j=1;j<=gg;j++)
if (num[j]>1&&g[j]*2==m[i]) //两条等长链
ans[i]+=z*num[j]*(num[j]-1);
while (l<r)
{
if (g[l]+g[r]>m[i]&&l<r) r--;
else
{
if (g[l]+g[r]==m[i]) //两条不等长链
ans[i]+=z*num[l]*num[r];
l++;
}
}
}
}
void dfs(int now)
{
vis[now]=1;
deep[now]=0;
cal(now,1);
for (int i=st[now];i;i=way[i].nxt)
if (!vis[way[i].y])
{
deep[way[i].y]=way[i].v;
cal(way[i].y,-1);
root=0; sum=sz[way[i].y];
getroot(way[i].y,0);
dfs(root);
}
}
int main()
{
scanf("%d%d",&n,&Q);
for (int i=1;i<n;i++)
{
int u,w,z;
scanf("%d%d%d",&u,&w,&z);
add(u,w,z); add(w,u,z);
}
for (int i=1;i<=Q;i++) scanf("%d",&m[i]);
memset(vis,0,sizeof(vis));
f[0]=N; root=0; sum=n;
getroot(1,0);
dfs(root);
for (int i=1;i<=Q;i++)
if (ans[i]>0||!m[i]) printf("Yes\n");
else printf("No\n");
return 0;
}