1081 线段树练习 2

1081 线段树练习 2 

codevs 1081

 时间限制: 1 s

 空间限制: 128000 KB

 题目等级 : 大师 Master 

 

题目描述 Description

给你N个数，有两种操作

1：给区间[a,b]的所有数都增加X

2：询问第i个数是什么？

输入描述 Input Description

第一行一个正整数n，接下来n行n个整数，再接下来一个正整数Q，表示操作的个数. 接下来Q行每行若干个整数。如果第一个数是1，后接3个正整数a,b,X，表示在区间[a,b]内每个数增加X,如果是2，后面跟1个整数i, 表示询问第i个位置的数是多少。

输出描述 Output Description

对于每个询问输出一行一个答案

样例输入 Sample Input

3

1

2

3

2

1 2 3 2

2 3

样例输出 Sample Output

5

数据范围及提示 Data Size & Hint

数据范围

1<=n<=100000

1<=q<=100000 

线段树：区间增减 与 单点查询。

但是不知道为什么，有一组则是数据运行时输出结果正常，但评测时，却不正确。求解答

输入数据 (显示前20行)

　　5 2 3 1 4 5 2 1 1 1 1 2 1

你的答案

　　< 1

正确答案

　　> 3

本题来自codevs 1081 

#include<iostream>
#include<cstdio>
#define lson l , m , rt << 1
#define rson m+1, r , rt << 1 | 1
#define LL long long  
using namespace std;

const int N = 100010 ;

int add[N<<2];
int sum[N<<2];
int n,q;

void pushup(int rt)
{
    sum[rt] = sum[rt<<1] + sum[rt<<1|1] ;
}
void pushdown(int rt,int m)  //rt线段,m长度 
{
    if (add[rt])
    {  
        add[rt<<1] += add[rt];  
        add[rt<<1|1] += add[rt];  
        sum[rt<<1] += add[rt] * (m - (m >> 1));  
        sum[rt<<1|1] += add[rt] * (m >> 1);  
        add[rt] = 0;  
    }  
}
void build(int l,int r,int rt) 
{  
    add[rt] = 0;  
    if (l == r) 
    {  
           scanf("%lld",&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) //[L,R]的区间加上c 
{  
    if (L <= l && r <= R)
    {
        add[rt] += c;
        sum[rt] += (LL)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);
} 
LL query(int x,int l,int r,int rt)
{  
    if (x==l && x==r)
    {
        return sum[rt];  
    }
    pushdown(rt , r - l + 1);  
    int m = (l + r) >> 1;  
    if (x <= m) return query(x , lson);  
    if (m < x) return query(x , rson);  
}
int main()
{
    scanf("%d",&n);
    build(1,n,1);
    scanf("%d",&q);
    while(q--)
    {
        int a,b,x,y;
        scanf("%d",&a);
        if(a==1)
        {
            scanf("%d%d%d",&x,&y,&b);
            update(x,y,b,1,n,1);
        }
        else
        {
            scanf("%d",&b);
            printf("%lld
",query(b,1,n,1));
        }
    }
    return 0;
}