7795번: 먹을 것인가 먹힐 것인가

https://www.acmicpc.net/problem/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보다 큰 쌍의 개수를 출력한다.

예제 입력 1

2
5 3
8 1 7 3 1
3 6 1
3 4
2 13 7
103 11 290 215

예제 출력 1

7
1

코드

package binary_search;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.StringTokenizer;

/**
 * 이분 탐색 기본 아이디어 : 오름차순 정렬된 배열에서 특정 숫자 x보다 작은 숫자가 몇개 있는지 세는 것
 * 정렬된 배열의 left, right을 정하고 그 중간값 mid를 x와 비교함
 * mid < x --> mid의 왼쪽 값은 안봐도 x보다 작다 --> left = mid + 1 / result = mid
 * mid > x --> mid의 오른쪽 값은 안봐도 x보다 크다 --> right = mid - 1
 * left > right --> 탐색 끝
 */
public class Main_7795 {
    
    private static int numberOfTestcase;
    private static int numberOfA;
    private static int numberOfB;
    private static int[] valuesForA;
    private static int[] valuesForB;

    private static BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
    private static StringBuilder sb = new StringBuilder();

    private static void input() throws IOException {
        StringTokenizer st;

        st = new StringTokenizer(br.readLine());
        numberOfA = Integer.parseInt(st.nextToken());
        numberOfB = Integer.parseInt(st.nextToken());

        valuesForA = new int[numberOfA];
        valuesForB = new int[numberOfB];

        st = new StringTokenizer(br.readLine());
        for (int i = 0; i < numberOfA; i++) {
            valuesForA[i] = Integer.parseInt(st.nextToken());
        }

        st = new StringTokenizer(br.readLine());
        for (int i = 0; i < numberOfB; i++) {
            valuesForB[i] = Integer.parseInt(st.nextToken());
        }

        Arrays.sort(valuesForB);
    }

    private static void process() {
        int sum = 0;

        for (int a : valuesForA) {
            sum += binarySearch(a);
        }

        sb.append(sum).append('\n');
    }

    private static int binarySearch(int a) {
        int start = 0;
        int end = numberOfB - 1;
        int mid;
        int result = 0;

        while (start <= end) {
            mid = (start + end) / 2;

            if (valuesForB[mid] < a) {
                result = mid + 1;
                start = mid + 1;
            }

            if (valuesForB[mid] >= a) {
                end = mid - 1;
            }
        }

        return result;
    }

    private static void output() {
        System.out.println(sb);
    }

    public static void main(String[] args) throws IOException {
        numberOfTestcase = Integer.parseInt(br.readLine());

        for (int i = 0; i < numberOfTestcase; i++) {
            input();
            process();
        }

        output();
    }
}