hdu 1698 線段樹區間修改

【題目鏈接】
http://acm.hdu.edu.cn/showproblem.php?pid=1698

【解題報告】
區間修改的模板題目。需要打一個set標記,表示區間(L,R)被修改爲v.
在這道題目裏。一開始所有結點被賦值爲1,之後給Q個修改,把(L,R)修改爲1,2,或3.最後求區間和。
我的寫法採用了劉汝佳在訓練指南中使用的寫法。

【參考代碼】

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
using namespace std;

const int maxn=100000+50;

int N,Q;
int setv[maxn*4],sumv[maxn*4];

void maintain(  int O, int L, int R  )
{
      if(  L>R )return;
      int lc=O*2,rc=O*2+1;
      sumv[O]=0;
      if(  L<R  &&  setv[O]==-1  )
      {
            sumv[O]=sumv[lc]+sumv[rc];
      }
      if(  setv[O]>=0 )sumv[O]=setv[O]*( R-L+1 );
}

void pushdown(  int O  )
{
      int lc=O*2, rc=O*2+1;
      if(  setv[O]>=0  )
      {
            setv[lc]=setv[rc]=setv[O];
            setv[O]=-1;
      }
}

void update(   int O, int L, int R, int x, int y, int z  )
{
      if( L>R )return;
      if(  x<=L && R<=y  )
      {
            setv[O]=z;
      }
      else
      {
            pushdown(  O );
            int mid=(L+R)/2;
            if(   x<=mid   )update(   O*2, L, mid, x, y, z  ); else  maintain(  O*2, L, mid  );
            if(  y>mid )update(  O*2+1, mid+1, R, x, y, z ); else maintain(  O*2+1, mid+1, R );
      }
      maintain(   O,  L,  R  );
}


int query(  int O, int L , int R, int qL, int qR  )
{
      if( L>R )return 0;
      if(   setv[O]>=0  )
      {
            return setv[O]*(    min(R, qR)-max(L, qL) +1  );
      }
      else  if(   qL<=L &&  R<=qR  )  return sumv[O];
      else
      {
            int ans=0,mid=( L+R )/2;
            if(   qL<=mid )ans+=query(    O*2, L, mid, qL, qR );
            if(  qR>mid ) ans+=query(  O*2+1, mid+1, R,  qL, qR );
            return ans;
      }
}

int main()
{
      int T,kase=0;
      cin>>T;
      while( T-- )
      {
            memset(  setv,0,sizeof setv );
            memset( sumv,0,sizeof sumv );
            scanf(  "%d%d", &N, &Q  );
            setv[1]=1;
            for(  int i=1; i<=Q; i++ )
            {
                  int x,y,z;
                  scanf(  "%d%d%d",   &x, &y, &z  );
                  update(   1,1,N, x,y,z   );

                 //  printf(  "Case %d: The total value of the hook is %d.\n" ,   ++kase, query(  1,1,N, 1,N  )  );
            }
            printf(  "Case %d: The total value of the hook is %d.\n" ,   ++kase, query(  1,1,N, 1,N  )  );
      }

      return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章