线段树入门题 hdu1166 敌兵布阵

题意:动态查询一段区间的和,支持单点更新。


开始刷线段树了,这是入门第一题。。。最基本的线段树操作,递归建树,递归查询,递归修改,向上更新节点,每个节点的值代表该节点对应区间的和 。

代码:

#include 
   
   
    
    
#include 
    
    
     
     
#include 
     
     
      
      
#include 
      
      
       
       
#include 
       
       
         #include 
        
          #include 
         
           #include 
          
            #include 
           
             #include 
            
              #include 
             
               #include 
               #include 
               
                 #include 
                
                  #include 
                 
                   #include 
                  
                    #include 
                   
                     #include 
                    
                      #include 
                     
                       #include 
                      
                        #include 
                       
                         #include 
                        
                          #include 
                         
                           #include 
                          
                            using namespace std; #define PB push_back #define MP make_pair #define REP(i,x,n) for(int i=x;i<(n);++i) #define FOR(i,l,h) for(int i=(l);i<=(h);++i) #define FORD(i,h,l) for(int i=(h);i>=(l);--i) #define SZ(X) ((int)(X).size()) #define ALL(X) (X).begin(), (X).end() #define RI(X) scanf("%d", &(X)) #define RII(X, Y) scanf("%d%d", &(X), &(Y)) #define RIII(X, Y, Z) scanf("%d%d%d", &(X), &(Y), &(Z)) #define DRI(X) int (X); scanf("%d", &X) #define DRII(X, Y) int X, Y; scanf("%d%d", &X, &Y) #define DRIII(X, Y, Z) int X, Y, Z; scanf("%d%d%d", &X, &Y, &Z) #define OI(X) printf("%d",X); #define RS(X) scanf("%s", (X)) #define MS0(X) memset((X), 0, sizeof((X))) #define MS1(X) memset((X), -1, sizeof((X))) #define LEN(X) strlen(X) #define F first #define S second #define Swap(a, b) (a ^= b, b ^= a, a ^= b) #define Dpoint strcut node{int x,y} #define cmpd int cmp(const int &a,const int &b){return a>b;} /*#ifdef HOME freopen("in.txt","r",stdin); #endif*/ const int MOD = 1e9+7; typedef vector 
                           
                             VI; typedef vector 
                            
                              VS; typedef vector 
                             
                               VD; typedef long long LL; typedef pair 
                              
                                PII; //#define HOME int Scan() { int res = 0, ch, flag = 0; if((ch = getchar()) == '-') //判断正负 flag = 1; else if(ch >= '0' && ch <= '9') //得到完整的数 res = ch - '0'; while((ch = getchar()) >= '0' && ch <= '9' ) res = res * 10 + ch - '0'; return flag ? -res : res; } /*----------------PLEASE-----DO-----NOT-----HACK-----ME--------------------*/ int sum[200000+10]; int a[50000+10]; void pushplus(int rt) { sum[rt]=sum[rt<<1]+sum[(rt<<1)+1]; } int cnt; void build(int l,int r,int rt) { if(l==r) { sum[rt]=a[++cnt]; return; } int m=(l+r)>>1; build(l,m,rt<<1); build(m+1,r,(rt<<1)+1); pushplus(rt); } void update(int node,int l,int r,int add,int rt) {if(l==r) { sum[rt]+=add; return; } int m=(l+r)>>1; if(node<=m) update(node,l,m,add,rt<<1); else update(node,m+1,r,add,(rt<<1)+1); pushplus(rt); } int query(int L,int R,int l,int r,int rt) {if(L<=l&&r<=R) { return sum[rt]; } int ans=0; int m=(l+r)>>1; if(L<=m) ans+=query(L,R,l,m,rt<<1); if(R>m) ans+=query(L,R,m+1,r,(rt<<1)+1); return ans; } int main() {int T; RI(T); for(int i=1;i<=T;i++) { printf("Case %d:\n",i); int n; RI(n); for(int i=1;i<=n;i++) RI(a[i]); cnt=0; build(1,n,1); char s[100]; while(scanf("%s",s)) { if(s[0]=='E') break; int a,b; RII(a,b); if(s[0]=='Q') printf("%d\n",query(a,b,1,n,1)); else { if(s[0]=='S') b=-b; update(a,1,n,b,1); } } } return 0; } 
                               
                              
                             
                            
                           
                          
                         
                        
                       
                      
                     
                    
                   
                  
                 
                
              
             
            
           
          
         
       
      
      
     
     
    
    
   
   


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值