HDU 4123--Bob’s Race（树形dp）

来源：

题意：n个点的树，每个点有个人，每个人会跑到离自己最远的点上。给你m个查询，对每个查询q，找一段连续的人的编号，使其中的最大值和最小最的差不大于q。

#include<iostream>
#include<algorithm>
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <string>
#include <cstring>
#include <map>
#include <queue>
#include <cmath>
#include <stack>
#include <set>
#include <vector>
using namespace std;
struct edge
{
    int x,len,sec;
    edge(int a,int b):x(a),len(b) {};
};
vector<edge> data[50005];
struct node
{
    int id,dis;
} qmax[50005],qmin[50005];
int dp[50005],n,m;
int vis[500005];
void dfs1(int s)
{
    vis[s]=true;
    for(int i=0; i<data[s].size(); i++)
    {
        int x=data[s][i].x;
        int len=data[s][i].len;
        if(!vis[x])
        {
            dfs1(x);
            dp[s]=max(dp[s],dp[x]+len);
            data[s][i].sec=dp[x]+len;
        }
    }
}
void dfs2(int s)
{
    for(int i=0; i<data[s].size(); i++)
    {
        int x=data[s][i].x;
        if(!vis[x])
        {
            vis[x]=true;
            int tmp=0;
            for(int j=0; j<data[s].size(); j++)
            {
                int curx=data[s][j].x;
                if(curx!=x)
                {
                    tmp=max(tmp,data[s][j].sec);
                }
            }
            for(int j=0; j<data[x].size(); j++)
            {
                if(data[x][j].x==s)
                {
                    data[x][j].sec=tmp+data[x][j].len;
                    break;
                }
            }
            for(int j=0; j<data[x].size(); j++)
            {
                dp[x]=max(dp[x],data[x][j].sec);
            }
            dfs2(x);
        }
    }
}
int query(int q)
{
    int maxh=1,minh=1,from=1;
    int maxl=0,minl=0,ans=0;
    node tmp;
    for(int i=1; i<=n; i++)
    {
        tmp.dis=dp[i];
        tmp.id=i;
        while (maxh<=maxl && tmp.dis >= qmax[maxl].dis) maxl--;
        qmax[++maxl] = tmp;
        while (minh<=minl && tmp.dis<= qmin[minl].dis) minl--;
        qmin[++minl] = tmp;
        while (qmax[maxh].dis - qmin[minh].dis > q)
        {
            if (qmax[maxh].id < qmin[minh].id)
            {
                from=qmax[maxh].id+1;
                maxh++;
            }
            else
            {
                from=qmin[minh].id+1;
                minh++;
            }
        }
        if ( qmax[maxh].dis - qmin[minh].dis<=q && qmax[maxh].dis-qmin[minh].dis>=-q)   //满足条件的要求下
        {
            ans=max(ans, i-from+1);
        }
    }
    return ans;
}

int main()
{
    while(cin>>n>>m&&n&&m)
    {
        for(int i=0;i<=n;i++)
        {
            data[i].clear();
            vis[i]=0;
            dp[i]=0;
        }
        for(int i=1; i<n; i++)
        {
            int a,b,c;
            scanf("%d%d%d",&a,&b,&c);
            data[a].push_back(edge(b,c));
            data[b].push_back(edge(a,c));
        }
        memset(vis,0,sizeof(vis));
        dfs1(1);
        memset(vis,0,sizeof(vis));
        dfs2(1);
        while(m--)
        {
            int q;
            scanf("%d",&q);
            cout<<query(q)<<endl;
        }
    }
    return 0;
}