3000. 중간값 구하기(D4)

STL multiset을 이용해 새로운 숫자가 입력될 때마다 정렬이 되도록했다.

그리고 새로 입력된 수와 이전 중간값을 비교하여 iterator의 위치를 변경해주어 중간 위치를 유지하도록 하였다.

#include <stdio.h>
#include <set>
using namespace std;
int main()
{
    int T;
    setbuf(stdout, NULL);
    scanf("%d", &T);
    for (int test_case = 1; test_case <= T; test_case++)
    {
        int n, a;
        scanf("%d%d", &n,&a);
        multiset<int> s;
        s.insert(a);
        long long ans = 0;
        auto it = s.begin();
        int mid = a;
        for (int i = 0; i < n; i++)
        {
            int t1, t2;
            scanf("%d%d", &t1,&t2);
            s.insert(t1);
            s.insert(t2);
            if (t1 >= mid && t2 >= mid)
            {
                it++;
            }
            else if (t1 >= mid && t2 <= mid)
            {
 
            }
            else if (t1 <= mid && t2 >= mid)
            {
 
            }
            else if (t1 <= mid && t2 <= mid)
            {
                it--;
            }
            ans += (*it);
            ans %= 20171109;
            mid = (*it);
        }
        printf("#%d %lld\n", test_case, ans);
    }
    return 0;
}