T1081 线段树练习 2 codevs

http://codevs.cn/problem/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

分类标签 Tags 点此展开 

  树状数组 线段树 树结构  

#include <iostream>
#define maxn 100005

using namespace std;

int n,m,x,a,b,X;
struct node
{
int now,dis,l,r,mid,flag;
}tree[maxn*4];

void tree_up(int now)
{
tree[now].dis=tree[now*2].dis+tree[now*2+1].dis;
return ;
}

void tree_build(int now,int l,int r)
{
tree[now].l=l,tree[now].r=r;
if(tree[now].l==tree[now].r)
{
cin>>tree[now].dis;
return ;
}
tree[now].mid=(tree[now].l+tree[now].r)/2;
tree_build(now*2,l,tree[now].mid);
tree_build(now*2+1,tree[now].mid+1,r);
tree_up(now);
}

void tree_down(int now)
{
tree[now*2].flag+=tree[now].flag;
tree[now*2].dis+=tree[now].flag*(tree[now*2].r-tree[now*2].l+1);
tree[now*2+1].flag+=tree[now].flag;
tree[now*2+1].dis+=tree[now].flag*(tree[now*2+1].r-tree[now*2+1].l+1);
tree[now].flag=0;    return ;
}

void tree_change(int now,int l,int r,int x)
{
if(tree[now].l==l&&tree[now].r==r)
{
tree[now].dis+=x*(r-l+1);
tree[now].flag+=x;
return ;
}
if(tree[now].flag) tree_down(now);
if(tree[now].mid>=r)    tree_change(now*2,l,r,x);
else if(tree[now].mid<l)    tree_change(now*2+1,l,r,x);
else
{
tree_change(now*2,l,tree[now].mid,x);
tree_change(now*2+1,tree[now].mid+1,r,x);
}
tree_up(now);
}

int tree_query(int now,int to)
{
if(tree[now].l==tree[now].r)    return tree[now].dis;
if(tree[now].flag) tree_down(now);
if(tree[now].mid>=to)    return tree_query(now*2,to);
else if(tree[now].mid<to)    return tree_query(now*2+1,to);
}

int main()
{
cin>>n;
tree_build(1,1,n);
cin>>m;
while(m--)
{
cin>>x;
if(x==1)
{
cin>>a>>b>>X;
tree_change(1,a,b,X);
}
else
{
cin>>X;
cout<<tree_query(1,X)<<endl;
}
}
return 0;
}

区间改变 单点查询