hdu4123-树的直径+ST算法记忆化-Bob’s Race

http://acm.hdu.edu.cn/showproblem.php?pid=4123
给定一个树，问你i点到其他点的最远距离为num[i]。
问你num[i] 之间绝对值小于k的 连续区间是多少。。
思路：看懂题就好办了，先求一下所有点到其他点的最短距离（用树形dp求树的直径，这样比用dfs求方便一点）。然后把距离打到一个st算法里。
尺取法是看了别人的qwq。自己没写好，还有就是st算法，存一下那个query用的k，不然会TLE。可能计算log比较用时间吧qwq

#include <iostream>
#include <cstdio>
#include <bits/stdc++.h>
using namespace std;
/* 
*/
const int maxn=6e5;
const int MAXN=1e6;
int smax_siz[maxn];//次长的距离
int max_siz[maxn];// 最长的距离叶子的距离
int vec1[maxn];//最长的距离叶子的距离的方向（总共有两种，不是爸爸就是儿子）
int vec2[maxn];
int FMin[MAXN][20],FMax[MAXN][20];
int m;
int mm[MAXN];
void Init(){
    int i,j;
    mm[0]=-1;
    for(i=1;i<=m;i++){
        FMin[i][0]=FMax[i][0]=max_siz[i];//以i为起点，长度为1的区间，初始值为a[i]；
         mm[i] = (i&(i-1))==0?mm[i-1]+1:mm[i-1];
    }
    for(i=1;(1<<i)<=m;i++){   //区间慢慢扩增
        for(j=1;j+(1<<i)-1<=m;j++){   //区间起点，
            FMin[j][i]=min(FMin[j][i-1],FMin[j+(1<<(i-1))][i-1]);
            FMax[j][i]=max(FMax[j][i-1],FMax[j+(1<<(i-1))][i-1]);
        }
    }
}
int Query_min(int l,int r){
    //int k=(int)(log(r-l+1)/log(2));
    int k=mm[r-l+1];
    return min(FMin[l][k],FMin[r-(1<<k)+1][k]);//0(1)的查询
}
int Query_max(int l,int r){
    //int k=(int)(log(r-l+1)/log(2));
    int k=mm[r-l+1];//把k存一下。
    return max(FMax[l][k],FMax[r-(1<<k)+1][k]);//0(1)的查询
}
struct Node
{   int to,next,w;
}node[maxn];
int len;
int head[maxn];
void add(int u,int v,int w){
     node[len].to=v;
     node[len].next=head[u];
     node[len].w=w;
     head[u]=len++;
}
int dfs1(int u,int pre){
    max_siz[u]=0;
    smax_siz[u]=0;
    for(int i=head[u];i!=-1;i=node[i].next){
        int to=node[i].to;
        int w=node[i].w;

        if(to==pre) continue;
        dfs1(to,u);
        if(max_siz[to]+w>smax_siz[u]){
            smax_siz[u]=max_siz[to]+w;
            vec2[u]=to;
            if(smax_siz[u]>max_siz[u]){
               swap(vec2[u],vec1[u]);//叶子节点方向
               swap(smax_siz[u],max_siz[u]);
            }
        }
    }
    //cout<<u<<" "<<max_siz[u]<<" "<<max_siz[u]<<endl;
return 0;
}
int dfs2(int u,int pre){
     for(int i=head[u];i!=-1;i=node[i].next){
         int to=node[i].to;
         int w=node[i].w;
         if(to==pre) continue;
         if(vec1[u]==to){
             if(smax_siz[u]+w>smax_siz[to]){
                smax_siz[to]=smax_siz[u]+w;
                 vec2[to]=u;
                 if(smax_siz[to]>max_siz[to]){
                    swap(smax_siz[to],max_siz[to]);
                    swap(vec1[to],vec2[to]);
                }
             }
         }
         else{
              if(max_siz[u]+w>smax_siz[to]){
                  smax_siz[to]=max_siz[u]+w;
                  vec2[to]=u;
                 if(smax_siz[to]>max_siz[to]){
                     swap(smax_siz[to],max_siz[to]);
                     swap(vec1[to],vec2[to]);
                 }
              }
         }
      dfs2(to,u);
     }
}
void init(){
    len=0;
    memset(head,-1,sizeof(head));
}
int lim;
bool solve(int l,int r,int nume){
      int a,b;
      a=Query_max(l,r);
     b=Query_min(l,r);
     if(a-b>nume)
        return true;
     return false;

}
int main()
{   int t,a,b,n,c;
    while(~scanf("%d%d",&m,&n)){
         if(!m&&!n) break;
         init();
         for(int i=1;i<=m-1;i++){
             scanf("%d%d%d",&a,&b,&c);
             add(a,b,c);
             add(b,a,c);
         }
         dfs1(1,-1);
         dfs2(1,-1);
         Init();
         /*for(int i=1;i<=m;i++){
            printf("%d\n",max_siz[i]);
         }*/
         int aaa,bbb;
         //cout<<Query_max(1,m)<<endl;
         for(int i=1;i<=n;i++){
              scanf("%d",&lim);
              int l=1;int r=1;
              int ans=-1;
              for(int i=1;i<=m;i++,r++){
                  while(solve(l,r,lim)&&l<=r)
                     l++;
                  ans=max(ans,r-l+1);
              }
             printf("%d\n",ans);
         }
    }
    return 0;
}