【线段树II：区间修改+点查询】hdu 1556 Color the ball_查询哪些区间包含给定的点-CSDN博客

本文链接：https://blog.csdn.net/u012717411/article/details/47154465

【线段树II：区间修改+点查询】hdu 1556 Color the ball

题目链接：hdu 1556 Color the ball

题目大意

给你N个气球，不断刷新指定区间的颜色，刷新N次，最后输出每一个气球的刷新次数。

上一篇文章是线段树的点修改、区间查询；
这篇文章是线段树的区间修改、点查询；
这里写图片描述

说一下思路

线段树的区间修改：利用线段树的区间查询，查询到叶节点segTree[root].sum++,而如果对区间进行多次点修改的话，注定超时
线段树的点查询：以为用之前的区间查询就可以了，实际上还是有改变，因为原理不是特别懂，后面慢慢搞懂！

参考代码

/*Author:Hacker_vision*/
#include<bits/stdc++.h>
using namespace std;

const int _max = 1e5 + 10;
int n,x,y;
struct segTree{
  int lc,rc;
  int sum;
}segTree[_max<<2];

void build(int root,int L,int R){//建树O(logN)
  segTree[root].lc=L;
  segTree[root].rc=R;
  if(L==R){
    segTree[root].sum=0;//初始值为0：A[1].A[2]..A[N]=0;
    return ;
  }
  int mid = (L + R) >> 1;
  build(root<<1,L,mid);
  build(root<<1|1,mid+1,R);
  segTree[root].sum=segTree[root<<1].sum+segTree[root<<1|1].sum;
}

void update(int root,int L,int R){//区间修改
  if(segTree[root].lc==L&&segTree[root].rc==R) {
    segTree[root].sum++;
    return;
  }
  int mid = (segTree[root].lc + segTree[root].rc) >> 1;
  if(R <= mid) update(root<<1,L,R);
  else if(mid < L) update(root<<1|1,L,R);
  else{
    update(root<<1,L,mid);
    update(root<<1|1,mid+1,R);
  }
}

int query(int root,int L,int R){//点(区间)查询
  if(segTree[root].lc==L&&segTree[root].rc==R) return segTree[root].sum;
  int mid = (segTree[root].lc + segTree[root].rc) >> 1 ;
  if(R <= mid) return segTree[root].sum+query(root<<1,L,R);
  else if(mid < L) return segTree[root].sum+query(root<<1|1,L,R);
  else return segTree[root].sum+query(root<<1,L,mid)+query(root<<1|1,mid+1,R);
}

int main(){
 // freopen("input.txt","r",stdin);
  while(scanf("%d",&n)==1&&n){
    build(1,1,n);
    for(int i = 0;i < n; ++ i){
        scanf("%d%d",&x,&y);
      //  printf("%d%d\n",x,y);
        update(1,x,y);
    }
    for(int i = 1; i <= n; ++ i)
      if(i==1) printf("%d",query(1,i,i));
      else printf(" %d",query(1,i,i));
      puts("");
  }
  return 0;
}