bzoj2500幸福的道路 树形dp+单调队列

2500: 幸福的道路

Time Limit: 20 Sec  Memory Limit: 256 MB
Submit: 434  Solved: 170
[Submit][Status][Discuss]

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

Sample Output

3
数据范围：
50%的数据N<=1000
80%的数据N<=100 000
100%的数据N<=1000 000

 

这其实是两个题强行合在一起啊。。
首先是对于每个节点求它在树上的最长路，可以树形dp 2次，一次从儿子转移一次从父亲转移
求最长连续区间的话维护两个单调队列max min就好

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#define N 1000005
#define ll long long
using namespace std;
int n,m,tot,w[N],hd[N],ans=1;
ll f[N],g[N],a[N];int mx[N],mn[N];
struct edge{int v,w,next;}e[N<<1];
char gc(){
    static char s[1000000],*p1,*p2;
    if(p1==p2)p2=(p1=s)+fread(s,1,1000000,stdin);
    if(p1==p2)return EOF;
    return *p1++; 
}
int read(){
    int x=0;char ch=gc();
    while(ch>'9'||ch<'0')ch=gc();
    while(ch<='9'&&ch>='0')x=x*10+ch-'0',ch=gc();
    return x; 
}
void adde(int u,int v,int w){
    e[++tot].v=v;
    e[tot].next=hd[u];
    e[tot].w=w;
    hd[u]=tot; 
}
void dfs1(int u,int fa){
     for(int i=hd[u];i;i=e[i].next){
         int v=e[i].v;
         if(v==fa)continue;
         dfs1(v,u);
         f[u]=max(f[u],f[v]+e[i].w); 
     }
}
void dfs2(int u,int fa){
    ll mx1=0,mx2=0;
    for(int i=hd[u];i;i=e[i].next){
        int v=e[i].v;if(v==fa)continue;
        if(f[v]+e[i].w>mx1)mx2=mx1,mx1=f[v]+e[i].w;
        else if(f[v]+e[i].w>mx2)mx2=f[v]+e[i].w;
        g[v]=g[u]+e[i].w;
    }
    for(int i=hd[u];i;i=e[i].next){
        int v=e[i].v;if(v==fa)continue;
        if(f[v]+e[i].w==mx1)g[v]=max(g[v],mx2+e[i].w);
        else if(mx1+e[i].w>g[v])g[v]=mx1+e[i].w;
        dfs2(v,u); 
    }
}
void solve(){
    for(register int i=1;i<=n;i++)a[i]=max(g[i],f[i]);
    int l1=1,l2=1,r2=0,r1=0,t=1;
    for(register int i=1;i<=n;i++){
        while(l1<=r1&&a[i]>=a[mx[r1]])r1--;
        while(l2<=r2&&a[i]<=a[mn[r2]])r2--;
        mx[++r1]=i;mn[++r2]=i;
        while(a[mx[l1]]-a[mn[l2]]>m){
            if(mx[l1]<=mn[l2])t=mx[l1]+1,l1++;
            else t=mn[l2]+1,l2++;
        }
        ans=max(ans,i-t+1);
    }
}
int main(){
     n=read();m=read();
     for(register int i=2;i<=n;i++){
         int v=read(),w=read();
         adde(i,v,w);adde(v,i,w);
     }dfs1(1,0);dfs2(1,0);
     solve();printf("%d",ans);
     return 0;
}