这是一道线段树的模板题。给定一个数列,有两种操作:1.给定区间[l,r],将区间内的所有数加上x。2.给定i,查询a[i]的值。
解题思路:使用线段树维护区间和,对于操作1,在线段树上对应的区间上打标记,表示区间内所有数都要加上x。对于操作2,在线段树上查询对应的叶子节点的值。
代码如下:
```C #include using namespace std; const int N = 1e6 + 10; int n, m, a[N], tr[N << 2];
void pushup(int x) { tr[x] = tr[x << 1] + tr[x << 1 | 1]; }
void build(int x, int l, int r) { if (l == r) { tr[x] = a[l]; return; } int mid = (l + r) >> 1; build(x << 1, l, mid), build(x << 1 | 1, mid + 1, r); pushup(x); }
void update(int x, int l, int r, int ql, int qr, int k) { if (ql <= l && qr >= r) { tr[x] += k * (r - l + 1); return; } int mid = (l + r) >> 1; if (ql <= mid) update(x << 1, l, mid, ql, qr, k); if (qr > mid) update(x << 1 | 1, mid + 1, r, ql, qr, k); pushup(x); }
int query(int x, int l, int r, int ql, int qr) { if (ql <= l && qr >= r) return tr[x]; int mid = (l + r) >> 1, ans = 0; if (ql <= mid) ans += query(x << 1, l, mid, ql, qr); if (qr > mid) ans += query(x << 1 | 1, mid + 1, r, ql, qr); return ans; }
int main() { scanf("%d%d", &n, &m); for (int i = 1; i <= n; i++) scanf("%d", &a[i