poj3162 树形dp|树的直径 + 双单调队列|线段树，好题啊

题解链接：https://blog.csdn.net/shiqi_614/article/details/8105149

 用树形dp是超时的，，

/*
先求出每个点可以跑的最长距离dp[i][0|1]表示向下|上可以跑到最大距离 
再用单调队列求最大连续天数 
*/
#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
#define maxn 1000005
#define ll long long
struct Edge{int to,nxt,w;}edge[maxn<<1];
int dp[maxn][2],head[maxn],tot,n,m;
void init(){
    memset(head,-1,sizeof head);
    tot=0;
}

void addedge(int u,int v,int w){
    edge[tot].to=v;edge[tot].w=w;
    edge[tot].nxt=head[u];head[u]=tot++;
}
void dfs0(int u,int pre){
    dp[u][0]=0;
    for(int i=head[u];i!=-1;i=edge[i].nxt){
        int v=edge[i].to;
        if(v!=pre){
            dfs0(v,u);
            dp[u][0]=max(dp[u][0],dp[v][0]+edge[i].w);
        } 
    }
}
void dfs1(int u,int pre){
    long long Max=0,Sec=0,v1,v2;//u的大儿子下标，二儿子下标 
    for(int i=head[u];i!=-1;i=edge[i].nxt){//这个循环处理出u的大儿子 
        int v=edge[i].to;
        if(v==pre)continue;
        int tmp=edge[i].w+dp[v][0];
        if(tmp>Max){//找到了更大的儿子树 
            Sec=Max,Max=tmp;
            v2=v1,v1=v;
        }
        else if(tmp==Max || tmp>Sec)
            Sec=tmp,v2=v;    
    } 
//printf("%d %d\n",u,Max);
    if(u!=-1){//和u的上面进行比较 
        long long tmp=dp[u][1],v=-1;
        if(tmp>Max){//找到了更大的儿子树 
            Sec=Max,Max=tmp;
            v2=v1,v1=v;
        }
        else if(tmp==Max || tmp>Sec)
            Sec=tmp,v2=v;    
    }
    
    for(int i=head[u];i!=-1;i=edge[i].nxt){//这个循环求dp进行递归 
        int v=edge[i].to;
        if(v==pre)continue;
        if(v==v1)dp[v][1]=Sec+edge[i].w;
        else dp[v][1]=Max+edge[i].w;
        dfs1(v,u);
    } 
//printf("%d %d\n",u,Max);
}
int d[maxn],qmax[maxn],qmin[maxn];
void solve()
{
    int ans=0,i,j,front1,front2,rear1,rear2;
    front1=rear1=0;
    front2=rear2=0;
     for(int i=1;i<=n;i++)d[i]=max(dp[i][0],dp[i][1]);
    for(i=1,j=1;j<=n;j++)
    {
        while(rear1>front1&&d[qmax[rear1-1]]<=d[j]) rear1--;
        qmax[rear1++]=j;
 
        while(rear2>front2&&d[qmin[rear2-1]]>=d[j]) rear2--;
        qmin[rear2++]=j;
 
        if(d[qmax[front1]]-d[qmin[front2]]>m)
        {
            ans=max(ans,j-i);
            while(d[qmax[front1]]-d[qmin[front2]]>m)
            {
                i=min(qmax[front1],qmin[front2])+1;
                while(rear1>front1&&qmax[front1]<i) front1++;
                while(rear2>front2&&qmin[front2]<i) front2++;
            }
        }
    }
    ans=max(ans,j-i);
    printf("%d\n",ans);
}

int main(){
    while(cin>>n>>m){
        init();
        int u,v,w;
        for(int v=2;v<=n;v++){
            cin>>u>>w;
            addedge(u,v,w);
            addedge(v,u,w);
        }

        dfs0(1,0);//先求dp[i][0]
        dfs1(1,0);//再求dp[i][1]
        solve(); 
    }
}