HDU 1166 单点更新 区间求和(ZKW 线段树)

题目链接:HDU 1166





题解:没啥特殊的, zkw线段树直接在单点更新时向上更新就可



AC code:

//lrl's submission
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
using namespace std;

#define debug 0
#define inf 0x3f3f3f3f
#define M(a, b) memset(a, b, sizeof(a))

const int maxn = 50000 + 5;
int sum[maxn << 2], a, b;					//四倍空间
int n,M;

void build()
{
    scanf("%d", &n);
    for(M = 1; M <= n + 1; M *= 2);
    for(int i = M + 1; i <= M + n; i++)
    {
        scanf("%d", &sum[i]);								//初始化
    }
    for(int i = M - 1; i >= 1; i--)
    {
        sum[i] = sum[i << 1] + sum[i << 1|1];		//处理
    }
}

void query(int a, int b)
{
    a += M - 1;												//开区间
    b += M + 1;
    int ans = 0;
    while(b - a > 1)
    {
        if(a % 2 == 0) 									//操作
        {
            ans += sum[a + 1];
        }
        if(b % 2)
        {
            ans += sum[b - 1];
        }
        a >>= 1;
        b >>= 1;
    }

    printf("%d\n", ans);
}

void Add(int a, int b){
	for(sum[a += M] +=b, a >>= 1; a; a >>= 1){
		sum[a] = sum[a << 1] + sum[a << 1|1];
	}
}

void Sub(int a, int b){
	for(sum[a += M] -=b, a >>= 1; a; a >>= 1){
		sum[a] = sum[a << 1] + sum[a << 1|1];
	}
}

int main()
{
#if debug
    freopen("in.txt", "r", stdin);
#endif // debug
    int t;
    scanf("%d", &t);
    char s[10];
    for(int kase = 1; kase <= t; kase++)
    {
        printf("Case %d:\n", kase);
        M(sum, 0);
        build();
        while(~scanf("%s", s))
        {
        	if(s[0] == 'E') break;
            scanf("%d%d", &a, &b);
            if(s[0] == 'A')
            {
                Add(a, b);
            }
            if(s[0] == 'S')
            {
                Sub(a, b);
            }
            if(s[0] == 'Q')
            {
                query(a, b);
            }
        }
    }
    return 0;
}


你可能感兴趣的:(tree),HDU)