poj3468 A Simple Problem with Integers

题意:O(-1) 思路:O(-1) 线段树功能:update:成段增减 query:区间求和
用long long 否则超时

#include <iostream>
#include <cmath>
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <set>
#include <queue>
#include <stack>
#define inf 0x3f3f3f3f
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
const long long mod = 1e9 + 7;
const int maxn = 55555;
using namespace std;
typedef long long ll;

int h,w,n;
int col[maxn<<2];
int sum[maxn<<2];

void pushup(int rt){
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}

void pushdown(int rt,int m){
    if(col[rt]){
        col[rt<<1]+=col[rt];
        col[rt<<1|1]+=col[rt];
        sum[rt<<1]+=(m-(m>>1))*col[rt];
        sum[rt<<1|1]+=(m>>1)*col[rt];
        col[rt]=0;
    }
}

void build(int l,int r,int rt){
    col[rt]=0;
    //sum[rt]=1;
    if(l==r){
        scanf("%d",&sum[rt]);
        return ;
    }
    int m=(l+r)>>1;
    build(lson);
    build(rson);
    pushup(rt);
}

void update(int L,int R,int c,int l,int r,int rt){
    if(L<=l&&r<=R){
        col[rt]+=c;
        sum[rt]+=c*(r-l+1);
        return ;
    }
    pushdown(rt,r-l+1);
    int m=(l+r)>>1;
    if(L<=m)update(L,R,c,lson);
    if(m<R)update(L,R,c,rson);
    pushup(rt);
}

int query(int L,int R,int l,int r,int rt){
    if(L<=l&&r<=R){
        return sum[rt];
    }
    pushdown(rt,r-l+1);
    int m=(l+r)>>1;
    int ret=0;
    if(L<=m)ret+=query(L,R,lson);
    if(m<R) ret+=query(L,R,rson);
    return ret;
}

int main(){
    int n,q;
    scanf("%d%d",&n,&q);
    build(1,n,1);
    while(q--){
        char op[2];
        int a,b,c;
        scanf("%s",op);
        if(op[0]=='Q'){
            scanf("%d%d",&a,&b);
            printf("%d\n",query(a,b,1,n,1));
        }
        else{
            scanf("%d%d%d",&a,&b,&c);
            update(a,b,c,1,n,1);
        }
    }
    return 0;
}

另一种风格加详解

#include <iostream>
#include <cstdio>
#include <queue>
#include <set>
#include <algorithm>
#include <cstring>
//priority_queue<int,vector<int>,greater<int> > q;
//priority_queue<int,vector<int>,less<int> > q;
#define lson l, mid, rt << 1
#define rson mid + 1, r, rt << 1 | 1
using namespace std;
typedef long long ll;
const int inf = 0x3f3f3f3f;
const int maxn = 2e5 + 10;
#define lowbit(x) (x&(-x))
//std::ios::sync_with_stdio(false);
//std::cin.tie(0);
//std::cout.tie(0);

ll Sum[maxn<<2],Add[maxn<<2];//Sum求和，Add为懒惰标记
ll A[maxn];//存原数组数据下标[1,n]
//PushUp函数更新节点信息 ，这里是求和
void PushUp(int rt)
{
    Sum[rt]=Sum[rt<<1]+Sum[rt<<1|1];
}
//Build函数建树
void Build(int l,int r,int rt)  //l,r表示当前节点区间，rt表示当前节点编号
{
    if(l==r)  //若到达叶节点
    {
        Sum[rt]=A[l];//储存数组值
        return;
    }
    int m=(l+r)>>1;
    //左右递归
    Build(l,m,rt<<1);
    Build(m+1,r,rt<<1|1);
    //更新信息
    PushUp(rt);
}
//点修改 假设A[L]+=C:
//void Update(int L,int C,int l,int r,int rt) //l,r表示当前节点区间，rt表示当前节点编号
//{
  //  if(l==r) //到叶节点，修改
 //   {
  //      Sum[rt]+=C;
 //       return;
  //  }
  //  int m=(l+r)>>1;
    //根据条件判断往左子树调用还是往右
  //  if(L <= m) Update(L,C,l,m,rt<<1);
 //   else       Update(L,C,m+1,r,rt<<1|1);
  //  PushUp(rt);//子节点更新了，所以本节点也需要更新信息
//}
//下推标记
void PushDown(int rt,int ln,int rn)
{
    //ln,rn为左子树，右子树的数字数量。
    if(Add[rt])
    {
        //下推标记
        Add[rt<<1]+=Add[rt];
        Add[rt<<1|1]+=Add[rt];
        //修改子节点的Sum使之与对应的Add相对应
        Sum[rt<<1]+=Add[rt]*ln;
        Sum[rt<<1|1]+=Add[rt]*rn;
        //清除本节点标记
        Add[rt]=0;
    }
}
//区间修改 假设A[L,R]+=C
void Update(int L,int R,int C,int l,int r,int rt) //L,R表示操作区间，l,r表示当前节点区间，rt表示当前节点编号
{
    if(L <= l && r <= R) //如果本区间完全在操作区间[L,R]以内
    {
        Sum[rt]+=C*(r-l+1);//更新数字和，向上保持正确
        Add[rt]+=C;//增加Add标记，表示本区间的Sum正确，子区间的Sum仍需要根据Add的值来调整
        return ;
    }
    int m=(l+r)>>1;
    PushDown(rt,m-l+1,r-m);//下推标记
    //这里判断左右子树跟[L,R]有无交集，有交集才递归
    if(L <= m) Update(L,R,C,l,m,rt<<1);
    if(R >  m) Update(L,R,C,m+1,r,rt<<1|1);
    PushUp(rt);//更新本节点信息
}
//区间查询
ll Query(int L,int R,int l,int r,int rt) //L,R表示操作区间，l,r表示当前节点区间，rt表示当前节点编号
{
    if(L <= l && r <= R)
    {
        //在区间内，直接返回
        return Sum[rt];
    }
    int m=(l+r)>>1;
    //下推标记，否则Sum可能不正确
    PushDown(rt,m-l+1,r-m);

    //累计答案
    ll ANS=0;
    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()
{
    std::ios::sync_with_stdio(false);
    std::cin.tie(0);
    std::cout.tie(0);
    int t,k=0;

        int n,m;
       while(cin>>n>>m)
       {
           for(int i=1;i<=n;i++)
        {
            cin>>A[i];
        }
        Build(1,n,1);
        string s;
        int a,b;
        ll c;
     //   cout<<"Case "<<++k<<":"<<endl;
        while(m--)
        {
            cin>>s;
            if(s[0]=='Q')
            {
                cin>>a>>b;
                cout<<Query(a,b,1,n,1)<<endl;
            }
            else if(s[0]=='C')
            {
                cin>>a>>b>>c;
                Update(a,b,c,1,n,1);
            }
            //else if(s[0]=='S')
            //{
            //    Update(a,-b,1,n,1);
            //}
        }
       }

    return 0;
}