BZOJ 1316: 树上的询问 点分治

1316: 树上的询问

Time Limit: 10 Sec  Memory Limit: 162 MB
Submit: 691  Solved: 187
[Submit][Status][Discuss]

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． 
做完此题可看下POJ 3237 Tree

树上路径问题

考虑点分治

有q组询问

那我们每组询问都搞一遍就好了。。。

点分治中间搞会比搞q次快很多。。。

#include<cmath>
#include<ctime>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<complex>
#include<iostream>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<string>
#include<bitset>
#include<queue>
#include<set>
#include<map>
using namespace std;
typedef long long ll;
inline int read()
{
	int x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch<='9'&&ch>='0'){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
	return x*f;
}
const int N=1000100;
const int M=10010;
const int Q=111;
int n,m,ecnt,sum,last[M],dis[M],t[N],d[M],f[M],g[M],num[M],size[M],q[Q],book[Q],root;
bool vis[M];
struct EDGE{int to,nt,val;}e[M<<1];
inline void add(int u,int v,int val)
{e[++ecnt]=(EDGE){v,last[u],val};last[u]=ecnt;}
void getroot(int u,int fa)
{
	d[u]=d[fa]+1;size[u]=1;f[u]=0;
	for(int i=last[u];i;i=e[i].nt)
	if(e[i].to!=fa&&!vis[e[i].to])
	{
		getroot(e[i].to,u);
		size[u]+=size[e[i].to];
		f[u]=max(f[u],size[e[i].to]);
	}
	f[u]=max(f[u],sum-size[u]);
	if(f[u]<f[root])root=u;
}
void seek(int u,int fa)
{
	t[++t[0]]=dis[u];
	for(int i=last[u];i;i=e[i].nt)
	if(!vis[e[i].to]&&e[i].to!=fa)
	{
		dis[e[i].to]=dis[u]+e[i].val;
		seek(e[i].to,u);
	}
}
inline void cal(int u,int val,int opt)
{
	dis[u]=val;t[0]=0;seek(u,0);
	sort(t+1,t+1+t[0]);
	g[0]=0;
	for(int i=1;i<=t[0];i++)
	if(i==1||t[i]!=t[i-1])
	g[++g[0]]=t[i],num[g[0]]=1;
	else num[g[0]]++;
	for(int i=1;i<=m;i++)
	{
		int l=1,r=g[0];
		for(int j=1;j<=g[0];j++)
		if(num[j]>1&&g[j]*2==q[i])
		book[i]+=opt*num[j]*(num[j]-1);
		while(l<r)
		{
			if(g[l]+g[r]<q[i])l++;
			else 
			{
				if(g[l]+g[r]==q[i])book[i]+=opt*num[l]*num[r];
				r--;
			}
		}
	}
}
void solve(int u)
{
	cal(u,0,1);vis[u]=1;
	for(int i=last[u];i;i=e[i].nt)
	if(!vis[e[i].to])
	{
		cal(e[i].to,e[i].val,-1);root=0;sum=size[e[i].to];
		getroot(e[i].to,u);solve(root);
	}
}
int main()
{
	n=read();m=read();int u,v,val;
	for(int i=1;i<n;i++)
	{
		u=read();v=read();val=read();
		add(u,v,val);add(v,u,val);
	}
	for(int i=1;i<=m;i++)q[i]=read();f[0]=0X3f3f3f3f;
	sum=n;getroot(1,0);solve(root);
	for(int i=1;i<=m;i++)if(book[i]>0||!q[i])puts("Yes");
	else puts("No");
	return 0;
}
/*
6 4 
1 2 5 
1 3 7 
1 4 1 
3 5 2 
3 6 3 
1 
8 
13 
14 

Yes 
Yes 
No 
Yes 
*/