기존 구간합을 구하며, 특정 인덱스의 수를 바꾸는 세그먼트 트리문제에서 구간의 인덱스를 바꾸는 방식으로 바꾼 문제. 첫 접근은 단순히 업데이트 함수를 여러번 호출하면 될 것 같았으나 당연하게도 다른 방법이 존재하였다. lazy propagation이란 방법인데, 할 일을 나중으로 미루는 방법… 이다. 해당 글을 참고하여 풀었으나 아직도 어렵기만 하다. 다른 부분은 기존 세그먼트 트리 문제와 동일하며 query와 updaterange 함수 호출 시 updatelazy를 호출하여 tree의 크기와 같은 lazy배열을 이용하여 나중에 더할 값을 저장하는 연산을 수행한다. 세그먼트 트리는 감을 잡기 시작한 것 같으나 lazy propagation은 아직 멀은 것 같다.
\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 mid = (start + end) // 2\n tree[node] = init(node*2, start, mid) + init(node*2+1, mid+1, end)\n return tree[node]\n\ndef query(node, start, end, left, right):\n update_lazy(node, start, end)\n if left > end or right < start:\n return 0\n\n if left <= start and right >= end:\n return tree[node]\n\n mid = (start + end) // 2\n return query(node*2, start, mid, left, right) + query(node*2+1, mid+1, end, left, right)\n\ndef update_lazy(node, start, end):\n if lazy[node] != 0:\n tree[node] += (end-start+1) * lazy[node]\n if start != end:\n lazy[node*2] += lazy[node]\n lazy[node*2+1] += lazy[node]\n lazy[node] = 0\n\ndef update_range(node, start, end, left, right, diff):\n update_lazy(node, start, end)\n if left > end or right < start:\n return 0\n if left <= start and right >= end:\n tree[node] += (end-start+1) * diff\n if start != end:\n lazy[node*2] += diff\n lazy[node*2+1] += diff\n return\n mid = (start + end) // 2\n update_range(node*2, start, mid, left, right, diff)\n update_range(node*2+1, mid+1, end, left, right, diff)\n tree[node] = tree[node*2] + tree[node*2+1]\n\nn, m, k = map(int, input().split())\n\nl = []\nfor _ in range(n): l.append(int(input()))\n\nheigth = math.ceil(math.log2(n))\nsize = 2 ** (heigth + 1)\ntree = [0 for _ in range(size)]\nlazy = [0 for _ in range(size)]\ninit(1, 0, n-1)\n\nfor _ in range(m+k):\n t = list(map(int, input().split()))\n if t[0] == 1:\n update_range(1, 0, n-1, t[1]-1, t[2]-1, t[3])\n else:\n print(query(1, 0, n-1, t[1]-1, t[2]-1))