2016年4月12日 星期二

[TIOJ 1895][IOI 2014] Wall

原題連結
題目本身並不複雜,這邊簡單的說一下。



不妨將新增操作視為取一個下界,移除操作視為取一個上界,將每個操作依照時間加入線段樹之中,最後輸出答案時走訪一遍線段樹就好了。

那麼本題唯一的難點就是:處理上下界。
然而這也並不是很難......手算模擬一下可能會遇到的狀況,用力分析一下就行了。
可以看看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);
}

沒有留言:

張貼留言