hdu_4718_The LCIS on the Tree(树链剖分+线段树合并)

题目连接:http://acm.hdu.edu.cn/showproblem.php?pid=4718

题意:给你一棵树,每个节点有一个值,然后任给树上的两点,问这两点的最长连续递增区间是多少

题解:先树链剖分,然后结合线段树的区间合并来搞,注意的是要记录递增和递减两个状态,因为线段树的区间都是从根到子节点,如果询问从子节点到子节点,那么就是一增一减

  1 #include<cstdio>
  2 #include<algorithm>
  3 #define F(i,a,b) for(int i=a;i<=b;i++)
  4 #define root 1,n,1
  5 #define ls l,m,rt<<1
  6 #define rs m+1,r,rt<<1|1
  7 using namespace std;
  8 const int N=1e5+7;
  9 int t,x,y,n,ic=1,q,cnt,g[N],ed,nxt[N],v[N],a[N];
 10 inline void adg(int x,int y){v[++ed]=y,nxt[ed]=g[x],g[x]=ed;}
 11 //树链剖分----------------------------
 12 int dep[N],sz[N],fa[N],hs[N],tid[N],top[N],idx,cot[N];
 13 void dfs1(int u,int pre){
 14     dep[u]=dep[pre]+1,sz[u]=1,fa[u]=pre,hs[u]=0;
 15     for(int i=g[u];~i;i=nxt[i]){
 16         dfs1(v[i],u),sz[u]+=sz[v[i]];
 17         if(sz[v[i]]>sz[hs[u]])hs[u]=v[i];
 18     }
 19 }
 20 void dfs2(int u,int tp){
 21     top[u]=tp,tid[u]=++idx,cot[tid[u]]=a[u];
 22     if(hs[u])dfs2(hs[u],tp);
 23     for(int i=g[u];~i;i=nxt[i])if(v[i]!=hs[u])dfs2(v[i],v[i]);
 24 }
 25 //线段树-------------------------------
 26 int Rn[N<<3],Ln[N<<3],len[N<<3],ml[N<<3],Rl[N<<3];
 27 int Ll[N<<3],dml[N<<3],dRl[N<<3],dLl[N<<3];
 28 
 29 inline void up(int rt,int li,int ri){
 30     Ln[rt]=Ln[li],Rn[rt]=Rn[ri];
 31     Ll[rt]=Ll[li],Rl[rt]=Rl[ri];
 32     dLl[rt]=dLl[li],dRl[rt]=dRl[ri];
 33     ml[rt]=max(ml[li],ml[ri]);
 34     dml[rt]=max(dml[li],dml[ri]);
 35     if(Rn[li]<Ln[ri]){//左右区间可以合并
 36         ml[rt]=max(ml[rt],Rl[li]+Ll[ri]);
 37         if(Rl[li]==len[li])Ll[rt]=Ll[li]+Ll[ri];
 38         if(Rl[ri]==len[ri])Rl[rt]=Rl[li]+Rl[ri];
 39     }
 40     if(Rn[li]>Ln[ri]){
 41         dml[rt]=max(dml[rt],dRl[li]+dLl[ri]);
 42         if(dRl[li]==len[li])dLl[rt]=dLl[li]+dLl[ri];
 43         if(dRl[ri]==len[ri])dRl[rt]=dRl[li]+dRl[ri];
 44     }
 45 }
 46 
 47 void build(int l,int r,int rt){
 48     len[rt]=r-l+1;
 49     if(l==r){
 50         Ln[rt]=Rn[rt]=cot[l],ml[rt]=Ll[rt]=Rl[rt]=1;
 51         dml[rt]=dLl[rt]=dRl[rt]=1;
 52         return;
 53     }
 54     int m=(l+r)>>1;
 55     build(ls),build(rs),up(rt,rt<<1,rt<<1|1);
 56 }
 57 //将两个区间合并
 58 inline void adt(int li,int ri,int rt){len[rt]=len[li]+len[ri],up(rt,li,ri);}
 59 
 60 int anson(int l,int r){
 61     int ans=max(dml[l],ml[r]);
 62     if(Ln[l]<Ln[r])return max(ans,dLl[l]+Ll[r]);
 63     return ans;
 64 }
 65 
 66 int query(int L,int R,int l,int r,int rt){
 67     if(L==l&&R==r)return rt;
 68     int m=(l+r)>>1;
 69     if(m>=R)return query(L,R,ls);
 70     else if(m<L)return query(L,R,rs);
 71     else{
 72         int lss=query(L,m,ls),rss=query(m+1,R,rs);
 73         adt(lss,rss,++cnt);
 74         return cnt;
 75     }
 76 }
 77 
 78 int lca(int x,int y){
 79     if(x==y)return 1;
 80     cnt=N<<2;
 81     int xp=-1,yp=-1,op;
 82     while(top[x]!=top[y]){
 83         if(dep[top[x]]>dep[top[y]]){
 84             op=query(tid[top[x]],tid[x],root),x=fa[top[x]];
 85             if(xp==-1)xp=op;
 86             else adt(op,xp,++cnt),xp=cnt;
 87         }else{
 88             op=query(tid[top[y]],tid[y],root),y=fa[top[y]];
 89             if(yp==-1)yp=op;
 90             else adt(op,yp,++cnt),yp=cnt;
 91         }
 92     }
 93     if(dep[x]>=dep[y]){
 94         op=query(tid[y],tid[x],root);
 95         if(xp==-1)xp=op;
 96         else adt(op,xp,++cnt),xp=cnt;
 97     }else{
 98         op=query(tid[x],tid[y],root);
 99         if(yp==-1)yp=op;
100         else adt(op,yp,++cnt),yp=cnt;
101     }
102     if(xp==-1)return ml[yp];
103     if(yp==-1)return dml[xp];
104     return anson(xp,yp);
105 }
106 
107 int main(){
108     scanf("%d",&t);
109     while(t--){
110         scanf("%d",&n);
111         F(i,0,N-1)g[i]=-1;ed=0;
112         F(i,1,n)scanf("%d",a+i);
113         F(i,2,n)scanf("%d",&x),adg(x,i);
114         dfs1(1,0),idx=0,dfs2(1,1),build(root);
115         scanf("%d",&q);
116         printf("Case #%d:\n",ic++);
117         while(q--)scanf("%d%d",&x,&y),printf("%d\n",lca(x,y));
118         if(t)puts("");
119     }
120 }
View Code

 

posted @ 2016-06-30 17:31  bin_gege  阅读(233)  评论(0编辑  收藏  举报