洛谷P3368 【模板】树状数组 2 题解

题目描述

如题，已知一个数列，你需要进行下面两种操作：

1. 将某区间每一个数数加上 x；

2. 求出某一个数的值。

输入格式

第一行包含两个整数 N、M，分别表示该数列数字的个数和操作的总个数。

第二行包含 N 个用空格分隔的整数，其中第 i 个数字表示数列第 i 项的初始值。

接下来 M 行每行包含 2 或 4个整数，表示一个操作，具体如下：

操作 1： 格式：1 x y k 含义：将区间 [x,y] 内每个数加上 k；

操作 2： 格式：2 x 含义：输出第 x 个数的值。

输出格式

输出包含若干行整数，即为所有操作 2 的结果。

AC代码：

#include<bits/stdc++.h>
#define ll long long
using namespace std;

ll n,m,c[5000001];

ll lowbit(ll x)
{
    return x&(-x);
}

void change(ll k,ll v)
{
    while (k<=n)
    {
        c[k]+=v;
        k+=lowbit(k);
    }
}

void change_range(ll l,ll r,ll t)
{
    change(l,t);
    change(r+1,-t);
}

ll getsum(ll k)
{
    int ans=0;
    while (k>0)
    {
        ans+=c[k];
        k-=lowbit(k);
    }
    return ans;
}

int main()
{
    scanf("%lld%lld",&n,&m);
    ll t;
    for (int i=1;i<=n;i++)
    {
        scanf("%lld",&t);
        change_range(i,i,t);
    }
    while (m--)
    {
        ll op,x,y,k;
        scanf("%lld",&op);
        if (op==1)
        {
            scanf("%lld%lld%lld",&x,&y,&k);
            change_range(x,y,k);
        }
        if (op==2)
        {
            scanf("%lld",&x);
            cout<<getsum(x)<<endl;
        }
    }
    return 0;
}