关闭

poj3468(线段树成段更新)

标签: 线段树数据结构
510人阅读 评论(0) 收藏 举报
分类:

链接:poj3468

/*poj3468 A Simple Problem with Integers成段更新,区间求和
题目大意:
首先给出一个数列,表示初始值,然后两种操作,查询某段区间的和,
和更新某区间所有数,每个数加上add

思路:lazy思想
add不为零时表示区间内所有的数同时加上add
更新时不直接更新到底
*/
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define N 100005
#define LL long long
int n,add;
struct node
{
    int l,r;
    long long sum;
    int add;
}s[N<<2];
int a[N];
void build(int l, int r, int n)
{
    s[n].l = l;
    s[n].r = r;
    s[n].add = 0;
    if(l == r)
    {
        s[n].sum = a[l];
        return;
    }
    int mid = (l+r)>>1;
    build(l, mid, n<<1);
    build(mid+1, r, n<<1|1);
    s[n].sum = s[n<<1].sum + s[n<<1|1].sum;
}
void update(int l, int r, int n)
{
    if(l <= s[n].l && s[n].r <= r)//要更新的区间将该节点表示的区间囊括在内
    {
        s[n].sum += (s[n].r - s[n].l + 1)*add;
        s[n].add += add;
        return;
    }
    if(s[n].add != 0)
    {
        //根据父节点的值来更新子节点
        s[n<<1].sum += (LL)(s[n<<1].r - s[n<<1].l + 1)*s[n].add;
        s[n<<1|1].sum += (LL)(s[n<<1|1].r - s[n<<1|1].l + 1)*s[n].add;
        s[n<<1].add += s[n].add;
        s[n<<1|1].add += s[n].add;
        s[n].add = 0;
    }
    int mid = (s[n].l + s[n].r) >> 1;
    if(r <= mid)
        update(l, r, n<<1);
    else if(l > mid)
        update(l, r, n<<1|1);
    else
    {
        update(l, mid, n<<1);
        update(mid+1, r, n<<1|1);
    }
    s[n].sum = s[n<<1].sum + s[n<<1|1].sum;
}
LL getsum(int l, int r, int n)
{
    if(s[n].l == l && s[n].r == r)
        return s[n].sum;
    if(s[n].add != 0)
    {
        s[n<<1].sum += (LL)(s[n<<1].r - s[n<<1].l + 1)*s[n].add;
        s[n<<1|1].sum += (LL)(s[n<<1|1].r - s[n<<1|1].l + 1)*s[n].add;
        s[n<<1].add += s[n].add;
        s[n<<1|1].add += s[n].add;
        s[n].add = 0;
    }
    int mid = (s[n].l + s[n].r) >> 1;
    if(r <= mid)
        return getsum(l, r, n<<1);
    else if(l > mid)
        return getsum(l, r, n<<1|1);
    else
        return getsum(l, mid, n<<1) + getsum(mid+1, r, n<<1|1);
}

int main()
{

	int i,m,x,y;
	char ch[2];
	while(~scanf("%d%d",&n,&m))
	{
	    for(i = 1; i <= n; i ++)
            scanf("%d",&a[i]);
        build(1,n,1);
	    while(m--)
	    {
	        scanf("%s",ch);
	        if(ch[0] == 'Q')
	        {
	            scanf("%d%d",&x,&y);
	            printf("%lld\n",getsum(x,y,1));
	        }
	        else if(ch[0] == 'C')
	        {
	            scanf("%d%d%d",&x,&y,&add);
	            if(!add) continue;
	            update(x,y,1);
	        }
	    }
	}
	return 0;
}


0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:133572次
    • 积分:3126
    • 等级:
    • 排名:第11465名
    • 原创:186篇
    • 转载:13篇
    • 译文:0篇
    • 评论:2条
    最新评论