线段树与树状数组

线段树是一种将数组而分成若干段的二分查找树，通过段记录段修改来更新数据来求得答案的一种数据结构算法。

题目：

http://acm.hdu.edu.cn/showproblem.php?pid=1166

思路：

线段树水题。

AC代码：

#include<cstdio>
#include<cstring>
using namespace std;
const int M = 50000 + 10;
struct Tree
{
    int l, r, v;
}tree[3 * M];
int value[M];
void build(int a, int b, int dx)//建树，不断二分
{
    tree[dx].l = a;
    tree[dx].r = b;
    if(tree[dx].r == tree[dx].l)
        tree[dx].v = value[a];
    else 
    {
        int mid = (tree[dx].l + tree[dx].r) >> 1;
        build(a, mid, dx + dx);
        build(mid + 1, b, dx + dx + 1);
        tree[dx].v = tree[dx + dx].v + tree[dx + dx + 1].v;
    }
}
void update(int a, int b, int dx)//从上至下 更新叶子的值
{
    tree[dx].v += b;
    if(tree[dx].r != tree[dx].l)
    {
        int mid = (tree[dx].l + tree[dx].r) >> 1;
        if(a <= mid)
            update(a, b, dx + dx);
        else
            update(a, b, dx + dx + 1);
    }
}
int query(int a, int b, int dx)//获取a到b范围的叶子值总和，将其分成段记录，如n = 8，1～6取1～4 + 5～6
{
    if(tree[dx].l == a && tree[dx].r == b)
        return tree[dx].v;
    int mid = (tree[dx].l + tree[dx].r) >> 1;
    if(b <= mid)
        return query(a, b, dx + dx);
    else if(a > mid)
        return query(a, b, dx + dx + 1);
    else
        return query(a, mid, dx + dx) + query(mid + 1, b, dx + dx + 1);
}
main()
{
    int T;
    scanf("%d", &T);
    int t = T;
    while(T--)
    {
        int n;
        scanf("%d", &n);
        for(int i = 1; i <= n; i++)
            scanf("%d", &value[i]);
        build(1, n, 1);
        char str[10];
        int a, b;
        getchar();
        printf("Case %d:\n", t - T);
        while(~scanf("%s", str))
        {
            if(str[0] == 'Q')
            {
                scanf("%d %d", &a, &b);
                getchar();
                int res = query(a, b, 1);
                printf("%d\n", res);
            }
            else if(str[0] == 'A')
            {
                scanf("%d %d", &a, &b);
                getchar();
                update(a, b, 1);
            }
            else if(str[0] == 'S')
            {
                scanf("%d %d", &a, &b);
                getchar();
                update(a, -b, 1);
            }
            else 
                break;
        }
    }
}

树状数组是线段树的缩减，在它的结构中线段树中重复的部分被删去了，所以代码更加简洁，速度更快，但是功能也有缩减。

如图所示，数组C为一个或多个数组A的值的和。将数组C的下标转为二进制，如果最后一位是1，则该位置的数只存有一个A中元素。功能是查询段数据总和，但是每次只能更新一个节点的值。

同样的题，树状数组AC代码：

#include<stdio.h>
#include<string.h>
#define M 50010
#define l(x) x & (-x)
int nums[M], n;
void add(int a, int x)//更新树状数组下标为a的值
{
	while(a <= n)
	{
		nums[a] += x;
		a += l(a);
	}
}
int sum(int a)//计算原数组中1到a的所有数的和
{
	int total = 0;
	while(a > 0)
	{
		total += nums[a];
		a -= l(a);
	}
	return total;
}
main()
{
	int T;
	scanf("%d", &T);
	for(int i = 1; i <= T; i++)
	{
		memset(nums, 0, sizeof(nums));
		scanf("%d", &n);
		for(int j = 1; j <= n; j++)
		{
			int v;
			scanf("%d", &v);
			add(j, v);
		}
		char str[20];
		printf("Case %d:\n", i);
		while(scanf("%s", str) && str[0] != 'E')
		{
			if(str[0] == 'Q')
			{
				int a, b;
				scanf("%d %d", &a, &b);
				printf("%d\n", sum(b) - sum(a - 1));
			}
			else if(str[0] == 'A')
			{
				int a, b;
				scanf("%d %d", &a, &b);
				add(a, b);
			}
			else if(str[0] == 'S')
			{
				int a, b;
				scanf("%d %d", &a, &b);
				add(a, -b);
			}
		}
	}
}