[백준 2042 c++ V] 구간합구하기

- 풀이 링크:

https://github.com/xhaktmchl/Algorithm_study/blob/main/BOJ/%ED%8A%B8%EB%A6%AC/%EC%9D%B8%EB%8D%B1%EC%8A%A4%20%ED%8A%B8%EB%A6%AC/%5B%EB%B0%B1%EC%A4%80%202042%20c%2B%2B%20V%5D%20%EA%B5%AC%EA%B0%84%ED%95%A9%EA%B5%AC%ED%95%98%EA%B8%B0.cpp

GitHub - xhaktmchl/Algorithm_study: 알고리즘 이론 및 문제풀이

#include <iostream>
#include <algorithm> // fill_n, min,max, swap
//#include <map> // 중복 x 
//#include <string> // getline
//#include <cstring> // memset, strok, strstr
//#include <vector>
//#include <queue> // priority_queue<자료형, 구현체, 비교 연산자> (중복허용)
//#include <set> // 트리, 중복 x , multiset
//#include <unordered_set>
//#include <cmath>
using namespace std;
/*
[백준 2042 c++ V] 구간합구하기 
문제: 
접근: 세그, 인덱스드트리 -> 반복적인 구간합 구하기
시간복잡도: 
풀이1: 인덱스드 트리
    //1.입력
    //2.인덱스드 트리 초기화
    //1)n보다 크거나 같은 수중 가장 작은 2의 제곱수 구하기
    //2) 찾은 제곱수부터 원래 배열 채우기, 뒤에는 0으로 채우기
    //3) 트리 왼쪽 배열 채우기: 찾은 제곱수-1 부터 0까지 왼자식+오른자식 합 저장 tree[i] = tree[i*2]+tree[i*2+1]
    
    //3.트리 업데이트
    //1) 트리 인덱스부터 시작
    //2) 숫자 업데이트
    //3) 변경된 숫자 조상 부모들 탐색하며 모두 왼자식+오른자식으로 업데이트

    //4.트리 구간합 구하기
    //1) 트리의 인덱스로 이동
    //트리구간의 시작과 끝이 역전되기 전 까지 반복
    //2) 트리 구간의 시작이 홀수번째 일 때: 더하고 오른부모로 올라감
    //3) 트리 구간의 끝이 짝수일때: 더하고 왼 부모로 올라감
    //구간합 반환
*/
#define MAX_N 1000000
#define MAX_TREE 2097152
typedef long long ll;
int n, m, k,a;
ll ori[MAX_N];
ll OFFSET = 1;
ll tree[MAX_TREE];
ll b, c;

void init() {
    while (OFFSET < n) OFFSET *= 2;
    for (int i = 0; i < n; i++) tree[OFFSET + i] = ori[i];
    for (int i = n; i < OFFSET; i++) tree[OFFSET+i] = 0;
    for (int i = OFFSET - 1; i > 0; i--) tree[i] = tree[i * 2] + tree[i * 2 + 1];
}

void update(int idx, ll val) {
    idx += OFFSET;
    tree[idx] = val;
    while (idx > 0) {
        idx /= 2;
        tree[idx] = tree[idx * 2] + tree[idx * 2 + 1];
    }
}

ll query(int from, int to) {
    ll sum = 0;
    from += OFFSET; to += OFFSET;
    
    while (from <= to) {
        if (from % 2 == 1) sum += tree[from++];
        if (to % 2 == 0)sum += tree[to--];

        from /= 2; to /= 2;
    }
    return sum;
}

int main()
{
    ios::sync_with_stdio(false); cin.tie(nullptr); cout.tie(0);
    //1.입력
    cin >> n >> m >> k;
    for (int i = 0; i < n; i++) {
        cin >> ori[i];
    }
    //2.인덱스드 트리 초기화
    //1)n보다 크거나 같은 수중 가장 작은 2의 제곱수 구하기
    //2) 찾은 제곱수부터 원래 배열 채우기, 뒤에는 0으로 채우기
    //3) 트리 왼쪽 배열 채우기: 찾은 제곱수-1 부터 0까지 왼자식+오른자식 합 저장 tree[i] = tree[i*2]+tree[i*2+1]
    init();

    //3.트리 업데이트
    //1) 트리 인덱스부터 시작
    //2) 숫자 업데이트
    //3) 변경된 숫자 조상 부모들 탐색하며 모두 왼자식+오른자식으로 업데이트

    //4.트리 구간합 구하기
    //1) 트리의 인덱스로 이동
    //트리구간의 시작과 끝이 역전되기 전 까지 반복
    //2) 트리 구간의 시작이 홀수번째 일 때: 더하고 오른부모로 올라감
    //3) 트리 구간의 끝이 짝수일때: 더하고 왼 부모로 올라감
    //구간합 반환
    for (int i = 0; i < m + k; i++) {
        cin >> a >> b >> c;
        if (a == 1) update(b - 1, c);
        if (a == 2) cout << query(b - 1, c - 1) << '\n';
    }
    return 0;
}