hdu4124

对着题解敲了一遍，彻底敲晕了，还敲的不对

#include<bits/stdc++.h>
using namespace std;
#define N 50004
#define M 100004
int n,m,d[N];
int p[N],eid;
struct Edge{
    int u,v,cost,nxt,mlen;
}e[M];
void init(int n){
    for(int i=1;i<=n;i++) { p[i]=-1; d[i]=0; } eid=0;
}
void addedge(int u,int v,int cost){
    e[eid].u=u; e[eid].v=v; e[eid].mlen=-1; e[eid].nxt=p[u]; p[u]=eid++; swap(u,v);
    e[eid].u=u; e[eid].v=v; e[eid].mlen=-1; e[eid].nxt=p[u]; p[u]=eid++;
}
int dfs(int id){
    if(e[id].mlen!=-1) return e[id].mlen;
    int u=e[id].u, v=e[id].v; //e[id].v为叶子
    if(e[ p[v] ].nxt==-1) return e[id].mlen=e[id].cost;
    for(int j=p[v];j!=-1;j=e[j].nxt){
        if(e[j].v=u) continue;
        e[id].mlen = max(e[id].mlen,e[id].cost+dfs(j));
    }return e[id].mlen;
}
int dpmin[N][17],dpmax[N][17],delta[N];//delta用来预处理log值
void initRMQ(int n){
    for(int i=1;i<=n;i++) dpmax[i][0]=d[i];
    for(int j=1;j<=log((double)n)/log(2.0);j++) for(int i=1;i+(1<<j)-1<=n;i++) dpmax[i][j]=max( dpmax[i][j-1], dpmax[i+(1<<(j-1))][j-1] );
    for(int i=1;i<=n;i++) dpmin[i][0]=d[i];
    for(int j=1;j<=log((double)n)/log(2.0);j++) for(int i=1;i+(1<<j)-1<=n;i++) dpmin[i][j]=min( dpmin[i][j-1], dpmin[i+(1<<(j-1))][j-1] );
}
int maxRMQ(int a,int b){
    int k=delta[b-a+1]; return max(dpmax[a][k],dpmax[b-(1<<k)+1][k]);
}
int minRMQ(int a,int b){
    int k=delta[b-a+1]; return min(dpmin[a][k],dpmin[b-(1<<k)+1][k]);
}
int main()
{
    while(scanf("%d%d",&n,&m)!=EOF){
        if(n==0&&m==0) break;
        init(n); for(int i=1,u,v,cost;i<n;i++){
        scanf("%d%d%d",&u,&v,&cost); addedge(i,v,cost);
        }
        for(int i=0;i<eid;i++) if(e[i].mlen==-1) dfs(i);
        for(int i=0;i<eid;i++) d[ e[i].u ]=max(d[ e[i].u ],e[i].mlen);
        initRMQ(n);
        int Q,ans;
        while(m--){
            scanf("%d",&Q); ans=1;
            int l=1,r=1,MAX,MIN;
            while(r<=n){
                MAX=maxRMQ(l,r);
                MIN=minRMQ(l,r);
                if(MAX-MIN>Q) break; r++;
            }
            r--;
            ans=max(ans,r-l+1);
            //以上找到第一个情况
            while(r+1<=n){
                r++;
                while(l<=r){
                    MAX=maxRMQ(l,r);
                    MIN=minRMQ(l,r);
                    if(MAX-MIN>Q) l++;
                    else break;
                }
                ans=max(ans,r-l+1);
            }
            printf("%d\n",ans);
        }
    }
    return 0;
}