본문 바로가기

BOJ/C++

구간 합 구하기

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
//구간 합 구하기
#include <iostream>
#include <vector>
#include <cmath>
 
#define MAXX 1000001
using namespace std;
 
typedef long long ll;
 
ll sum(vector<ll> &tree, int node, int left, int right, int start, int end)
{
    //[start, end] 밖에 [left, right]가 있는 경우
    //[left, right]가 [start, end]를 포함하는 경우
    //그 외
 
    if (right < start || end < left)
        return 0;
    else if (left <= start && end <= right)
        return tree[node];
    else
    {
        return sum(tree, node * 2, left, right, start, (start + end/ 2+ sum(tree, node * 2 + 1, left, right, (start + end/ 2 + 1end);
    }
}
void update(vector<ll> &tree, int node, int diff, int idx, int start, int end)
{
    //해당 범위 안에 idx가 있느냐 없느냐
    if (idx < start || end < idx)
        return;
 
    tree[node] += diff;
    if (start != end)
    {
        update(tree, node * 2, diff, idx, start, (start + end/ 2);
        update(tree, node * 2 + 1, diff, idx, (start + end/ 2 + 1end);
    }
}
 
ll init(vector<ll> &tree, vector<ll> &nums, int node, int start, int end)
{
    if (start == end)
    {
        return tree[node] = nums[start];
    }
    return tree[node] = init(tree, nums, node * 2, start, (start + end/ 2+ init(tree, nums, node * 2 + 1, (start + end/ 2 + 1end);
}
int main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(0);
 
    int n, m, k;
 
    cin >> n >> m >> k;
    //수의 개수, 수의 변경이 일어나는 횟수, 구간의 합을 구하는 횟수
 
    int h = (int)ceil(log2(n));
    int tree_h = 1 << (h + 1);
 
    vector<ll> nums, tree(tree_h);
 
    for (int i = 0; i < n; i++)
    {
        int tmp;
        cin >> tmp;
        nums.push_back(tmp);
    }
 
    init(tree, nums, 10, n - 1);
 
    for (int i = 0; i < m + k; i++)
    {
        int a, b;
        cin >> a >> b;
 
        if (a == 1)
        {
            //b번째 수를 c로 바꾼다.
            ll c;
            cin >> c;
            b--;
 
            int diff = c - nums[b]; //차이만큼만 업데이트 한다.
            nums[b] = c;
 
            update(tree, 1, diff, b, 0, n - 1);
 
        }
        else
        {
            int c;
            cin >> c;
 
            //b부터 c까지의 합을 구한다.
            b--;
            c--;
            cout << sum(tree, 1, b, c, 0, n - 1<< '\n';
        }
    }
    return 0;
}
http://colorscripter.com/info#e" target="_blank" style="color:#4f4f4ftext-decoration:none">Colored by Color Scripter
http://colorscripter.com/info#e" target="_blank" style="text-decoration:none;color:white">cs

'BOJ > C++' 카테고리의 다른 글

커피숍2  (0) 2020.02.14
가운데를 말해요  (0) 2020.02.14
카드게임  (0) 2020.02.14
  (0) 2020.02.14
Dance Dance Revolution  (0) 2020.02.14