您的位置:首页 > 其它

【模板】树状数组 2

2017-01-21 18:15 253 查看
区间修改+单点查询

在[x,y]区间上加k

另开一个数组

在x处+k,y+1处-k

该数组对应位置的前缀和即为改数字的变化量

#include<cstdio>
using namespace std;

int n,//点数
m,//操作数
C[500003],//树状数组
//树状数组用以求前缀和
a[500003];//原数列

void add(int loc,int value)
{//加标记
for(int i=loc;i<=n;i+=i&(-i))
C[i]+=value;
}

int query(int loc)
{//求前缀和/变化量
int ans=0;
for(int i=loc;i>0;i-=i&(-i))
ans+=C[i];
return ans;
}

int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;++i)
scanf("%d",&a[i]);
//读入原数列
for(int i=1;i<=m;++i)
{
int t;
scanf("%d",&t);
if(t==1)
{
//在[x,y]区间上加k
int x,y,k; scanf("%d%d%d",&x,&y,&k);
add(x,k);add(y+1,-k);//打标记
}
else
{
//查询第k个数
int x; scanf("%d",&x);
printf("%d\n",a[x]+query(x));
}
}
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: