인프런 커뮤니티 질문&답변

안녕하세요!

8-I 나무심기 해설 코드를 보다 작은 의문점이 생겨서 질문드립니다!

바로 중복된 좌표에 나무를 심는 경우인데요.

나무의 개수를 저장하는 부분인 update(treeCnt, num, 1)에서 좌표가 중복되는 경우에는 1이 아니라 나무의 개수로 갱신되어야 하지 않나하는 궁금증이 생겨버렸습니다. 중복되는 경우, 나무를 뽑고 다시 심는 개념으로 이해해야 하는 걸까요?

제가 아직 펜윅트리를 완벽하게 익힌 것이 아니라...많이 헷갈립니다!

#include<iostream>
#include<vector>

using namespace std;

typedef long long ll;
int N; 
vector<ll> t;
vector<ll> tcnt;
vector<ll> treeSum;
vector<ll> treeCnt;
ll answer = 0;

void update(vector<ll> &tree, int i, int diff)
{
    while(i < tree.size())
    {
        tree[i] += diff;
        i += (i & -i);
    }
}

ll sum(vector<ll> &tree, int i)
{
    ll r = 0;
    while(i > 0)
    {
        r += tree[i];
        i -= (i & -i);
    }
    return r;
}

int main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);  
    cin >> N; 
    t = vector<ll>(N + 1, 0);
    tcnt = vector<ll>(200004, 0);
    treeSum = vector<ll>(200004, 0);
    treeCnt = vector<ll>(200004, 0);
    for(int i = 1; i <= N; i++)
    {
        ll num;
        cin >> num; num++;
        t[i] = num;
        tcnt[num]++;
        if (i != 1)
        {
            ll l, r;
            ll value;
            l = sum(treeCnt, t[i] - 1) * t[i] - sum(treeSum, t[i] - 1);
            r = (sum(treeSum, 200004) - sum(treeSum, t[i])) - (sum(treeCnt, 200004) - sum(treeCnt, t[i])) * t[i];
            value = l + r;
            if (answer == 0) answer = 1;
            answer = ((answer % 1000000007) * (value % 1000000007)) % 1000000007;
        }
        update(treeSum, num, num);
        update(treeCnt, num, 1);
    }

    cout << answer << "\n";
    return 0;
}