HDU 4123 Bob’s Race （树形dp+RMQ+尺取）

stduy_ing

于 2020-05-31 01:53:45 发布

阅读量277

点赞数

本文链接：https://blog.csdn.net/hddddh/article/details/106450421

版权

博客介绍了如何解决HDU 4123问题，即在一棵树上寻找权值（最远节点距离）之差小于k的最长连续区间。通过两次DFS计算节点的权值，使用RMQ（Range Minimum Query）和尺取法找到符合条件的区间。难点在于处理子树内外最大值的更新，涉及到次大值的查找和路径判断。

摘要由CSDN通过智能技术生成

链接： HDU 4123 Bob’s Race

题意：
给一棵树，边有权值，一个节点的权值定义为该点到树中距离该点最远的点的距离。求一个最长的连续区间，区间中点的最大权值减去最小权值小于k .
思路：
这题可以分为两部分，第一部分是求距离每个点最远的点到该点的距离。第二部分就是 rmq +尺取求最长的合法区间了。

求点的权值，可以用两个 dfs,一次求子树内的最大值，一次求子树外的最大值。子树内的最大值很好求，自下而上，依次更新最大值就好了。
关键在于求子树外的最大值，对于每个节点子树外的最大值，可以由父节点的最大值得到。该节点子树外的最大值，就是父节点子树内的最大值和子树外的最大值中较大的加上边的权值用 ans [u] 记录子树外的最大值，ma[u] 记录子树内最大值的话就是ans[u]=max(ans[pre]+v,ma[pre]+v)
那么很容易想到一个矛盾的地方，要是父节点子树内的最大值路径刚好经过的是当前节点，这样显然不能直接用最大值了，所以我们必须重新找一个最大值，也就是次大值，用次大值来更新。所以我们要在第一个dfs记录次大值maa。ans[u]=max(ans[pre]+v,maa[pre]+v);
那么还有一个问题，怎么确定用最大值更新还是次大值，对于每一个节点我们可以用 vis 记录最大值路径走的是哪个子节点。在下面判断当前节点是不是记录的那个节点，是的话就用次大值。

if(vis[pre]==u) ans[u]=max(ans[pre]+v,maa[pre]+v);
if(vis[pre]!=u) ans[u]=max(ans[pre]+v,ma[pre]+v);

最后比较子树内和子树外的最大值，就能得到每个点的权值了。
接下来就和树没有关系了，用rmq 记录区间最值，然后尺取求最大合法区间就好了。（用二分试了试 t了）。
代码：

#include<iostream>
#include<cstdio>
#include<map>
#include<math.h>
#include<queue>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
const int maxn=5e4+7;
struct node{
       int to,next,w;
}e[maxn<<1];
int num,head[maxn],n,q,ma[maxn],maa[maxn],ans[maxn],vis[maxn];
int dpma[maxn][32],dpmi[maxn][32];
void add(int u,int v,int w){
    e[num].next=head[u];
    e[num].to=v;
    e[num].w=w;
    head[u]=num++;
}
void dfs1(int u,int pre){
     ma[u]=0,maa[u]=0;
     for(int i=head[u];i!=-1;i=e[i].next){
         int to=e[i].to,w=e[i].w;
         if(to==pre) continue;
         dfs1(to,u);
         if(ma[to]+w>=ma[u]){    //记录最大和次大
            vis[u]=to;
            maa[u]=ma[u];
            ma[u]=ma[to]+w;
         }
         if(ma[to]+w<ma[u]&&ma[to]+w>maa[u]){
            maa[u]=ma[to]+w;
         }
     }
}
void dfs2(int u,int pre,int v){   //这个dfs自上到下，因为要用到更新后的父节点的信息
     if(u==1) ans[u]=0;
     else{               //判断用最大还是次大
        if(vis[pre]==u) ans[u]=max(ans[pre]+v,maa[pre]+v);
        if(vis[pre]!=u) ans[u]=max(ans[pre]+v,ma[pre]+v); 
     }
     for(int i=head[u];i!=-1;i=e[i].next){
        int to=e[i].to,w=e[i].w;
        if(to==pre) continue;
        dfs2(to,u,w);
    }
}
int qa(int l, int r){
    int k=0;
    while ((1 << (k + 1)) <= (r - l + 1)) k++;
    return max(dpma[l][k],dpma[r-(1<<k)+1][k]);
}
int qi(int l,int r){
    int k=0;
    while ((1 << (k + 1)) <= (r - l + 1)) k++;
    return min(dpmi[l][k], dpmi[r - (1 << k) + 1][k]);
}
void rmq(){
    for(int i=1;i<=n;i++){
        dpma[i][0]=ans[i],dpmi[i][0]=ans[i];
    }
    for(int j=1;(1<<j)<=n;j++){
        for(int i=1;i+(1<<j)-1<=n;i++){
            dpma[i][j]=max(dpma[i][j-1],dpma[i+(1<<(j-1))][j-1]);
            dpmi[i][j]=min(dpmi[i][j-1],dpmi[i+(1<<(j-1))][j-1]);
        }
    }
}

int main(){
   int u,v,w;
   while(scanf("%d%d",&n,&q)!=EOF){
   if(n==0&&q==0) break;
   for(int i=1;i<=n;i++) head[i]=-1;num=0;
   for(int i=0;i<n-1;i++){
       scanf("%d%d%d",&u,&v,&w);
       add(u,v,w);
       add(v,u,w);
   }
   dfs1(1,-1);dfs2(1,-1,0);
   for(int i=1;i<=n;i++) ans[i]=max(ans[i],ma[i]);
   rmq();
   while(q--){
        int k;scanf("%d",&k);
        int l=1,r=1,mid,anss=0;
        while(r<=n){
            while(r<=n&&qa(l,r)-qi(l,r)<=k) r++;
            anss=max(anss,r-l);
            l++;
        }
        printf ("%d\n",anss);
    }
   }
}

/*
7 5
1 2 3
2 3 4
4 5 3
3 4 2
2 7 10
4 6 5
*/