树状数组及例题动态求连续区间和 Java代码

树状数组

作用：
1、在某个位置上的数加上一个数
2、求前缀和
数组中 $c[x]$ 存的是 $(x-lowbit(x),x]$ 的和

注意： 树状数组下标必须从 $1$ 开始，否则会无限循环

例题 Acwing 1264. 动态求连续区间和

给定 $n$ 个数组成的一个数列，规定有两种操作，一是修改某个元素，二是求子数列 $[a,b]$的连续和。

输入格式

第一行包含两个整数 $n$和 $m$，分别表示数的个数和操作次数。

第二行包含 $n$ 个整数，表示完整数列。

接下来 $m$ 行，每行包含三个整数 $k,a,b$（ $k=0$，表示求子数列$[a,b]$的和；$k=1$，表示第 $a$ 个数加 $b$ ）。

数列从 $1$ 开始计数。

输出格式

输出若干行数字，表示 $k=0$ 时，对应的子数列 $[a,b]$ 的连续和。

数据范围

$1\le n\le 100000$,
$1\le m\le 100000$，
$1\le a\le b\le n$,
数据保证在任何时候，数列中所有元素之和均在 $int$ 范围内。

输入样例：

10 5
1 2 3 4 5 6 7 8 9 10
1 1 5
0 1 3
0 4 8
1 7 5
0 4 8

输出样例：

11
30
35

Java代码快读快写版本

import java.io.*;
class Main
{
    public static int[] a = new int[100010];    //原数组
    public static int[] tr = new int[100010];   //树状数组
    public static int n;
    public static void main(String[] args) throws IOException
    {
        StreamTokenizer in = new StreamTokenizer(new BufferedReader(new InputStreamReader(System.in)));
        PrintWriter out = new PrintWriter(new OutputStreamWriter(System.out));
        in.nextToken();
        n = (int)in.nval;
        in.nextToken();
        int m = (int)in.nval;
        for(int i=1;i<=n;i++)
        {
            in.nextToken();
            a[i]=(int)in.nval;
        }
        for(int i=1;i<=n;i++)    add(i,a[i]);
        int k,a,b;
        while(m-->0)
        {
            in.nextToken();
            k=(int)in.nval;
            in.nextToken();
            a=(int)in.nval;
            in.nextToken();
            b=(int)in.nval;
            
            if(k==1)    add(a,b);
            if(k==0)    out.println(query(b)-query(a-1));
        }
        out.flush();
    }
    public static int lowbit(int x)
    {
        return x & -x;
    }
    public static void add(int x,int y) //在x位置的数加上y
    {
        for(int i=x;i<=n;i+=lowbit(i))    tr[i]+=y;
    }
    public static int query(int x) //返回某位置的前缀和
    {
        int res=0;
        for(int i=x;i>=1;i-=lowbit(i))  res+=tr[i];
        return res;
    }
}

Java代码正常读写版本

这个代码也能 $AC$

import java.util.*;
class Main
{
    public static int[] a = new int[100010];    //原数组
    public static int[] tr = new int[100010];   //树状数组
    public static int n;
    public static void main(String[] args)
    {
        Scanner sc = new Scanner(System.in);
        n = sc.nextInt();
        int m = sc.nextInt();
        for(int i=1;i<=n;i++)
        {
            a[i]=sc.nextInt();
        }
        for(int i=1;i<=n;i++)    add(i,a[i]);
        int k,a,b;
        while(m-->0)
        {
            k=(int)sc.nextInt();
            a=(int)sc.nextInt();
            b=(int)sc.nextInt();
            
            if(k==1)    add(a,b);
            if(k==0)    System.out.println(query(b)-query(a-1));
        }
    }
    public static int lowbit(int x)
    {
        return x & -x;
    }
    public static void add(int x,int y) //在x位置的数加上y
    {
        for(int i=x;i<=n;i+=lowbit(i))    tr[i]+=y;
    }
    public static int query(int x) //返回某位置的前缀和
    {
        int res=0;
        for(int i=x;i>=1;i-=lowbit(i))  res+=tr[i];
        return res;
    }
}