【BZOJ2500】幸福的道路
Description
小T与小L终于决定走在一起,他们不想浪费在一起的每一分每一秒,所以他们决定每天早上一同晨练来享受在一起的时光.
他们画出了晨练路线的草图,眼尖的小T发现可以用树来描绘这个草图.
他们不愿枯燥的每天从同一个地方开始他们的锻炼,所以他们准备给起点标号后顺序地从每个起点开始(第一天从起点一开始,第二天从起点二开始……). 而且他们给每条道路定上一个幸福的值.很显然他们每次出发都想走幸福值和最长的路线(即从起点到树上的某一点路径中最长的一条).
他们不愿再经历之前的大起大落,所以决定连续几天的幸福值波动不能超过M(即一段连续的区间并且区间的最大值最小值之差不超过M).他们想知道要是这样的话他们最多能连续锻炼多少天(hint:不一定从第一天一直开始连续锻炼)?
现在，他们把这个艰巨的任务交给你了!
Input
第一行包含两个整数N, M(M<=10^9).
第二至第N行,每行两个数字Fi , Di, 第i行表示第i个节点的父亲是Fi,且道路的幸福值是Di.
Output
最长的连续锻炼天数
Sample Input
3 2
1 1
1 3
1 1
1 3
Sample Output
3
数据范围：
50%的数据N<=1000
80%的数据N<=100 000
100%的数据N<=1000 000
数据范围：
50%的数据N<=1000
80%的数据N<=100 000
100%的数据N<=1000 000
题解：这题显然可以被分成两个子任务
1.求树上距离点i最远的点到i的距离
方法：维护每个点子树中到这个点距离的最大值和次大值，然后搞一搞~
2.求最长的一段区间，使得区间中最大值和最小值的差≤M
方法：先用RMQ求出区间最大值最小值，然后上双指针法
#include <cstdio>
#include <cstring>
#include <iostream>
using namespace std;
const int maxn=1000010;
int n,m,cnt,l,r,ans;
int fa[maxn],to[maxn<<1],next[maxn<<1],head[maxn];
int d1[maxn],d2[maxn],q[maxn],from[maxn];
int Log[maxn],dm[maxn][20],dn[maxn][20];
void updata(int x,int tmp)
{
if(d1[x]<tmp) d2[x]=d1[x],d1[x]=tmp;
else d2[x]=max(d2[x],tmp);
}
void dfs1(int x)
{
int i,tmp;
q[++q[0]]=x;
for(i=head[x];i!=-1;i=next[i])
dfs1(to[i]),updata(x,d1[to[i]]+from[to[i]]);
}
void add(int a,int b)
{
to[cnt]=b;
next[cnt]=head[a];
head[a]=cnt++;
}
int gm(int a,int b)
{
int k=Log[b-a+1];
return max(dm[a][k],dm[b-(1<<k)+1][k]);
}
int gn(int a,int b)
{
int k=Log[b-a+1];
return min(dn[a][k],dn[b-(1<<k)+1][k]);
}
int main()
{
scanf("%d%d",&n,&m);
int i,j,a,b,c;
memset(head,-1,sizeof(head));
for(i=2;i<=n;i++)
{
scanf("%d%d",&fa[i],&from[i]);
add(fa[i],i);
}
dfs1(1);
for(i=2;i<=n;i++)
{
if(d1[fa[q[i]]]==d1[q[i]]+from[q[i]]) updata(q[i],d2[fa[q[i]]]+from[q[i]]);
else updata(q[i],d1[fa[q[i]]]+from[q[i]]);
}
for(i=1;i<=n;i++) dm[i][0]=dn[i][0]=d1[i];
for(i=2;i<=n;i++) Log[i]=Log[i>>1]+1;
for(j=1;(1<<j)<=n;j++)
{
for(i=1;i+(1<<j)-1<=n;i++)
{
dm[i][j]=max(dm[i][j-1],dm[i+(1<<j-1)][j-1]);
dn[i][j]=min(dn[i][j-1],dn[i+(1<<j-1)][j-1]);
}
}
int h=1;
ans=-1;
for(i=1;i<=n;i++)
{
while(gm(h,i)-gn(h,i)>m) h++;
ans=max(ans,i-h+1);
}
printf("%d",ans);
return 0;
}