n개의 정수로 이루어진 수열을 입력받은 후, 구간합을 출력한 후, 인덱스 값을 바꾸는 연산을 수행해야하는 문제. 세그먼트 트리를 이용하여 풀었다. 아직은 이해도가 부족하다 생각된다.
\nimport sys\nimport math\ninput = sys.stdin.readline\n\ndef init(node, start, end):\n if start == end:\n tree[node] = l[start]\n return tree[node]\n\n mid = (start + end) // 2\n temp = init(node*2, start, mid) + init(node*2+1, mid+1, end)\n tree[node] = temp\n return tree[node]\n\ndef query(node, start, end, left, right):\n if start > right or end < left:\n return 0\n if start >= left and end <= right:\n return tree[node]\n\n mid = (start + end) // 2\n temp = query(node*2, start, mid, left, right) + query(node*2+1, mid+1, end, left, right)\n return temp\n\ndef update(node, start, end, index, diff):\n if index < start or index > end:\n return 0\n\n tree[node] += diff\n if start != end:\n mid = (start + end) // 2\n update(node*2, start, mid, index, diff)\n update(node*2+1, mid+1, end, index, diff)\n\n\nn, q = map(int, input().split())\nl = list(map(int, input().split()))\n\n# 세그먼트 트리 크기\nheight = math.ceil(math.log2(n))\nsize = 2 ** (height + 1)\ntree = [0 for _ in range(size)]\ninit(1, 0, n-1)\n\nfor _ in range(q):\n left, right, cindex, cvalue = map(int, input().split())\n if right < left:\n left, right = right, left\n print(query(1, 0, n-1, left-1, right-1))\n\n diff = cvalue - l[cindex-1]\n l[cindex-1] = cvalue\n update(1, 0, n-1, cindex-1, diff)