HDU 5266 pog loves szh III

题目：pog loves szh III

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

题意：给一棵树，有m个询问，每个询问问所有下标在[l, r]之中的结点的最近公共祖先。

思路：

　　线段树 + LCA

　　这道题数据有点大，n、m均在30万，所以有点卡内存，做这道题才知道自己平时做题有多浪费内存。。。

　　题目提示小心栈溢出，我AC以后再用普通的dfs试，发现不会出现该问题，但最好还是自己写栈，用while代替递归。

　　题目思路并不难，线段树处理区间，建树时使用lca在线计算即可。

AC代码：

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<math.h>
#include<set>
#include<map>
#include<list>
#include<stack>
#include<queue>
#include<vector>
#include<string>
#include<iostream>
#include<algorithm>
using namespace std;
#define lson rt<<1
#define rson rt<<1|1
#define N 300005
#define M 300005
#define Mod 1000000007
#define LL long long
#define INF 0x7fffffff
vector<int> e[M];
int xu[N*2],xo,pos[M],h[M];

struct SSS
{
  int rt,fa;
  int i;
};

void dfs(int rt,int fa,int ceng)  //普通递归
{
  h[rt]=ceng;
  for(int i=0;i<e[rt].size();i++)
  {
    if(e[rt][i]!=fa) dfs(e[rt][i],rt,ceng+1);
    pos[rt]=xo;
    xu[xo++]=rt;
  }
}

SSS st[M];
void dfs(int rt)  //while型递归
{
  int top=0;
  st[top].rt=rt;
  st[top].fa=-1;
  st[top].i=0;
  h[rt]=1;
  top++;
  while(top)
  {
    SSS a=st[top-1];
    st[top-1].i++;
    if(a.i<e[a.rt].size())
    {
      int ad=e[a.rt][a.i];
      if(ad!=a.fa)
      {
        h[ad]=h[a.rt]+1;
        st[top].rt=ad;
        st[top].fa=a.rt;
        st[top++].i=0;
      }
      if(a.i!=0) xu[xo++]=a.rt;
    }
    else
    {
      pos[a.rt]=xo;
      xu[xo++]=a.rt;
      top--;
    }
  }
}
int f[N<<1][24];
void rmq()
{
  for(int j=0;j<xo;j++) f[j][0]=xu[j];
  for(int i=1;i<24;i++)
  {
    for(int j=0;j<xo;j++)
    {
      if(j+(1<<i)-1 < xo)
      {
        if(h[f[j][i-1]]<h[f[j+(1<<i-1)][i-1]])
        {
          f[j][i]=f[j][i-1];
        }
        else
        {
          f[j][i]=f[j+(1<<i-1)][i-1];
        }
      }
    }
  }
}
int look(int l,int r)
{
  if(l>r) l^=r^=l^=r;
  int k=(int)log2((double)(r-l+1));
  if(h[f[l][k]]<h[f[r-(1<<k)+1][k]]) return f[l][k];
  else return f[r-(1<<k)+1][k];
}

int v[M<<2];
void build(int l,int r,int rt)
{
  if(l==r)
  {
    v[rt]=l;
    return ;
  }
  build(l,(l+r)/2,rt<<1);
  build((l+r)/2+1,r,rt<<1|1);
  v[rt] = look(pos[v[rt<<1]],pos[v[rt<<1|1]]);
}
int query(int ll,int rr,int l,int r,int rt)
{
  if(l==ll&&r==rr) return v[rt];
  int mid=(ll+rr)/2;
  if(l>mid) return query(mid+1,rr,l,r,rt<<1|1);
  else if(r<=mid) return query(ll,mid,l,r,rt<<1);
  else return look(pos[query(ll,mid,l,mid,rt<<1)],pos[query(mid+1,rr,mid+1,r,rt<<1|1)]);
}
int main()
{
  int x,y,n,m;
  while(scanf("%d",&n)!=EOF)
  {
    for(int i=1;i<=n;i++)
    {
      e[i].clear();
      h[i]=INF;
    }
    for(int i=1;i<n;i++)
    {
      scanf("%d%d",&x,&y);
      e[x].push_back(y);
      e[y].push_back(x);
    }
    xo=0;
    dfs(1,-1,1);
    rmq();
    build(1,n,1);
    scanf("%d",&m);
    while(m--)
    {
      scanf("%d%d",&x,&y);
      printf("%d
",query(1,n,x,y,1));
    }
  }
  return 0;
}