带标记的线段树 - Acwing： 一个简单的整数问题 Python题解

给定长度为 NN 的数列 AA，然后输入 MM 行操作指令。

第一类指令形如 C l r d，表示把数列中第 l∼rl∼r 个数都加 d。

第二类指令形如 Q x，表示询问数列中第 xx 个数的值。

对于每个询问，输出一个整数表示答案。

输入格式

第一行包含两个整数 NN 和 MM。

第二行包含 NN 个整数 A[i]A[i]。

接下来 MM 行表示 MM 条指令，每条指令的格式如题目描述所示。

输出格式

对于每个询问，输出一个整数表示答案。

每个答案占一行。

数据范围

1≤N,M≤1051≤N,M≤105,
|d|≤10000|d|≤10000,
|A[i]|≤109|A[i]|≤109

输入样例：

10 5
1 2 3 4 5 6 7 8 9 10
Q 4
Q 1
Q 2
C 1 6 3
Q 2

输出样例：

4
1
2
5

引用：242. 一个简单的整数问题 - AcWing题库

>> 算法打卡：区间操作问题。比较简单，采用带标记的线段树，接着实现树上标记下传，详细见码。 Python题解，后续更新。

import sys

def buildTree(k,l,r):
    if l == r:
        tree[k] = a[l]
        return
    mid = l + r >> 1
    buildTree(k << 1, l, mid)
    buildTree(k << 1 | 1, mid + 1, r)
    tree[k] = tree[k << 1] + tree[k << 1 | 1]

def SearCh(k,l,r,x,p):
    p += v[k]
    if l == r:
        return tree[k] + p
    mid = l + r >> 1
    if x <= mid:
        return SearCh(k << 1, l, mid, x, p)
    else:
        return SearCh(k << 1 | 1, mid + 1, r, x, p)

def add(k,l,r,x,y,z):
    if l == x and r == y:
        v[k] += z
        return
    tree[k] += z * (y - x + 1)
    m = l + r >> 1
    if y <= m:
        add(k << 1, l, m, x, y, z)
    else:
        if x > m:
            add(k << 1 | 1,m + 1, r, x, y, z)
        else:
            add(k << 1, l, m, x, m, z)
            add(k << 1 | 1, m + 1, r, m + 1, y, z)

def main():
    global tree,a,v
    n,m = map(int,sys.stdin.readline().split())
    a = [0] + list(map(int,sys.stdin.readline().split()))
    tree = [0] * (4 * n)
    v = [0] * (4 * n)
    buildTree(1,1,n)
    for i in range(m):
        tmp = list(input().split())
        tmps = list(map(int,tmp[1:]))
        if tmp[0] == "Q":
            print(SearCh(1,1,n,tmps[0],0))
        elif tmp[0] == "C":
            x,y,z = tmps
            add(1, 1, n, x, y, z)
            
            
main()