数组计算机 单点更新和区间查询

7 篇文章 0 订阅
7 篇文章 0 订阅

bLue 有一个神器的机器,这个机器可以读入一个数组,并按照用户要求快速地进行数组的处理和计算,它支持如下两种操作:

  • 操作 1:把数组中第 p 个元素的值增加 v。
  • 操作 2:计算数组中 [l, r] 区间内所有数的和。

这个机器就是这么的神奇,但是 bLue 的计算机坏掉了,你能帮他修一下吗?

Input

输入数据有多组(数据组数不超过 20),到 EOF 结束。

对于每组数据:

  • 第 1 行输入一个整数 n (1 <= n <= 10^5),表示数组中元素的个数。
  • 第 2 行输入 n 个用空格隔开的整数 ai (1 <= ai <= 10^10),表示初始输入到计算机中的数组。
  • 第 3 行输入一个整数 q (1 <= q <= 50000),表示用户的操作次数。
  • 接下来 q 行,每行输入先输入 1 个整数,表示操作类型,根据不同的操作类型:
    • 如果类型为 1,则紧接着输入 2 个用空格隔开的整数 p (1 <= p <= n) 和 v (1 <= v <= 10^10),表示要把数组中第 p 个数的值增加 v。
    • 如果类型为 2,则紧接着输入 2 个用空格隔开的整数 l, r (1 <= l <= r <= n),表示要计算区间 [l, r] 内所有数的和(数组下标从 1 开始)。
Output

对于每组数据中的每次类型为 2 的操作,输出 1 行,包含一个整数,表示计算出的和。

Example Input
5
1 2 3 4 5
5
2 1 2
2 1 5
1 4 10
2 4 5
2 1 5
Example Output
3
15
19
25


这个应该是最最基本的线段树了吧,模板题,毕竟刚刚接触嘛~ 单点更新和区间查询

#include <iostream>  
#include <algorithm>  
#include <cstdio>  
#include <cstring>

using namespace std;  
const int N = 1e5;

struct tree
{
    int l, r;
    long long w;
}SegTree[N << 2];

long long v, ans;
int p, x, y;

void Build(int l, int r, int k)
{
    SegTree[k].l = l;
    SegTree[k].r = r;
    if(l == r)
    {
        scanf("%lld", &SegTree[k].w);
        return;
    }
    int mid = (l + r) >> 1;
    Build(l, mid, k << 1);
    Build(mid + 1, r, k << 1 | 1);
    SegTree[k].w = SegTree[k << 1].w + SegTree[k << 1 | 1].w;
}
void Update(int k)
{
    if(SegTree[k].l == SegTree[k].r)
    {
        SegTree[k].w += v;
        return;
    }
    int mid = (SegTree[k].l + SegTree[k].r) >> 1;
    if(p <= mid)
        Update(k << 1);
    else
        Update(k << 1 | 1);
    SegTree[k].w = SegTree[k << 1].w + SegTree[k << 1 | 1].w;
}
void Sum(int k)
{
    if(x <= SegTree[k].l && y >= SegTree[k].r)
    {
        ans += SegTree[k].w;
        return;
    }
    int mid = (SegTree[k].l + SegTree[k].r) >> 1;
    if(x <= mid)
        Sum(k << 1);
    if(y > mid)
        Sum(k << 1 | 1);
}
int main()  
{         
    int n, q, m;
    while(~scanf("%d", &n))
    {
        Build(1, n, 1);
        scanf("%d", &q);
        while(q--)
        {
            scanf("%d", &m);
            if(m == 1)
            {
                scanf("%d%lld", &p, &v);
                Update(1);
            }
            else
            {
                scanf("%d%d", &x, &y);
                ans = 0;
                Sum(1);
                printf("%lld\n", ans);
            }
        }
    }
    return 0;  
}  


当然,这道题用树状数组也能做,而且比较简单

#include <iostream>
#include <cstdio>
#include <cstring>
#define lowbit(x) (x & (-x))

const int N = 1e5 + 5;

long long c[N];
int n;

void Update(int p, long long v)
{
    for(int i = p; i <= n; i += lowbit(i))
        c[i] += v;
}
long long GetSum(int k)
{
    long long sum = 0;
    for(int i = k; i > 0; i -= lowbit(i))
        sum += c[i];
    return sum;
}
int main()
{
    int q, t, p, l, r;
    long long x, v;
    while(~scanf("%d", &n))
    {
        memset(c, 0, sizeof(c));
        for(int i = 1; i <= n; i++)
        {
            scanf("%lld", &x);
            Update(i, x);
        }
        scanf("%d", &q);
        while(q--)
        {
            scanf("%d", &t);
            if(t == 1)
            {
                scanf("%d%lld", &p, &v);
                Update(p, v);
            }
            else
            {
                scanf("%d%d", &l, &r);
                printf("%lld\n", GetSum(r) - GetSum(l - 1));
            }
        }
    }
    return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值