1572 중앙값

아.. 어려웠다. 구현 하려고 하다가 못 하겠어서 다른 사람 블로그 참고해서 풀었다. 정말 깔끔하게 잘 푸셨다..

그냥 참고해서 공부하고 익히기로. 저렇게 반반 나눠서 사용하는 거 신기하다. 

 

#include <iostream>
#include <algorithm>
#include <vector>
#include <string>
#include <functional>
#include <string>
#include <queue>
#include <deque>
#include <stack>
#include <set>
#include <map>
#include <cmath>
#include <cstring>
#include <bitset>

#define xx first
#define yy second
#define all(x) (x).begin(), (x).end()

using namespace std;
using i64 = long long int;
using ii = pair<int, int>;
using iis = pair<int, string>;
using ii64 = pair<i64, i64>;
using iii = tuple<int, int, int>;
 
int main() {
    int n, k;
    scanf("%d %d", &n, &k);
    
    vector<int> v(n);
    multiset<int> lo, hi;
    for (int i = 0; i < n; i++) {
        scanf("%d", &v[i]);
        if (i < k)
            lo.emplace(v[i]);
    }
    
    while (lo.size() >= hi.size()) {
        hi.emplace(*lo.rbegin());
        lo.erase(lo.find(*lo.rbegin()));
    }
    
    i64 ans = *hi.begin();
    
    for (int i = k; i < n; i++) {
        auto r = lo.find(v[i - k]);
        if (r != lo.end())
            lo.erase(r);
        else
            hi.erase(hi.find(v[i - k]));
        lo.emplace(v[i]);
        
        while (lo.size() < hi.size()) {
            lo.emplace(*hi.begin());
            hi.erase(hi.begin());
        }
        while (lo.size() >= hi.size()) {
            hi.emplace(*lo.rbegin());
            lo.erase(lo.find(*lo.rbegin()));
        }
        ans += *hi.begin();
    }
    
    printf("%lld", ans);
    
    return 0;
}