hdu 4123 Bob’s Race (树上最长路+RMQ)

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=4123

思路：

首先利用树的一个性质，树上个每点的最长路的另外一个端点一定是树直径的两个端点中的一个，利用这个性质做三遍bfs就能处理出来每个点在树上的最长路。

然后利用RMQ和尺取法扫一遍就能得到答案了~

写得代码1968ms C++ 卡过，不知道为什么会这么慢。。。

code：

#include <cstdio>
#include <cstdlib>
#include <iostream>
#include <algorithm>
#include <cstring>
#include <cmath>
#define INF 1e9
using namespace std;

const int maxn=50500;
const int maxe=150000;
const int maxq=500000;

struct edge
{
    int to,cost,next;
} P[maxe];

int head[maxn],si;
int dismax[maxn],dis0[maxn];
bool vis[maxn];
int que[maxq],he,ta;
int V,E,M;
int mm[600];
int flog[maxn];

void add_edge(int s,int t,int cc)
{
    P[si].to=t;
    P[si].cost=cc;
    P[si].next=head[s];
    head[s]=si++;
}

void spfa(int s,int &ss,int dis[])
{
    fill(dis,dis+V+1,INF);
    fill(vis,vis+V+1,false);
    dis[s]=0;
    he=ta=0;
    que[ta++]=s;
    vis[s]=true;
    ss=s;
    while(he!=ta){
        int v=que[he++];
        for(int i=head[v];i!=-1;i=P[i].next){
            int u=P[i].to;
            if(dis[v]+P[i].cost<dis[u]){
                dis[u]=dis[v]+P[i].cost;
                if(!vis[u]){
                    vis[u]=true;
                    que[ta++]=u;
                }
            }
        }
        if(dis[v]>dis[ss]) ss=v;
        vis[v]=false;
    }
}
int dmax[maxn][30],dmin[maxn][30];
void RMQ_init(int A[])
{
    int n=V;
    for(int i=0;i<n;i++) dmax[i][0]=dmin[i][0]=A[i];
    for(int j=1;(1<<j)<=n;j++){
        for(int i=0;i+(1<<j)-1<n;i++){
            dmax[i][j]=max(dmax[i][j-1],dmax[i+(1<<(j-1))][j-1]);
            dmin[i][j]=min(dmin[i][j-1],dmin[i+(1<<(j-1))][j-1]);
        }
    }
}
int RMQ_max(int L,int R)
{
    int k=0;
    k=flog[R-L+1];
    return max(dmax[L][k],dmax[R-(1<<k)+1][k]);
}
int RMQ_min(int L,int R)
{
    int k=0;
    k=flog[R-L+1];
    return min(dmin[L][k],dmin[R-(1<<k)+1][k]);
}
int cal(int MM)
{
    int ans=0;
    he=ta=0;
    while(ta<V){
        if(RMQ_max(he,ta)-RMQ_min(he,ta)<=MM){
            ans=max(ans,ta-he+1);
            ta++;
        }
        else he++;
    }
    return ans;
}

void solve()
{
    int uu,vv;
    spfa(1,uu,dis0);
    spfa(uu,vv,dis0);
    spfa(vv,uu,dismax);
    for(int i=0;i<V;i++) dismax[i]=max(dismax[i+1],dis0[i+1]);
    RMQ_init(dismax);
    for(int i=0;i<M;i++){
        printf("%d\n",cal(mm[i]));
    }
}

int main()
{
    flog[0]=-1;
    for(int i=1;i<2*maxn;i++) flog[i]=flog[i>>1]+1;

    int ss,tt,cc;
    while(scanf("%d%d",&V,&M),V!=0||M!=0){
        memset(head,-1,sizeof(head));
        si=0;
        for(int i=1;i<V;i++){
            scanf("%d%d%d",&ss,&tt,&cc);
            add_edge(ss,tt,cc);
            add_edge(tt,ss,cc);
        }
        for(int i=0;i<M;i++) scanf("%d",&mm[i]);
        solve();
    }
}

