本文共 2541 字,大约阅读时间需要 8 分钟。
线段树是一个高效的数据结构,能够在O(log n)的时间复杂度内完成区间更新和查询操作。对于这个问题,我们需要处理两种操作:区间加法和区间求和。线段树的懒标记机制可以帮助我们高效地处理这些操作。
线段树结构:我们使用线段树来处理区间操作。每个节点表示一个区间,叶子节点对应数组中的单个元素。
懒标记:懒标记用于记录对子节点的更新操作,避免在每次查询时都遍历整个树,从而提高效率。
区间更新:对于“C a b c”操作,我们通过线段树更新区间[a, b]中的每个元素c,并使用懒标记记录这些更新操作。
区间查询:对于“Q a b”操作,我们通过线段树查询区间[a, b]的和,并返回结果。
输入处理:使用高效的输入读取方法,避免因读取速度慢而导致超时。
#include#include #include #include #include #include using namespace std;#define ll long long#define maxn 100002#define lson(k) k << 1#define rson(k) k << 1 | 1#define build(k, l, r) build(k, l, r)#define push_down(k, l, r) push_down(k, l, r)struct SegmentTree { ll *a, *s, *lazy; int size; SegmentTree(int n) : size(1) { while (size < n) size <<= 1; a = new ll[size << 1]; s = new ll[size << 1]; lazy = new ll[size << 1](); fill(a + size, a + size + n, 0); } void build(int k, int l, int r) { if (l == r) { s[k] = a[k + size]; return; } int m = lson(k); build(m, l, mid); build(rson(k), mid + 1, r); s[k] = s[m] + s[rson(k)]; } void push_down(int k, int l, int r) { if (lazy[k]) { lazy[lson(k)] += lazy[k]; lazy[rson(k)] += lazy[k]; int m = (l + r) >> 1; s[lson(k)] += (m - l) * lazy[k]; s[rson(k)] += (r - m) * lazy[k]; lazy[k] = 0; } } void update(int k, int l, int r, int ql, int qr, ll val) { push_down(k, l, r); if (ql > r || qr < l) return; if (ql <= l && r <= qr) { lazy[k] += val; s[k] += (r - l + 1) * val; return; } int m = (l + r) >> 1; update(lson(k), l, m, ql, qr, val); update(rson(k), m + 1, r, ql, qr, val); s[k] = s[lson(k)] + s[rson(k)]; } ll query(int k, int l, int r, int ql, int qr) { push_down(k, l, r); if (ql > r || qr < l) return 0; if (ql <= l && r <= qr) return s[k]; int m = (l + r) >> 1; return query(lson(k), l, m, ql, qr) + query(rson(k), m + 1, r, ql, qr); }};int main() { int t, n, q; scanf("%d %d", &t, &n); while (t--) { SegmentTree st(n); for (int i = 0; i < n; ++i) scanf("%lld", &st.a[size + i]); st.build(1, 1, n); for (int i = 0; i < q; ++i) { char op[5]; int a, b, c; scanf("%s", op); if (op[0] == 'Q') { scanf("%d %d", &a, &b); a = a < 1 ? 1 : a; b = b > n ? n : b; ll res = st.query(1, 1, n, a, b); printf("%lld\n", res); } else { scanf("%d %d %d", &a, &b, &c); a = a < 1 ? 1 : a; b = b > n ? n : b; st.update(1, 1, n, a, b, c); } } } return 0;}
线段树结构:使用数组a
存储原始数据,s
存储线段树节点的值,lazy
存储懒标记。
构建函数:递归构建线段树,合并左右子节点的值。
懒标记推送:将懒标记传播到子节点,更新子节点的值。
区间更新:递归更新指定区间的值,并使用懒标记记录更新操作。
区间查询:递归查询指定区间的和,并返回结果。
输入处理:读取输入数据,初始化线段树,处理每个操作并输出结果。
该实现确保了线段树的高效性,能够在较短的时间内处理大量操作,适用于大规模数据的区间操作问题。
转载地址:http://szrp.baihongyu.com/