P3368 【模板】差分树状数组

题目描述
如题,已知一个数列,你需要进行下面两种操作:

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

2.求出某一个数的和

输入输出格式
输入格式:

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

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

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

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

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

输出格式:

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

输入输出样例
输入样例#1:
5 5
1 5 4 2 3
1 2 4 2
2 3
1 1 5 -1
1 3 5 7
2 4
输出样例#1:
6
10
说明
时空限制:1000ms,128M

数据规模:

对于30%的数据:N<=8,M<=10

对于70%的数据:N<=10000,M<=10000

对于100%的数据:N<=500000,M<=500000

样例说明:

在这里插入图片描述

故输出结果为6、10


推一篇博客:http://blog.csdn.net/qq_21841245/article/details/43956633

最主要的做法其实是差分数组

用树状数组维护数组的前缀和只是为了达到优化时间复杂度的目的

存板子:

#include <iostream>
#include <cmath>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <queue>
#include <string>
#include <stack>
#include <set>
#include <vector>
#include <map>
#define mod 1e9+7
#define inf 0x3f3f3f3f
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
typedef unsigned long long ull;
typedef long long ll;
using namespace std;

const int maxn = 250;

const int MAXN=5e5+10;
int n,m,k,x,y,z,a[MAXN],tree[MAXN];

int lowbit(int x)     //get the lowest "1" in x(binary form)    eg. lowbit(1100(2))=100(2)=4
{
    return x&(-x);
}

void add(int i,int x)
{
    while(i<=n)
    {
        tree[i]+=x;
        i+=lowbit(i);
    }
}

long long sum(int i)
{
    ll ans=0;
    while(i)
    {
        ans+=tree[i];
        i-=lowbit(i);
    }
    return ans;
}
int main()
{
    cin>>n>>m;
    for(int i=1; i<=n; i++)
        scanf("%d",&a[i]);
    for(int i=1; i<=n; i++)
        add(i,a[i]-a[i-1]);
    int flag,x,y,z,j;
    for(int i=1; i<=m; i++)
    {
        scanf("%d",&flag);
        if(flag==1) scanf("%d%d%d",&x,&y,&z),add(x,z),add(y+1,-z);
        if(flag==2) scanf("%d",&j),printf("%lld\n",sum(j));
    }

    return 0;
}

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值