算法——树状数组，线段树

一.树状数组定义：核心是用来解决两个问题

（1）单点修改

（2）区间查询

 2.计算前缀和的写法

3.当给原数组A[ ] 的某个位置加上一个数字V，则数组数组的改变如下

 

树状数组的三个核心函数如下：

int lowbit(int x)
{
    return x & -x;
}
void add(int x,int v)
{
    for(int i = x ;i <= n; i += lowbit(i)) tr[i] += v;
}
int query(int x)
{
    int res = 0;
    for(int i = x; i>0;i -= lowbit(i)) res += tr[i];
    return res;
}

可以使用add函数进行树状数组的初始化，默认当前树状数组的值都为0，然后在每一个位置i上加上a[i]即可

例题：动态求连续区间和

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

输入格式

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

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

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

数列从 11 开始计数。

输出格式

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

数据范围

1≤n≤100000,
1≤m≤100000，
1≤a≤b≤n1,
数据保证在任何时候，数列中所有元素之和均在 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

代码实现：

#include<iostream>
#include<cstdio>
#include<algorithm>

const int N = 100010;
int n,m;
int a[N],tr[N];

//三个函数
int lowbit(int x)
{
    return x & -x;
}
void add(int x,int v)
{
    for(int i = x ;i <= n; i += lowbit(i)) tr[i] += v;
}
int query(int x)
{
    int res = 0;
    for(int i = x; i>0;i -= lowbit(i)) res += tr[i];
    return res;
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    for(int i=1;i<=n;i++) add(i,a[i]);
    
    while(m--)
    {
        int k,x,y;
        scanf("%d%d%d",&k,&x,&y);
        if(k==0) printf("%d\n",query(y)-query(x-1));
        else{
           add(x,y);
        }
    }
    return 0;
}

例题2：数星星

天空中有一些星星，这些星星都在不同的位置，每个星星有个坐标。

如果一个星星的左下方（包含正左和正下）有 k 颗星星，就说这颗星星是 k 级的。

例如，上图中星星 5 是3 级的（1,2,4 在它左下），星星 2,4, 是 1 级的。

例图中有 1 个 0 级，2 个 1 级，1 个 2 级，1 个 3 级的星星。

给定星星的位置，输出各级星星的数目。

换句话说，给定 N 个点，定义每个点的等级是在该点左下方（含正左、正下）的点的数目，试统计每个等级有多少个点。

输入格式

第一行一个整数 N，表示星星的数目；

接下来 N 行给出每颗星星的坐标，坐标用两个整数 x,y 表示；

不会有星星重叠。星星按 y 坐标增序给出，y 坐标相同的按 x 坐标增序给出。

输出格式

N 行，每行一个整数，分别是 0 级，1 级，2 级，……，N−1 级的星星的数目。

数据范围

1≤N≤15000,
0≤x,y≤32000

输入样例：

5
1 1
5 1
7 1
3 3
5 5

输出样例：

1
2
1
1
0

代码实现：

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;

const int N = 32010;
int n,m;
int tr[N],leval[N];

int lowbit(int x)
{
    return x & -x;
}
void add(int x,int v)
{
    for(int i=x;i<N;i+=lowbit(i)) tr[i]+=v;
}
int query(int x)
{
    int res = 0;
    for(int i=x;i>0;i-=lowbit(i)) res += tr[i];
    return res;
}
int main()
{
    scanf("%d",&n);
    for(int i=0;i<n;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        x++;
        leval[query(x)]++;
        add(x,1);
    }
    for(int i=0;i<n;i++) printf("%d\n",leval[i]);
    return 0;
}