題意:給你n個整數,有兩種操作,U A B把第A個數變成B,Q A B查詢區間[A,B]的最長連續上升序列。
線上段樹的節點裡保留了端點的值即lval,rval。同時維護左端點最長的上長序列的長度lmx,右端點的最長的下降序列的長度rmx,以及這個區間的最長上升序列的長度mx。在合并區間的時候,如果當前區間的左兒子的lmx等於左兒子的區間長度,那麼當前區間的lmx還要加上右兒子的lmx。維護當前區間的rmx也是類似的,當前區間的mx除了在左右兒子的區間中的mx取一個最大值外,還要注意,如果左兒子的右端點小於右兒子的左端點那麼當前區間的mx還有可能是左兒子的rmx加上右兒子的lmx。
在查詢的時候,要注意,如果查詢的範圍是在當前區間的左右兒子裡,並且左兒子的右端點小於右兒子的左端點,那麼不能直接取左兒子的rmx加上右兒子的lmx,因為有可能查詢的範圍的長度就已經小於左兒子的rmx加上右兒子的lmx,所以要加上一個判斷。
(代碼風格更新前,沒有線上段樹節點裡儲存端點的值)之前做得比較複雜,在節點裡還儲存了這個區間端點的值,然後在合并的時候去比較,後來在網上看到一種作法,更新到葉子結點的時候直接對數組y進行操作。合并的時候,直接對數組y裡的元素進行比較。
/*代碼風格更新後*/#include <iostream>#include <cstdio>#include <cstring>using namespace std;#define LL(x) (x<<1)#define RR(x) (x<<1|1)#define MID(a,b) (a+((b-a)>>1))const int N=1e5+5;struct node{int lft,rht;int lval,rval;int lmx,rmx,mx;int mid(){return MID(lft,rht);}int len(){return rht-lft+1;}};int y[N],n,m;struct Segtree{node tree[N*4];void up(int ind){tree[ind].lmx=tree[LL(ind)].lmx;tree[ind].rmx=tree[RR(ind)].rmx;tree[ind].lval=tree[LL(ind)].lval;tree[ind].rval=tree[RR(ind)].rval;tree[ind].mx=max(tree[LL(ind)].mx,tree[RR(ind)].mx);if(tree[LL(ind)].rval<tree[RR(ind)].lval){if(tree[LL(ind)].len()==tree[LL(ind)].lmx)tree[ind].lmx+=tree[RR(ind)].lmx;if(tree[RR(ind)].len()==tree[RR(ind)].rmx)tree[ind].rmx+=tree[LL(ind)].rmx;tree[ind].mx=max(tree[ind].mx,tree[LL(ind)].rmx+tree[RR(ind)].lmx);}tree[ind].mx=max(tree[ind].mx,max(tree[ind].lmx,tree[ind].rmx));}void build(int lft,int rht,int ind){tree[ind].lft=lft; tree[ind].rht=rht;if(lft==rht) {tree[ind].lval=tree[ind].rval=y[lft];tree[ind].lmx=tree[ind].rmx=tree[ind].mx=1;}else {int mid=tree[ind].mid();build(lft,mid,LL(ind));build(mid+1,rht,RR(ind));up(ind);}}void updata(int pos,int ind,int valu){int lft=tree[ind].lft,rht=tree[ind].rht;if(lft==rht) tree[ind].lval=tree[ind].rval=valu;else {int mid=tree[ind].mid();if(pos<=mid) updata(pos,LL(ind),valu);else updata(pos,RR(ind),valu);up(ind);}}int query(int st,int ed,int ind){int lft=tree[ind].lft,rht=tree[ind].rht;if(st<=lft&&rht<=ed) return tree[ind].mx;else {int mid=tree[ind].mid();if(ed<=mid) return query(st,ed,LL(ind));else if(st>mid) return query(st,ed,RR(ind));else {int tmp1=0,tmp2=0;int mx1=query(st,ed,LL(ind)),mx2=query(st,ed,RR(ind));if(tree[LL(ind)].rval<tree[RR(ind)].lval){tmp1=min(mid-st+1,tree[LL(ind)].rmx);tmp2=min(ed-mid,tree[RR(ind)].lmx);}return max(max(mx1,mx2),tmp1+tmp2);}}}}seg;int main(){int t;scanf("%d",&t);while(t--){scanf("%d%d",&n,&m);for(int i=0;i<n;i++) scanf("%d",&y[i]);seg.build(0,n-1,1);while(m--){char cmd[5]; int a,b;scanf("%s%d%d",cmd,&a,&b);if(cmd[0]=='Q') printf("%d\n",seg.query(a,b,1));else seg.updata(a,1,b);}}return 0;}
/*代碼風格更新前*/#include <iostream>#include <cstdio>using namespace std;const int N=100005;int y[N];struct node{ int left,right; int rmax,lmax,mmax; int mid(){return left+(right-left)/2;} int dis(){return right-left+1;}};void unin(node &a,node &b,node &c){ a.lmax=b.lmax; a.rmax=c.rmax; a.mmax=max(b.mmax,c.mmax); int mid=a.mid(); if(y[mid]<y[mid+1]) { if(b.lmax==b.dis()) a.lmax+=c.lmax; if(c.rmax==c.dis()) a.rmax+=b.rmax; a.mmax=max(a.mmax,b.rmax+c.lmax); }}struct Segtree{ node tree[N*4]; void build(int left,int right,int r) { tree[r].left=left; tree[r].right=right; tree[r].rmax=tree[r].lmax=0; if(left==right) { tree[r].lmax=tree[r].rmax=tree[r].mmax=1; } else { int mid=tree[r].mid(); build(left,mid,r*2); build(mid+1,right,r*2+1); unin(tree[r],tree[r*2],tree[r*2+1]); } } void updata(int pos,int r,int co) { if(tree[r].left==tree[r].right) { y[tree[r].left]=co; } else { int mid=tree[r].mid(); if(pos<=mid) updata(pos,r*2,co); else if(pos>mid) updata(pos,r*2+1,co); unin(tree[r],tree[r*2],tree[r*2+1]); } } int query(int be,int end,int r) { if(be<=tree[r].left&&tree[r].right<=end) { return tree[r].mmax; } else { int mid=tree[r].mid(); if(end<=mid) return query(be,end,r*2); else if(be>mid) return query(be,end,r*2+1); else { int mid=tree[r].mid(); int max1=query(be,end,r*2); int max2=query(be,end,r*2+1); if(y[mid]>=y[mid+1]) return max(max1,max2); else { int max3=min(tree[r*2].right-be+1,tree[r*2].rmax)+min(end-tree[r*2+1].left+1,tree[r*2+1].lmax); return max(max(max1,max2),max3); } } } }}seg;int main(){ int t; scanf("%d",&t); while(t--) { int n,m; scanf("%d%d",&n,&m); for(int i=0;i<n;i++) scanf("%d",&y[i]); seg.build(0,n-1,1); for(int i=0;i<m;i++) { char cmd[5]; int a,b; scanf("%s%d%d",cmd,&a,&b); if(cmd[0]=='Q') { printf("%d\n",seg.query(a,b,1)); } else if(cmd[0]=='U') { seg.updata(a,1,b); } } } return 0;}