hdu 4123 Bob’s Race

题解：

三个模板的叠加：

求树中一个点的最远距离（DP）+最值（RMQ）+单调性

这个题值得注意的地方是：log函数很慢，需要预处理。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<cmath>
#include<cstdlib>
#include<algorithm>
using namespace std;
int n,m;
const int maxn=51000;
int max1[maxn],max2[maxn],rec1[maxn],rec2[maxn];
int b[maxn];
int f[maxn];
int rmq_max[maxn][20],rmq_min[maxn][20];
int q[maxn];
int l[maxn],r[maxn];
int flog[maxn];
bool v[maxn];
struct node
{
    int u,next,d;
}a[maxn*10];
int head[maxn];

void dp(int x)
{
    if(v[x])return;
    v[x]=true;
    bool flag=false;
    //max1
    for(int i=head[x];i!=-1;i=a[i].next)
    if(!v[a[i].u])
    {
        flag=true;
        dp(a[i].u);
        if(max1[a[i].u]+a[i].d>=max1[x])
        {
            max2[x]=max1[x];
            rec2[x]=rec1[x];

            max1[x]=max1[a[i].u]+a[i].d;
            rec1[x]=a[i].u;
        }
        else if(max1[a[i].u]+a[i].d>max2[x])
        {
            max2[x]=max1[a[i].u]+a[i].d;
            rec2[x]=a[i].u;
        }
    }
    if(!flag)max1[x]=0;
}
void dfs(int x,int father,int d)
{
    v[x]=true;
    if(x==1)f[x]=0;
    else
    {
        if(rec1[father]!=x&&max1[father]!=-1)
        f[x]=max1[father]+d;
        else if(rec2[father]!=x&&max2[father]!=-1)
        f[x]=max2[father]+d;

        f[x]=max(f[x],f[father]+d);
    }

    for(int i=head[x];i!=-1;i=a[i].next)
    if(!v[a[i].u])
    {
        dfs(a[i].u,x,a[i].d);
    }

}
inline int getmax(int l,int r)
{
    int x=flog[r-l+1];
    return max(rmq_max[l][x],rmq_max[r+1-(1<<x)][x]);
}
inline int getmin(int l,int r)
{
    int x=flog[r-l+1];
    return min(rmq_min[l][x],rmq_min[r+1-(1<<x)][x]);
}

int main()
{
    for(int i=1;i<=maxn-1;i++)
    flog[i]=log2(i);
    while(1)
    {
        scanf("%d%d",&n,&m);if(n==0 && m==0)break;
        for(int i=1;i<=n;i++)
        head[i]=-1;
        int num=0;
        for(int i=1;i<=n-1;i++)
        {
            int x,y,z;
            scanf("%d%d%d",&x,&y,&z);
            num++;a[num].u=y;a[num].d=z;a[num].next=head[x];head[x]=num;
            num++;a[num].u=x;a[num].d=z;a[num].next=head[y];head[y]=num;
        }
        //pre
        memset(rec1,0,sizeof(rec1));
        memset(rec2,0,sizeof(rec2));
        memset(max1,-1,sizeof(max1));
        memset(max2,-1,sizeof(max2));
        memset(v,false,sizeof(v));
        dp(1);
        memset(f,-1,sizeof(f));
        memset(v,false,sizeof(v));
        dfs(1,0,0);

        for(int i=1;i<=n;i++)
        b[i]=max(max1[i],f[i]);
        //pre
        for(int i=1;i<=n;i++)
        {
            rmq_max[i][0]=b[i];
            rmq_min[i][0]=b[i];
        }
        for(int j=1;j<=flog[n];j++)
        {
            for(int i=1;i<=n+1-(1<<j);i++)
            {
                rmq_max[i][j]=max(rmq_max[i][j-1],rmq_max[i+(1<<(j-1))][j-1]);
                rmq_min[i][j]=min(rmq_min[i][j-1],rmq_min[i+(1<<(j-1))][j-1]);
            }
        }
        //

        for(int i=1;i<=m;i++)
        {
            int lim;
            scanf("%d",&lim);
            int l=1,r=1,ans=0;
            while(r<=n)
            {
                int mx=getmax(l,r);
                int mi=getmin(l,r);
                if(mx-mi<=lim)
                {
                    ans=max(ans,r-l+1);
                    r++;
                }
                else
                    l++;
            }
            printf("%d\n",ans);
        }
        //do
    }
    return 0;
}