poj 3162 WalkingRace 树形dp+线段树

题目大意：

从任意一个结点出发，都要走到距离当前距离的最远的点，然后把这些距离存储下来，求差值在一定范围内最长的连续序列长度是多长；

解题思路：

求树上任意一点到树上某一点的最远距离，我们可以证明这一点一定是树的直径中的某一个端点；我们可以通过3次dfs找出树上每一点的最远距离；

然后通过控制l,r来寻找区间最长的连续序列，这期间通过线段树查找区间的最大最小值；

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define maxn 1000010
#define L(a) a<<1
#define R(a) a<<1|1
int n,m,ans;
int cnt,head[maxn];
bool vis[maxn];
int dp[maxn],maxlen,end;
struct Edge{
	int to,next,w;
}edge[3*maxn];
void init()
{
	maxlen=ans=cnt=0;
	end=0;
	memset(vis,0,sizeof(vis));
	memset(head,-1,sizeof(head));
	memset(dp,0,sizeof(dp));
}
void addedge(int u,int v,int w)
{
	edge[cnt].to=v;
	edge[cnt].w=w;
	edge[cnt].next=head[u];
	head[u]=cnt++;
}
void dfs(int u,int len)
{
	vis[u]=1;
	if(len>maxlen) maxlen=len,end=u;
	for(int i=head[u];i!=-1;i=edge[i].next)
	{
		int v=edge[i].to;
		if(!vis[v])
		{
			dp[v]=max(dp[v],len+edge[i].w);
			dfs(v,len+edge[i].w);
		}
	}
}
int t1,t2;
struct Node{
	int l,r,nummax,nummin;
}node[3*maxn];
void pushup(int k)
{
	node[k].nummax=max(node[L(k)].nummax,node[R(k)].nummax);
	node[k].nummin=min(node[L(k)].nummin,node[R(k)].nummin);
}
void create(int k,int l,int r)
{
	node[k].l=l;
	node[k].r=r;
	if(l==r)
	{
		node[k].nummax=node[k].nummin=dp[l];
		return ;
	}
	int mid=(l+r)>>1;
	create(L(k),l,mid);
	create(R(k),mid+1,r);
	pushup(k);
}
void query(int k,int l,int r)
{
	if(node[k].l>=l&&node[k].r<=r)
	{
		t1=max(t1,node[k].nummax);
		t2=min(t2,node[k].nummin);
		return ;
	}
	int mid=(node[k].l+node[k].r)>>1;
	if(l<=mid)
	query(L(k),l,r);
	if(r>mid)
	query(R(k),l,r);
}
int main()
{
	while(~scanf("%d%d",&n,&m))
	{
		init();
		for(int i=1;i<n;i++)
		{
			int a,b;
			scanf("%d%d",&a,&b);
			addedge(i+1,a,b);
			addedge(a,i+1,b);
		}
		dfs(1,0);
		memset(vis,0,sizeof(vis));
		dfs(end,0);
		memset(vis,0,sizeof(vis));
		dfs(end,0);
		
		create(1,1,n);
		int l=1,r=1;
		t1=t2=dp[1];
		while(l<=r&&r<=n)
		{
			if(t1-t2<=m)
			{
				ans=max(ans,r-l+1);
				r++;
				t1=max(t1,dp[r]);
				t2=min(t2,dp[r]);
			}
			else
			{
				l++;
				t1=0,t2=0x3f3f3f3f;
				query(1,l,r);
			}
			if(n-l<ans) break;
		}
		printf("%d\n",ans);
	}
}