1 10 1 2 3 4 5 6 7 8 9 10 Query 1 3 Add 3 6 Query 2 7 Sub 10 2 Add 6 3 Query 3 10 End
Case 1: 6 33 59
学习下线段树。这题是更新点询问区间
代码:
#include <stdio.h> #include <string.h> const int N = 50005; int t, n, num; struct Tree { int l, r, value; }st[3 * N]; void build(int l, int r, int id) { if (l != r) { int mid = (l + r) / 2; build(l, mid, 2 * id); build(mid + 1, r, 2 * id + 1); } st[id].l = l; st[id].r = r; st[id].value = 0; } void Insert(int l, int r, int value, int id) { if (st[id].l != st[id].r) { int mid = (st[id].l + st[id].r) / 2; if (l <= mid && r > mid) {Insert(l, mid, value, 2 * id); Insert(mid + 1, r, value, 2 * id + 1);} else if (l <= mid && r <= mid) {Insert(l, r, value, 2 * id);} else {Insert(l, r, value, 2 * id + 1);} } st[id].value += value; } int Query(int l, int r, int id) { if (l == st[id].l && r == st[id].r) return st[id].value; int mid = (st[id].l + st[id].r) / 2; if (l <= mid && r > mid) {return Query(l, mid, 2 * id) + Query(mid + 1, r, 2 * id + 1);} else if (l <= mid && r <= mid) {return Query(l, r, 2 * id);} else {return Query(l, r, 2 * id + 1);} } void init() { scanf("%d", &n); build(1, n, 1); for (int i = 1; i <= n; i++) { scanf("%d", &num); Insert(i, i, num, 1); } } void solve() { char s[10]; int a, b; while (~scanf("%s", &s) && s[0] != 'E') { scanf("%d%d", &a, &b); if (s[0] == 'Q') printf("%d\n", Query(a, b, 1)); else if (s[0] == 'A') Insert(a, a, b, 1); else Insert(a, a, -b, 1); } } int main() { int cas = 0; scanf("%d", &t); while (t--) { init(); printf("Case %d:\n", ++cas); solve(); } return 0; }