[백준] 7795

먹을 것인가 먹힐 것인가

문제

심해에는 두 종류의 생명체 A와 B가 존재한다. A는 B를 먹는다. A는 자기보다 크기가 작은 먹이만 먹을 수 있다. 예를 들어, A의 크기가 {8, 1, 7, 3, 1}이고, B의 크기가 {3, 6, 1}인 경우에 A가 B를 먹을 수 있는 쌍의 개수는 7가지가 있다. 8-3, 8-6, 8-1, 7-3, 7-6, 7-1, 3-1.

두 생명체 A와 B의 크기가 주어졌을 때, A의 크기가 B보다 큰 쌍이 몇 개나 있는지 구하는 프로그램을 작성하시오.

입력

첫째 줄에 테스트 케이스의 개수 T가 주어진다. 각 테스트 케이스의 첫째 줄에는 A의 수 N과 B의 수 M이 주어진다. 둘째 줄에는 A의 크기가 모두 주어지며, 셋째 줄에는 B의 크기가 모두 주어진다. 크기는 양의 정수이다. (1 ≤ N, M ≤ 20,000)

출력

각 테스트 케이스마다, A가 B보다 큰 쌍의 개수를 출력한다.

풀이

#include <iostream>
#include <vector>
#include <algorithm>
#include <cmath>

int main(void) {
    std::ios_base::sync_with_stdio(false);
    std::cin.tie(0);
    std::cout.tie(0);

    int t, a, b, temp, result;
    int min, max, mid;
    std::vector<int> av, bv;

    std::cin >> t;

    for (int i = 0; i < t; i++) {
        result = 0;
        av.clear();
        bv.clear();
        std::cin >> a >> b;
        for (int j = 0; j < a; j++) {
            std::cin >> temp;
            av.push_back(temp);
        }
        for (int j = 0; j < b; j++) {
            std::cin >> temp;
            bv.push_back(temp);
        }

        std::sort(av.begin(), av.end());
        std::sort(bv.begin(), bv.end());

        for (int j = 0; j < a; j++) {
            min = 0;
            max = b - 1;
            while (min <= max) {
                mid = (min + max) / 2;
                if (av[j] > bv[mid])
                    min = mid + 1;
                else
                    max = mid - 1;
            }
            result += min;
        }
        std::cout << result << "\n";
    }
}

반성회

이분탐색으로 풀었다

av[j] 랑 bv[mid] 가 같을 때 max를 이동시키지 않고 min을 이동시켜 버리면 min이 가리키는 인덱스가 더 커져서 “A가 B보다 클 때” 가 아니라 “A가 B보다 크거나 같을 때” 의 개수를 구하게 되기 때문에 max를 이동해야 min값을 사용할 수 있다

매 테스트케이스마다 벡터 초기화를 안 해줘서 몇 번을 틀렸다…