POJ 3162 Walking Race 树的直径+单调队列（其实暴力也可以>_<）

最新推荐文章于 2019-08-28 11:55:00 发布

NeighThorn

最新推荐文章于 2019-08-28 11:55:00 发布

阅读量309

点赞数 1

分类专栏： poj题解

本文链接：https://blog.csdn.net/neighthorn/article/details/52102207

版权

poj题解专栏收录该内容

94 篇文章 0 订阅

订阅专栏

传送门

题目大意：
给出一棵树，找到每个节点在树上最长延伸的距离，组成d[i]数组，找到d[i]数组中最长的区间，这个区间中最大值和最小值的差小于等于m

分析：
找d[i]数组，这个很简单，就求一下树的直径，然后i到直径两个端点的距离的较大值就是d[i]…怎么求树的直径？？请见http://blog.csdn.net/neighthorn/article/details/51628711
内含详细讲解及代码（顺便赠送一道练手题O(∩_∩)O哈哈~）
接下来有了d[i]数组，n<=100W，>_<，这肯定不能用暴力，应该用单调队列，维护最大最小值（然而本人数据结构很水Ｏ(≧口≦)Ｏ，单调队列WA了4遍还没过，一气之下写了暴力，结果RP++，数据很水，居然3000ms+就过了）

代码如下：
（这是暴力的代码，等我研究完单调队列怎么写再贴正解>_<）

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
const int maxn=1000000+5;
int n,m,hd[maxn],to[maxn*2],nxt[maxn*2],w[maxn*2],cnt,dis[2][maxn],d[maxn],lala,ans;
inline void add(int x,int y,int s){
    w[cnt]=s;
    to[cnt]=y;
    nxt[cnt]=hd[x];
    hd[x]=cnt++;
}
inline void dfs(int root,int fa,int k){
    for(int i=hd[root];i!=-1;i=nxt[i])
        if(to[i]!=fa)
            dis[k][to[i]]=dis[k][root]+w[i],dfs(to[i],root,k);
}
signed main(void){
    while(scanf("%d%d",&n,&m)!=EOF){
        ans=cnt=0,memset(hd,-1,sizeof(hd));
        for(int i=1,x,y;i<n;i++)
            scanf("%d%d",&x,&y),add(i+1,x,y),add(x,i+1,y);
        dis[0][1]=0,dfs(1,-1,0),lala=1;
        for(int i=2;i<=n;i++)
            lala=dis[0][i]>dis[0][lala]?i:lala;
        dis[0][lala]=0,dfs(lala,-1,0);
        for(int i=1;i<=n;i++)
            lala=dis[0][i]>dis[0][lala]?i:lala;
        dis[1][lala]=0,dfs(lala,-1,1);
        for(int i=1;i<=n;i++)
            d[i]=max(dis[0][i],dis[1][i]);
        int i,j,mx,mi;
        for(i=1;i<=n;i++){
            mx=0,mi=0x3f3f3f3f;
            for(j=i;j<=n;j++){
                mx=max(mx,d[j]),mi=min(mi,d[j]);
                if(mx-mi>m)
                    break;  
            }
            ans=max(ans,j-i);
        }
        cout<<ans<<endl;
    }
    return 0;
}