洛谷3368(【模板】树状数组 2)(树状数组+差分优化)

题目链接:https://www.luogu.org/problemnew/show/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

正常情况下用树状数组做是会T掉的,但是好像听说一些巨佬可以靠着O2优化直接A掉,我也试了好多次但是真的。。。。。。每次都是那样。话说我代码哪里打丑了吗

这题的标准解法应该就是树状数组+差分优化了(表示不会线段树QWQ)

至于差分就不多讲了,就是将原来的数组用来存当前元素减去前一个元素
然后更新的时候也有点变化,只要更新区间第一个点区间最后一个点后的一个点就好了。

AC代码如下:

#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
const int maxn=500001;
int c[maxn];
int n,m;
inline int lowbit(int x){
    return x&(-x);
}

inline void update(int x,int y){
    for(register int i=x;i<=n;i+=lowbit(i))
        c[i]+=y;
}

inline int sum(int x){
    int sum=0;
    for(register int i=x;i>0;i-=lowbit(i))
        sum+=c[i];
    return sum;
}

int main(){
    scanf("%d %d",&n,&m);
    register int k,x,y=0;
    for(register int i=1;i<=n;++i){
        //scanf("%d",&a[i]);
        scanf("%d",&x);
        update(i,x-y);
        y=x;
    }
    for(register int i=1;i<=m;++i){
        scanf("%d",&k);
        if(k==1){
            int b,c,d;
            scanf("%d %d %d",&b,&c,&d);
                update(b,d);
                update(c+1,-d);
        }
        if(k==2){
            int b;
            scanf("%d",&b);
            printf("%d\n",sum(b));
        }
    }
    return 0;
}

不得不说真的是快了好多。。。快了4秒左右。。。。。看来真的是代码打丑了。。。(逃~~)

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值