[Baekjoon/C++] 7795번 - 먹을 것인가 먹힐 것인가

[백준] Baekjoon Online Judge

문제로 이동

 

문제

심해에는 두 종류의 생명체 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 <algorithm>
using namespace std;

int N, M;
int A[20001], B[20001];

int binary(int num);

int binary(int num) {
    int low = 0;
    int high = M - 1;

    while (low <= high) {
        int mid = (low + high) / 2;

        if (B[mid] < num) low = mid + 1;
        else high = mid - 1;
    }

    return low;
}

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

    int T;
    cin >> T;

    while (T--) {
        cin >> N >> M;

        for (int i = 0; i < N; i++) cin >> A[i];
        for (int i = 0; i < M; i++) cin >> B[i];

        sort(B, B + M);

        int sum = 0;
        for (int i = 0; i < N; i++) sum += binary(A[i]);

        cout << sum << '\n';
    }

    return 0;
}

 이분 탐색으로 문제를 풀었다.