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<<];
int dp[maxn][],head[maxn],tot,n,m;
void init(){
    memset(head,-,sizeof head);
    tot=;
}

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

        while(rear2>front2&&d[qmin[rear2-]]>=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])+;
                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=;v<=n;v++){
            cin>>u>>w;
            addedge(u,v,w);
            addedge(v,u,w);
        }

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