求每个点的最长路径还可以用树形dp的方法，做两遍dfs，代码如下，思路肯定是对的，但是我这样做就一直T，求人指点~

code：

#include <cstdio>
#include <cstdlib>
#include <iostream>
#include <algorithm>
#include <cstring>
#include <cmath>
#define INF 1e9
using namespace std;

const int maxn=50010;
const int maxe=100200;

struct edge
{
    int to,cost,next;
} P[maxe];

int head[maxn],si;
int dismax[maxn],d[maxn][2],id[maxn];
int V,E,M;
int mm[600];
int flog[maxn];

void add_edge(int s,int t,int cc)
{
    P[si].to=t;
    P[si].cost=cc;
    P[si].next=head[s];
    head[s]=si++;
}

void dfs1(int u,int fa)
{
    d[u][0]=d[u][1]=0;
    id[u]=u;
    for(int i=head[u];i!=-1;i=P[i].next){
        int v=P[i].to;
        if(v==fa) continue;
        dfs1(v,u);
        if(d[v][0]+P[i].cost>d[u][0]){
            d[u][1]=d[u][0];
            id[u]=v;    d[u][0]=d[v][0]+P[i].cost;
        }
        else if(d[v][0]+P[i].cost>d[u][1]) d[u][1]=d[v][0]+P[i].cost;
    }
}

void dfs2(int u,int fa,int ll)
{
    dismax[u]=max(ll,d[u][0]);
    for(int i=head[u];i!=-1;i=P[i].next){
        int v=P[i].to;
        if(v==fa) continue;
        if(v==id[u]) dfs2(v,u,max(ll,d[u][1])+P[i].cost);
        else          dfs2(v,u,dismax[u]+P[i].cost);
    }
}

int dmax[maxn][30],dmin[maxn][30];
void RMQ_init(int A[])
{
    int n=V;
    for(int i=0;i<n;i++) dmax[i][0]=dmin[i][0]=A[i+1];
    for(int j=1;j<=flog[n];j++){
        for(int i=0;i+(1<<j)-1<n;i++){
            dmax[i][j]=max(dmax[i][j-1],dmax[i+(1<<(j-1))][j-1]);
            dmin[i][j]=min(dmin[i][j-1],dmin[i+(1<<(j-1))][j-1]);
        }
    }
}
int RMQ_max(int L,int R)
{
    int k=flog[R-L+1];
    return max(dmax[L][k],dmax[R-(1<<k)+1][k]);
}
int RMQ_min(int L,int R)
{
    int k=flog[R-L+1];
    return min(dmin[L][k],dmin[R-(1<<k)+1][k]);
}
int cal(int MM)
{
    int ans,he,ta;
    he=ta=ans=0;
    while(ta<V){
        if(RMQ_max(he,ta)-RMQ_min(he,ta)<=MM){
            ans=max(ans,ta-he+1);
            ta++;
        }
        else he++;
    }
    return ans;
}

int main()
{
    flog[0]=-1;
    for(int i=1;i<2*maxn;i++) flog[i]=flog[i>>1]+1;

    int ss,tt,cc;
    while(scanf("%d%d",&V,&M)){
        if(!V&&!M) break;
        for(int i=0;i<=V;i++) head[i]=-1;
        si=0;
        for(int i=1;i<V;i++){
            scanf("%d%d%d",&ss,&tt,&cc);
            add_edge(ss,tt,cc);
            add_edge(tt,ss,cc);
        }
        for(int i=0;i<M;i++) scanf("%d",&mm[i]);
        dfs1(1,-1);
        dfs2(1,-1,0);
        RMQ_init(dismax);
        for(int i=0;i<M;i++){
            int ans,he,ta;
            he=ta=ans=0;
            while(ta<V){
                if(RMQ_max(he,ta)-RMQ_min(he,ta)<=mm[i]){
                    ans=max(ans,ta-he+1);
                    ta++;
                }
                else he++;
            }
            printf("%d\n",ans);
        }
    }
}