题目传送:A Simple Problem with Integers
思路:线段树,成段增减,区间求和,注意延迟标记需要累加,还有会爆int
AC代码:
#include <cstdio> #include <cstring> #include <iostream> #include <algorithm> #include <cmath> #include <queue> #include <stack> #include <vector> #include <map> #include <set> #include <deque> #include <cctype> #define LL long long #define INF 0x7fffffff using namespace std; const int maxn = 100005; LL sum[maxn << 2]; LL lazy[maxn << 2]; void pushdown(int rt, int m) { if(lazy[rt] != 0) { lazy[rt << 1] += lazy[rt];//延迟标记需累加,因为可能有别的区间在这里产生延迟标记 lazy[rt << 1 | 1] += lazy[rt]; sum[rt << 1] += (m - (m >> 1)) * lazy[rt];//区间加减更新 sum[rt << 1 | 1] += (m >> 1) * lazy[rt]; lazy[rt] = 0;//用完后记得置为0 } } void build(int l, int r, int rt) {//建树 lazy[rt] = 0;//延迟标记赋初值 if(l == r) { scanf("%I64d", &sum[rt]);//题目中说会爆int,用64位输入输出 return; } int mid = (l + r) >> 1; build(l, mid, rt << 1); build(mid + 1, r, rt << 1 | 1); sum[rt] = sum[rt << 1] + sum[rt << 1 | 1]; } void update(int L, int R, int c, int l, int r, int rt) {//更新 if(L <= l && r <= R) { sum[rt] += (LL)(r - l + 1) * c;//虽然这里不会爆int,但是看到乘法要特别注意会不会爆int lazy[rt] += c;//累加延迟标记 return; } pushdown(rt, r - l + 1); int mid = (l + r) >> 1; if(L <= mid) update(L, R, c, l, mid, rt << 1); if(R >= mid + 1) update(L, R, c, mid + 1, r, rt << 1 | 1); sum[rt] = sum[rt << 1] + sum[rt << 1 | 1]; } LL query(int L, int R, int l, int r, int rt) {//查询 if(L <= l && r <= R) { return sum[rt]; } pushdown(rt, r - l + 1); int mid = (l + r) >> 1; LL ret = 0; if(L <= mid) ret += query(L, R, l, mid, rt << 1); if(R >= mid + 1) ret += query(L, R, mid + 1, r, rt << 1 | 1); return ret; } int main() { int n, q; while(scanf("%d %d", &n, &q) != EOF) { build(1, n, 1); char op[10]; int a, b, c; for(int i = 0; i < q; i ++) { scanf("%s", op); if(op[0] == 'Q') { scanf("%d %d", &a, &b); printf("%I64d\n", query(a, b, 1, n, 1)); } else { scanf("%d %d %d", &a, &b, &c); update(a, b, c, 1, n, 1); } } } return 0; }
POJ - 3468 - A Simple Problem with Integers (线段树 - 成段更新)
原文地址:http://blog.csdn.net/u014355480/article/details/45727251