首页 > 代码库 > Wikioi 1081 线段树练习2

Wikioi 1081 线段树练习2

线段树练习飘逸的写法,自从自己改成这种写法之后,线段树就没再练过,现在终于练得上了。

因为这里查询只是查询了叶子结点,所以pushUp函数就用不上了,不过我没去掉之前是3ms,去掉之后反而变成4ms了,搞不懂怎么原因,没用到,去掉之后应该更快才对啊,竟然变慢了,真搞不明白?

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <deque>
#include <vector>
#include <queue>
#include <string>
#include <cstring>
#include <map>
#include <stack>
#include <set>
#define PI acos(-1.0)
#define mem(a,b) memset(a,b,sizeof(a))
#define sca(a) scanf("%d",&a)
#define sc(a,b) scanf("%d%d",&a,&b)
#define pri(a) printf("%d\n",a)
#define lson i<<1,l,mid
#define rson i<<1|1,mid+1,r
#define MM 100004
#define MN 1008
#define INF 2000000000
#define eps 1e-8
using namespace std;
typedef long long ll;
typedef unsigned long long ULL;
int sum[MM],val[MM];
//void pushUp(int i)
//{
//    sum[i]=sum[i<<1]+sum[i<<1|1];
//}
void pushDown(int i) //处理lazy标记
{
    if(val[i])
    {
        val[i<<1]+=val[i],val[i<<1|1]+=val[i];
        sum[i<<1]+=val[i],sum[i<<1|1]+=val[i];
        val[i]=0;
    }
}
void build(int i,int l,int r)
{
    sum[i]=val[i]=0;
    if(l==r) return ;
    int mid=(l+r)>>1;
    build(lson),build(rson);
}
void update(int i,int l,int r,int L,int R,int v)
{
    if(L<=l&&r<=R)
    {
        val[i]+=v;
        sum[i]+=v;
        return ;
    }
    int mid=(l+r)>>1;
    pushDown(i);
    if(L<=mid) update(lson,L,R,v);
    if(R>mid) update(rson,L,R,v);
    //pushUp(i);
}
int query(int i,int l,int r,int x)
{
    if(l==x&&r==x) return sum[i];
    int mid=(l+r)>>1;
    pushDown(i);
    if(x<=mid) return query(lson,x);
    else return query(rson,x);
}
int main()
{
    int n,q,mm,i,a,b,s;
    sca(n);
    build(1,1,n);
    for(i=1;i<=n;i++)
    {
        sca(a);
        update(1,1,n,i,i,a);
    }
    sca(q);
    while(q--)
    {
        sca(mm);
        if(mm==1)
        {
            scanf("%d%d%d",&a,&b,&s);
            update(1,1,n,a,b,s);
        }
        else
        {
            sca(s);
            pri(query(1,1,n,s));
        }
    }
    return 0;
}