【树状数组】数列操作

2 篇文章 0 订阅

题目描述
给定n个数列,规定有两种操作,一是修改某个元素,二是求子数列[A,B]的连续和。数列的元素个数最多10万个,询问操作最多10万次。

输入
第一行2个整数n,m(n表示输入n个数列,m表示有m个操作)
第二行输入n个数列。
接下来M行,每更好行有三个数k,a,b(k=0表示求子数列[a,b]的和,k=1表示第a个数列加b)

输出
输出若干行数字,表示每次K=0时对应输出一个子数列[a,b]的连续和。

样例输入
10 5
1 2 3 4 5 6 7 8 9 10
1 1 5
0 1 3
0 4 8
1 7 5
0 4 8
样例输出
11
30
35

思路:建立一个树状数组进行求解。树状数组板子题

代码:

#include<bits/stdc++.h>
#define ll long long
using namespace std;
ll a[100005],c[100005],n,k;
ll lowbit(ll x)//
{
    return x&(-x);//获取两个相关位置的下标差值
}
void update(int i, ll k)//在a[i]上加k
{
    while(i<=n)
    {
        c[i]+=k;//相关的前缀和加k
        i+=lowbit(i);//找到下一个相关位置的下标
    }
}
ll sum(int i)//取得1到i的前缀和
{
    ll ans=0;//累加
    while(i>0)
    {
        ans+=c[i];
        i-=lowbit(i);
    }
    return ans;
}
int main()
{
    scanf("%d %d",&n,&k);
    ll q,qq,qqq;
    for(int i = 1; i <= n; i++)
    {
        scanf("%lld",&a[i]);
        update(i,a[i]);//用函数将相关的c加上值
    }
    for(int i = 1; i <= k; i++)
    {
        scanf("%lld %lld %lld",&q,&qq,&qqq);
        if(!q) printf("%lld\n",sum(qqq)-sum(qq-1));
        else update(qq,qqq);
    }
    return 0;
}

  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值