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

树状数组区间修改

作者头像
用户2965768
发布2019-03-15 14:22:19
8110
发布2019-03-15 14:22:19
举报
文章被收录于专栏:wymwym
代码语言:javascript
复制
//c2[n] = (n-1)*c1[n];
//sum(1,k)=k*(c1(1)+c1(2)+c1(3)+…+c1(k))-(0*c1*(1)+1*c1(2)+2*c1(3)+…+(k-1)*c1(k))。
#include <bits/stdc++.h>
#define ll long long
#define LEN 1000000
using namespace std;
ll c1[LEN],c2[LEN],a[LEN];
ll lowbit(ll k)
{
	return k&(-k);
}
void update(ll *q,ll k,ll v)
{
	while(k<=LEN)
	{
		q[k]+=v;
		k+=lowbit(k);
	}
}
ll getsum(ll *q,ll k)//求c1或c2前k项和
{
	ll re = 0;
	while(k)
	{
		re+=q[k];
		k-=lowbit(k);
	}
	return re;
}
ll sum(ll k)//求数列前k项和
{
	ll ans1,ans2;
	ans1 = k*getsum(c1,k);
	ans2 = getsum(c2,k);
	return ans1-ans2;
}
int main()
{
	ll n,q;
	scanf("%lld %lld",&n,&q);
	for(ll i = 1; i<=n ;i++)
		{
			scanf("%lld",&a[i]);
			update(c1,i,a[i]-a[i-1]);
			update(c2,i,(i-1)*(a[i]-a[i-1]));
		}
	while(q--)
	{
		ll tp,l,r,v;
		scanf("%lld",&tp);
		if(tp==0)
		{
			scanf("%lld%lld%lld",&l,&r,&v);
			update(c1,l,v);
			update(c1,r+1,-v);
			update(c2,l,(l-1)*v);
			update(c2,r+1,r*(-v));
		}else
		{
			scanf("%lld %lld",&l,&r);
			printf("%lld\n",sum(r)-sum(l-1));
		}
	}
	
	return 0;
}
本文参与 腾讯云自媒体分享计划,分享自作者个人站点/博客。
原始发表:2019年03月04日,如有侵权请联系 cloudcommunity@tencent.com 删除

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

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

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

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