hdu 1166 敌兵布阵(树状数组 | 线段树）

题目链接：hdu 1166 敌兵布阵

题目大意：略。

解题思路：

方法1：树状数组的水题。这篇题解很好，适合初学树状数组的人。题解连接

#include <stdio.h>
#include <string.h>

const int N = 50005;
const int M = 105;

int n, v
;

void add(int x, int val) {
while (x <= n) {
v[x] += val;
x += (x & (-x));
}
}

void init() {
memset(v, 0, sizeof(v));
int a;
scanf("%d", &n);
for (int i = 1; i <= n; i++) {
scanf("%d", &a);
add(i, a);
}
}

int sum(int x) {
int ans = 0;
while (x > 0) {
ans += v[x];
x -= (x & (-x));
}
return ans;
}

void solve() {
int x, y;
char order[M];

while (scanf("%s", order) == 1 && strcmp(order, "End") ) {
scanf("%d%d", &x, &y);
if (order[0] == 'A') add(x, y);
else if (order[0] == 'S') add(x, -y);
else printf("%d\n", sum(y) - sum(x-1));
}
}

int main() {
int cas;
scanf("%d", &cas);
for (int i = 1; i <= cas; i++) {
printf("Case %d:\n", i);
init();
solve();
}
return 0;
}

方法二：今天看了一下线段树，这题也可以用线段树做，更新点查询区间。

#include <stdio.h>
#include <string.h>

const int N = 50005;

struct node {
int l, r, v;
}s[N*3];

void build(int l, int r, int f) {
if (l != r) {
int mid = (l + r) / 2;
build(l, mid, f*2);
build(mid+1, r, f*2+1);
}
s[f].l = l; s[f].r = r; s[f].v = 0;
}

void insert(int l, int r, int val, int f) {
if (s[f].l != s[f].r) {
int mid = (s[f].l + s[f].r) / 2;
if (l <= mid && r > mid) { insert(l, mid, val, f*2); insert(mid+1, r, val, f*2+1); }
else if (l <= mid && r <= mid) insert(l, r, val, f*2);
else insert(l, r, val, f*2+1);
}
s[f].v += val;
}

int query(int l, int r, int f) {
if (l == s[f].l && r == s[f].r) return s[f].v;
int mid = (s[f].l + s[f].r) / 2;
if (l <= mid && r > mid) return query(l, mid, f*2) + query(mid+1, r, f*2+1);
else if (l <= mid && r <= mid) return query(l, r, f*2);
else return query(l, r, f*2+1);
}

void init () {
int n, a;
scanf("%d", &n);
build(1, n, 1);
for (int i = 1; i <= n; i++) {
scanf("%d", &a);
insert(i, i, a, 1);
}
}

void solve () {
int a, b;
char o[20];
while (scanf("%s", o), strcmp(o, "End")) {
scanf("%d%d", &a, &b);
if (o[0] == 'A') insert(a, a, b, 1);
else if (o[0] == 'S') insert(a, a, -b, 1);
else printf("%d\n", query(a, b, 1));
}
}

int main () {
int cas;
scanf("%d", &cas);
for (int i = 1; i <= cas; i++) {
printf("Case %d:\n", i);
init();
solve();
}
return 0;
}