一个简单的整数问题

bonely

于 2021-08-05 00:43:34 发布

阅读量87

点赞数

分类专栏：题解文章标签：数据结构

本文链接：https://blog.csdn.net/sjjzslhhh/article/details/119397294

版权

题解专栏收录该内容

171 篇文章 0 订阅

订阅专栏

本文介绍了一种利用树状数组（也称作二进制指数矩阵或 Fenwick Tree）来高效处理区间更新和单点查询的算法。在给定整数数组 a 和操作 m 的情况下，程序演示了如何通过 'C' 操作更新区间元素，以及通过 'Q' 操作查询单点累加和。代码中定义了 add 函数用于区间更新，ask 函数用于单点查询，从而实现 O(log n) 时间复杂度内的操作。

摘要由CSDN通过智能技术生成

题目:

题目链接:

题解:

树状数组

#include <bits/stdc++.h>
using namespace std;
long long a[100005],t[100005];
int n;
void add(int x, int k)
{
    for(; x <= n; x += x & -x) t[x] += k;
}
long long ask(int x)
{
    long long ans = 0;
    for(; x; x -= x & -x) ans += t[x];
    return ans;
}
int main()
{
    int m;
    cin>>n>>m;
    for(int i = 1; i <= n; i ++ ) cin>>a[i];
    while(m -- )
    {
        char op;
        cin>>op;
        if(op == 'C')
        {
            int L, R, d;
            cin>>L>>R>>d;
            add(L, d);
            add(R + 1, -d);
        }
        else 
        {
            int x;
            cin>>x;
            cout<<a[x] + ask(x)<<endl;
        }
    }
    return 0;
}