Luogu P3368 【模板】树状数组 2 [区间修改-单点查询]

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的结果。

输入输出样例

输入样例#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

Solution

我们定义$c[i]$表示$a[i]-a[i-1]$就是$a[i]$和$a[i-1]$之间的距离
  而且规定$c[1]=a[1]$
那么很明显
  $a[i-1]+c[i]=a[i]$
我们把c放入树状数组e里面
## 注意e是树状数组!!!

查询

然后我们可以得出
$query(x)=sum^{i<=x}_{i=1}c[i]$
$        =c[1]+c[2]+...+c[x-1]+c[x]$
$        =(a[1])+(a[2]-a[1])+...+(a[x-1]-a[x-2])+(a[x]-a[x-1])$
$        =a[1]-a[1]+a[2]+...-a[x-2]+a[x-1]-a[x-1]+a[x]$
$        =a[x]$

初始插入

刚开始将$c[i]$(也就是$now-last$)插入树状数组就可以啦

修改

 

因为给$a[x]+z$后
$a[x]$和$a[x-1]$的距离增加了z ，于是我们要给$a[x]-a[x-1]$(即$c[x]$)加上$z$
因为给$a[y]+z$后
$a[y]$和$a[y+1]$的距离减少了z ，于是我们要给$a[y]-a[y+1]$(即$c[y+1]$)加上$-z$
那中间$x<i<y$不用处理？
当然，因为$c[i]$存的只是$a[i]和a[i-1]$的距离啊！

Codes

program no;
var

  n,m,i,now,last,c,x,y,z:Longint;
  e:array[1..500000] of Longint;

function lowbit(apple:Longint):Longint ;
begin
    lowbit:=apple and -apple;
end;

procedure add(x,a:Longint);
begin
    while x<=n do
     begin
        e[x]:=e[x]+a;
        x:=x+lowbit(x);
     end;
end;

function query(x:Longint):longint;
begin
    query:=0;
    while x>0 do
    begin
        query:=query+e[x];
            x:=x-lowbit(x);
    end;
end;

begin
   //assign(input,'1.in'); assign(output,'1.out');
   reset(input); rewrite(output);

    readln(n,m);
    for i:= 1 to n do
    begin
     read(now);
     add(i,now-last);
     last:=now;
    end;

    for i:= 1 to m do
    begin
        read(c,x);
        if c=1 then
          begin
            readln(y,z);
            add(x,z);
            add(y+1,-z);
          end
        else writeln(query(x));
    end;

   close(input);  close(output);
end.