題目本身並不複雜,這邊簡單的說一下。
不妨將新增操作視為取一個下界,移除操作視為取一個上界,將每個操作依照時間加入線段樹之中,最後輸出答案時走訪一遍線段樹就好了。
那麼本題唯一的難點就是:處理上下界。
然而這也並不是很難......手算模擬一下可能會遇到的狀況,用力分析一下就行了。
可以看看code中的add和remove操作感受一下。code的其他部份就比較無趣了,就是經典的線段樹.....
註:code中Max[id]==-1的意思就是「整段區間都是Min[id]」的情況
寫法有很多種,可以自己嘗試一下。
//#define LOCAL #include<algorithm> #include<cstdio> #include "lib1895.h" using namespace std; const int maxn=2000000+5; const int INF=(1<<30); int Max[maxn*6],Min[maxn*6]; inline void add(int id,int val) { if(Max[id]==-1) Min[id]=max(Min[id],val); else if(val>=Max[id]) Min[id]=val,Max[id]=-1; else if(val>=Min[id]) Min[id]=val; } inline void remove(int id,int val) { if(Max[id]==-1) Min[id]=min(Min[id],val); else if(val<=Min[id]) Min[id]=val,Max[id]=-1; else if(val<=Max[id]) Max[id]=val; } inline void push(int id) { if(Max[id]==-1) { Max[id*2]=Max[id*2+1]=-1; Min[id*2]=Min[id*2+1]=Min[id]; Max[id]=INF,Min[id]=0; return; } if(Min[id]!=0) { add(id*2,Min[id]); add(id*2+1,Min[id]);} if(Max[id]!=INF) { remove(id*2,Max[id]); remove(id*2+1,Max[id]); } Max[id]=INF,Min[id]=0; } void modify(int id,int L,int R,int qL,int qR,int qtype,int qhei) { if(qL<= L && R <=qR) { if(qtype==1) add(id,qhei); else remove(id,qhei); return; } int M=L+(R-L)/2; push(id); if(qL<=M) modify(id*2,L,M,qL,qR,qtype,qhei); if(qR> M) modify(id*2+1,M+1,R,qL,qR,qtype,qhei); } void buildans(int id,int L,int R,int* ans) { if(L==R) {ans[L]=Min[id];return;} int M=L+(R-L)/2; push(id); buildans(id*2,L,M,ans); buildans(id*2+1,M+1,R,ans); } void buildWall(int n,int k,int op[],int left[],int right[],int height[],int finalHeight[]) { for(int i=0;i<n*3;i++) Max[i]=INF,Min[i]=0; for(int i=0;i<k;i++) modify(1,0,n-1,left[i],right[i],op[i],height[i]); buildans(1,0,n-1,finalHeight); }
沒有留言:
張貼留言