hdu 1166 敌兵布阵(数据结构:树状数组||线段树)

简单入门的树状数组题目

对应典型的树状数组操作

对照白书的写法:

#include <stdio.h>
#include <string.h>
#define MAXN 50010

int a[MAXN+100], c[MAXN+100];

int lowbit(int x) {
return x&(-x);
}

int sum(int x) {
int sum = 0;
while(x > 0) {
sum += c[x];
x -= lowbit(x);
}
return sum;
}

void add(int x, int d) {
while(x <= MAXN) {
c[x] += d;
x += lowbit(x);
}
}

int main(void) {
int T, t, i, j, N;
char str[100];
scanf("%d", &T);
for(t=1; t<=T; ++t) {
scanf("%d", &N);

memset(c, 0, sizeof(c));

for(i=1; i<=N; ++i) {
scanf("%d", &a[i]);
for(j=i-lowbit(i)+1; j<=i; ++j)
c[i] += a[j];
}
printf("Case %d:\n", t);
while(scanf("%s", str) && strcmp(str, "End")) {
scanf("%d%d", &i, &j);
switch(str[0]) {
case 'A':
add(i, j); break;
case 'S':
add(i, -j); break;
case 'Q':
printf("%d\n", sum(j)-sum(i-1)); break;
}
}
}
}

看了别人的代码,a数组并不是必要的

可以在输入的过程中构造c数组,而不是等a数组输入完成后再构造c数组

代码如下:

#include <stdio.h>
#include <string.h>
#define MAXN 50010

int a, c[MAXN+100];

int lowbit(int x) {
return x&(-x);
}

int sum(int x) {
int sum = 0;
while(x > 0) {
sum += c[x];
x -= lowbit(x);
}
return sum;
}

void add(int x, int d) {
while(x <= MAXN) {
c[x] += d;
x += lowbit(x);
}
}

int main(void) {
int T, t, i, j, N;
char str[100];
scanf("%d", &T);
for(t=1; t<=T; ++t) {
scanf("%d", &N);

memset(c, 0, sizeof(c));

for(i=1; i<=N; ++i) {
scanf("%d", &a);
add(i, a);//等价于上面代码中c的构造公式
}
printf("Case %d:\n", t);
while(scanf("%s", str) && strcmp(str, "End")) {
scanf("%d%d", &i, &j);
switch(str[0]) {
case 'A':
add(i, j); break;
case 'S':
add(i, -j); break;
case 'Q':
printf("%d\n", sum(j)-sum(i-1)); break;
}
}
}
}

学习线段树之后觉得线段树较树状数组易于理解

下面是线段树的代码:

#include <cstdio>
#include <iostream>
#include <algorithm>
#define MAXN 50010
#define LL long long
using namespace std;

int sum[MAXN<<2];

void pushUp(int rt) {//当前根对应的值为左右孩子值之和
sum[rt] = sum[rt<<1]+sum[rt<<1|1];
}

void build(int l, int r, int rt) {//递归建树过程
if(l == r) {
scanf("%d", &sum[rt]);
return ;
}
int m = (l+r) >> 1;
build(l, m, rt<<1);
build(m+1, r, rt<<1|1);
pushUp(rt);
}

void update(int p, int add, int l, int r, int rt) {//单点更新,令p点对应值+add, l r表示左右边界, rt表示根节点
if(l == r) {
sum[rt] += add;
return ;
}
int m = (l+r) >> 1;
if(p <= m)
update(p, add, l, m, rt<<1);
else update(p, add, m+1, r, rt<<1|1);
pushUp(rt);
}

int query(int L, int R, int l, int r, int rt) {//统计L R之间的叶节点之和,l r表示左右边界, rt表示根节点
if(L<=l && r<=R)
return sum[rt];
int m = (l+r) >> 1;
int ret = 0;
if(L <= m)
ret += query(L, R, l, m, rt<<1);
if(R > m)
ret += query(L, R, m+1, r, rt<<1|1);
return ret;
}

int main(void) {
int T, n;
scanf("%d", &T);
for(int t=1; t<=T; ++t) {
printf("Case %d:\n", t);
scanf("%d", &n);
build(1, n, 1);
char op[10];
while(scanf("%s", op)) {
if(op[0] == 'E')
break;
int a, b;
scanf("%d%d", &a, &b);
if(op[0] == 'Q')
printf("%d\n", query(a, b, 1, n, 1));
else if(op[0] == 'S')
update(a, -b, 1, n, 1);
else update(a, b, 1, n, 1);
}
}
return 0;
}