前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >Preprefix sum 差分+树状数组

Preprefix sum 差分+树状数组

作者头像
用户2965768
发布2019-04-18 14:15:01
4200
发布2019-04-18 14:15:01
举报
文章被收录于专栏:wym

 题意:求前缀和的前缀和,并有修改值操作。

Si = S(i-1)  + a[x]变为Si = S(i-1)  +y。 修改值就是y - a[x]

这里每个a[i]的值就是差分数组,一个树状数组维护就行。

  S1+S2+S3+....+Si

=a1+a1+a2+a1+a2+a3+.....+a1+a2+...+ai

=a1*i+a2*(i-1)+...+ai

=i*(a1+a2+...+ai) - (a2+a3*2+...+(i-1)*ai)

代码语言:javascript
复制
#include <bits/stdc++.h>
using namespace std;
#define ll long long
ll c[100005],c2[100005];
ll a[100005];
ll n,m; 
#define lowbit(x) x&-x
void update(ll pos,ll x)
{
    for(register int i=pos;i<=n;i+=lowbit(i))
        c[i]+=x,c2[i]+=(pos-1)*x;
}
ll getsum(ll pos)
{
    register ll res=0;
    for(register int i=pos;i;i-=lowbit(i))
        res+=c[i]*(pos)-c2[i];
    return res;
}
int main()
{

	scanf("%lld %lld",&n,&m);
	for(int i=1;i<=n;i++){
		cin>>a[i]; 
		update(i,a[i]);
	}
	ll y,x;
	while(m--){
		string s;
		cin>>s;
		if(s[0]=='Q'){
		cin>>x;
		printf("%lld\n",getsum(x));
		}else {
			cin>>x>>y;
			update(x,y-a[x]);
			a[x] = y;
		}
	}
	return 0;
}
本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2019年04月07日,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档