BOJ 14504 : 수열과 쿼리 18

• 문제 링크 : boj.kr/14504
• 난이도 : D5
• 태그 : 제곱근 분할법

---

코드

#include <bits/stdc++.h>

#define all(x) (x).begin(), (x).end()

#define INF 0x7FFFFFFF

using namespace std;

using ll = long long;
using ld = long double;
using pii = pair<int,int>;
using pll = pair<ll, ll>;

int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0); cout.tie(0);

    int n;
    cin >> n;
    vector<int> arr(n);
    for(auto &e : arr) cin >> e;
    int s = sqrt(n);
    vector<vector<int>> bucket(n/s+1);
    for(int i = 0; i < n; i++) {
        bucket[i/s].emplace_back(arr[i]);
    }
    for(auto &e : bucket) {
        sort(all(e));
    }

    int m;
    cin >> m;
    for(int i = 0; i < m; i++) {
        int q;
        cin >> q;
        if(q == 1) {
            int l, r, k;
            cin >> l >> r >> k;
            l--, r--;
            int ans = 0;
            for(int i = l; i <= r; i++) {
                if((i-1)/s != (i/s) && i+s-1 <= r) {
                    int idx = i/s;
                    ans += bucket[idx].end() - upper_bound(all(bucket[idx]), k);
                    i += s-1;
                }
                else {
                    ans += (arr[i] > k);
                }
            }
            cout << ans << "\n";
        }
        if(q == 2) {
            int i, k;
            cin >> i >> k;
            i--;
            int idx = i/s;
            *lower_bound(all(bucket[idx]), arr[i]) = k;
            arr[i] = k;
            sort(all(bucket[idx]));
        }
    }


    return 0;
}

---

풀이

제곱근 분할법을 이용하면 각 쿼리를 $O(\sqrt n\log n)$으로 처리할 수 있다.