POJ 3468 A Simple Problem with Integers

标签：线段树

题意很明确，区间修改+查询。

我用线段树写的。核心思想就是延时操作，update[]表示当前节点需要增加多少。

如果覆盖了这个区间就不向下传递了，修改update[]就行了。

#include<cstdio>
#include<cstring>
#include<string>
#include<queue>
#include<algorithm>
#include<map>
#include<stack>
#include<iostream>
#include<list>
#include<set>
#include<bitset>
#include<vector>
#include<cmath>

#define INF 0x7fffffff
#define eps 1e-8
#define LL long long
#define PI 3.141592654
#define CLR(a,b) memset(a,b,sizeof(a))
#define FOR(i,a,b) for(int i=a;i<b;i++)
#define FOR_(i,a,b) for(int i=a;i>=b;i--)
#define pub push_back
#define puf push_front
#define pob pop_back
#define pof pop_front
#define mp make_pair
#define ft first
#define sd second
#define sf scanf
#define pf printf
#define sz(v) ((int)(v).size())
#define all(v) (v).begin(),(v).end()
#define acfun std::ios::sync_with_stdio(false)

#define SIZE (100000+1)
#define MOD 1000000007
using namespace std;

LL a[SIZE];
LL t[SIZE*3];
//生成线段树
LL build(int o,int l,int r)
{
    if(l==r)return t[o]=a[l];
    int m=(l+r)>>1;
    return t[o]=build(o*2,l,m)+build(o*2+1,m+1,r);
}
//线段树区间更新
int ul,ur;
LL update[SIZE*3];
LL up_ans;
int up_site;
void update_interval(int o,int l,int r)
{
    if(ul<=l&&ur>=r)
    {
        if(l==r)t[o]+=up_ans;
        else
        update[o]+=up_ans;
    }
    else
    {
        int m=(l+r)>>1;
        if(ul<=m)update_interval(o*2,l,m);
        if(ur>m)update_interval(o*2+1,m+1,r);
    }
    if(r>l)
        t[o]=t[o*2]+t[o*2+1];
    t[o]+=update[o]*(r-l+1);
}
//线段树区间查询（对应区间更新）
LL ans;
int ql,qr;
void query_interval(int o,int l,int r,LL add)
{
    if(ql<=l&&qr>=r)
        ans+=t[o]+add*(r-l+1);
    else
    {
        int m=(l+r)>>1;
        if(ql<=m)query_interval(o*2,l,m,add+update[o]);
        if(qr>m)query_interval(o*2+1,m+1,r,add+update[o]);
    }
}
int main()
{
    int n,m;
    while(~sf("%d%d",&n,&m))
    {
        FOR(i,1,n+1)
        sf("%lld",&a[i]);
        build(1,1,n);
        CLR(update,0);
        while(m--)
        {
            char ch[2];
            sf("%s",ch);
            if(ch[0]=='C')
            {
                sf("%d%d%lld",&ul,&ur,&up_ans);
                update_interval(1,1,n);
            }
            else
            {
                ans=0;
                sf("%d%d",&ql,&qr);
                query_interval(1,1,n,0);
                pf("%lld\n",ans);
            }
        }
    }
}

POJ 3468 A Simple Problem with Integers

标签：线段树

原文地址：http://blog.csdn.net/dongshimou/article/details/43151707