POJ 3162Walking Race( 树形dp+线段树处理区间）

吐槽一下这题题意真是看了半天，，作者英文水平你妈嗨你妈嗨

题意最后是：一棵n个节点的树。wc爱跑步，跑n天，第i天从第i个节点开始跑步，每次跑到距第i个节点最远的那个节点（产生了n个距离），现在要在这n个距离里取连续的若干天，使得这些天里最大距离和最小距离的差小于M，问怎么取使得天数最多？

发现这题和Hdu 2196 Computer有相同的部分，先求一下树上每个点到其它点的最远距离，存在一个一维数组里。问题就转化为：怎么确定一个最大的符合要求的区间。即区间里 最大值 - 最小值 < M。

用很基础的线段树可以很容易的查询一个区间的最大最小值。再维护两个指针i，j；我一开始想的是i=1，j=n；然后i、j向中间靠，一旦找一个符合的就是最大的，但是发现反而不知道怎么判断移动i，j。

最后看了题解，直接让i=j=1；一起往右移就是了。

总的来说这题不难，没什么新意

【代码】

/* ***********************************************
Author        :angon

************************************************ */
#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
#include <stack>
#include <vector>
#include <queue>
#include <set>
#include <map>
#include <string>
#include <math.h>
#include <stdlib.h>
#include <time.h>
using namespace std;
#define showtime fprintf(stderr,"time = %.15f\n",clock() / (double)CLOCKS_PER_SEC)
#define lld %I64d
#define REP(i,k,n) for(int i=k;i<n;i++)
#define REPP(i,k,n) for(int i=k;i<=n;i++)
#define scan(d) scanf("%d",&d)
#define scanl(d) scanf("%I64d",&d)
#define scann(n,m) scanf("%d%d",&n,&m)
#define scannl(n,m) scanf("%I64d%I64d",&n,&m)
#define mst(a,k)  memset(a,k,sizeof(a))
#define LL long long
#define N 1000005
#define mod 1000000007
inline int read(){int s=0;char ch=getchar();for(; ch<'0'||ch>'9'; ch=getchar());for(; ch>='0'&&ch<='9'; ch=getchar())s=s*10+ch-'0';return s;}


struct Edge
{
    int v,w,next;
}edge[N*2];
int head[N],tot;
void addedge(int u,int v,int w)
{
    edge[tot].v=v;
    edge[tot].w=w;
    edge[tot].next=head[u];
    head[u]=tot++;
}
int dp[N][3],dis[N];
/*
dp[u][0]表示在u的子树下u的最远距离是多少
dp[u][1]表示在u的子树下（和dp[u][0]不是同一孩子）u的次远距离是多少
dp[u][2]表示通过u的父亲能走的最远距离是多少
*/
void dfs1(int u,int v) //每个节点子树下的最大和次大
{
    for(int i=head[u];i!=-1;i=edge[i].next)
    {
        int v=edge[i].v;
        dfs1(v,u);
        int w=edge[i].w;
        int temp = dp[v][0] + w;
        if(temp >= dp[u][0])
        {
            dp[u][1]=dp[u][0];
            dp[u][0] = temp;
        }
        else if(temp > dp[u][1])
            dp[u][1] = temp;
    }
}

void dfs2(int u,int v)  //通过父亲最远距离
{
    for(int i=head[u];i!=-1;i=edge[i].next)
    {
        int v=edge[i].v;
        if(dp[u][0] == dp[v][0] + edge[i].w)
        {
            dp[v][2] = max(dp[u][2],dp[u][1]) + edge[i].w;
        }
        else
        {
            dp[v][2] = max(dp[u][2],dp[u][0]) + edge[i].w;
        }
        dfs2(v,u);
    }
}
int n,m,mi,ma;
struct node
{
    int l,r;
    int minc,maxc;
}seg[N*4];

void build(int i,int l,int r)
{
    seg[i].l=l;
    seg[i].r=r;
    if(l==r)
    {
        seg[i].maxc=seg[i].minc=dis[l];
        return;
    }
    int mid = (l+r)>>1;
    build(i<<1,l,mid);
    build(i<<1|1,mid+1,r);
    seg[i].maxc = max(seg[i<<1].maxc,seg[i<<1|1].maxc);
    seg[i].minc = min(seg[i<<1].minc,seg[i<<1|1].minc);
}
void query(int i,int l,int r)
{
   // printf("%d %d %d\n",i,l,r);
    if(l<=seg[i].l && r>=seg[i].r)
    {
        ma = max(seg[i].maxc,ma);
        mi = min(seg[i].minc,mi);
        return;
    }
    int mid = (seg[i].l+seg[i].r)>>1;
    if(l<=mid) query(i<<1,l,r);
    if(r>mid)  query(i<<1|1,l,r);
}


int main()
{
    //freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
    while(~scann(n,m))
    {
        mst(head,-1); tot=0;
        for(int v=2;v<=n;v++)
        {
            int u,w; scann(u,w);
            addedge(u,v,w);
        }
        mst(dp,0);
        dfs1(1,-1);
        dfs2(1,-1);
        for(int i=1;i<=n;i++)
            dis[i] = max(dp[i][0],dp[i][2]);//,printf("%d\n",dis[i]);
        build(1,1,n);
        int i=1,j=2;
        int ans=1;
        while(i<=j && j<=n)
        {
            ma=0;   mi=1<<30;
            query(1,i,j);
            if(ma-mi<=m)
            {
                ans = max(ans,j-i+1);
                j++;
            }
            else
                i++;
            if(n-i < ans) break;
        }
        printf("%d\n",ans);
    }
    return 0;
}