POJ-3468-A Simple Problem with Integers（Lazy算法）

C - A Simple Problem with Integers
Time Limit:5000MS Memory Limit:131072KB 64bit IO Format:%I64d & %I64u
Submit

Status

Practice

POJ 3468
Description
You have N integers, A1, A2, … , AN. You need to deal with two kinds of operations. One type of operation is to add some given number to each number in a given interval. The other is to ask for the sum of numbers in a given interval.

Input
The first line contains two numbers N and Q. 1 ≤ N,Q ≤ 100000.
The second line contains N numbers, the initial values of A1, A2, … , AN. -1000000000 ≤ Ai ≤ 1000000000.
Each of the next Q lines represents an operation.
“C abc” means adding c to each of Aa, Aa+1, … , Ab. -10000 ≤ c ≤ 10000.
“Q ab” means querying the sum of Aa, Aa+1, … , Ab.

Output
You need to answer all Q commands in order. One answer in a line.

Sample Input
10 5
1 2 3 4 5 6 7 8 9 10
Q 4 4
Q 1 10
Q 2 4
C 3 6 3
Q 2 4
Sample Output
4
55
9
15
Hint
The sums may exceed the range of 32-bit integers.

题意：第一行给出N和Q，接下来一行有N个数字输入，接下来有Q行操作，每行首个字母为Q，后边跟上a和b，输出a到b区间所有数字和，首字母若为C，后边跟上abc，代表区间a到b的值都加上c。

裸的Lazy算法，线段树的延迟更新

代码

#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<string.h>
using namespace std;
const int maxn=100005;
long long int sum[maxn<<2];//i结点对应区间总和为sum[i]
long long int add[maxn<<2];//i结点对应区间每个元素都需要增加add[i]的数值
int N;//数据数量
int Q;//操作数量
void Push_Up(int root)//数据上浮
{
    sum[root]=sum[root<<1]+sum[root<<1|1];
}
void Push_Down(int root,int length)//数据下沉，传入下沉结点和下沉区间差值
{
    if(add[root])//如果root对应区间需要更新
    {
        add[root<<1]+=add[root];//左右子叶状态更新
        add[root<<1|1]+=add[root];
        sum[root<<1]+=add[root]*(length-(length>>1));//左右子叶元素和更新
        sum[root<<1|1]+=add[root]*(length>>1);
        add[root]=0;//标记root区间状态已更新
    }
}
void Build(int root,int left,int right)//建树
{
    add[root]=0;//初始化所有结点对应区间均不需要更新
    if(left==right)//最底层接收数据
    {
        scanf("%I64d",&sum[root]);
        return;
    }
    int mid=(left+right)>>1;
    Build(root<<1,left,mid);//建立左子叶
    Build(root<<1|1,mid+1,right);//建立右子叶
    Push_Up(root);//建立后的子叶开始数据上浮
}
long long int Query(int root,int left,int right,int find_left,int find_right)//传入根节点，最大查找区间，及目标查找区间
{
    if(find_left<=left&&find_right>=right)//当前区间完全在查询区间范围内
        return sum[root];//返回当前区间总和
    Push_Down(root,right-left+1);//由于是延迟更新，所以数据需要下沉一层
    int mid=(left+right)>>1;
    long long int sum_num=0;//设定总和
    if(find_right<=mid)//待查找区间完全在当前区间左子叶部分
        sum_num+=Query(root<<1,left,mid,find_left,find_right);
    else if(find_left>mid)//待查找区间完全在当前区间右子叶部分
        sum_num+=Query(root<<1|1,mid+1,right,find_left,find_right);
    else//待查找区间横跨当前区间左右子叶
    {
        sum_num+=Query(root<<1,left,mid,find_left,mid);
        sum_num+=Query(root<<1|1,mid+1,right,mid+1,find_right);
    }
    return sum_num;
}
void Update(int root,int left,int right,int find_left,int find_right,int num)//传入根节点，最大区间，更新区间，跟新数值
{
    if(find_left<=left&&find_right>=right)//当前区间完全处于待更新区间内
    {
        add[root]+=num;//更新状态
        sum[root]+=(long long int)num*(right-left+1);//更新总和
        return;
    }
    Push_Down(root,right-left+1);//由于是延迟更新，所以还需要数据下沉一层
    int mid=(left+right)>>1;
    if(find_right<=mid)//如果待查找区间完全在当前区间左子叶部分
        Update(root<<1,left,mid,find_left,find_right,num);
    else if(find_left>mid)//如果待查找区间完全在当前区间右子叶部分
        Update(root<<1|1,mid+1,right,find_left,find_right,num);
    else//如果待查找区间横跨当前区间
    {
        Update(root<<1,left,mid,find_left,mid,num);
        Update(root<<1|1,mid+1,right,mid+1,find_right,num);
    }
    Push_Up(root);
}
int main()
{
    scanf("%d%d",&N,&Q);
    Build(1,1,N);//建树的同时，接收N个数
    while(Q--)
    {
        char str[3];//判断字符
        int a,b,c;//含义如题
        scanf("%s",str);
        if(str[0]=='Q')//查询操作
        {
            scanf("%d%d",&a,&b);
            printf("%I64d\n",Query(1,1,N,a,b));
        }
        else//更新操作
        {
            scanf("%d%d%d",&a,&b,&c);
            Update(1,1,N,a,b,c);
        }
    }
    return 0;
}