[백준] 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 의 원소들중 순회 대상 원소보다 작은 원소들 중 가장 큰 값을
이분 탐색으로 찾는다.
찾은 원소의 인덱스를 반환한다.

인덱스의 시작은 0 부터 시작하므로 반환 값 + 1 한 값을
answer에 누적 시킨다.

def binary_search(arr, target):
    start, end = 0, len(arr) - 1
    res = -1
    while start <= end:
        mid = (start + end) // 2
        if arr[mid] < target:
            res = mid
            start = mid + 1
        else:
            end = mid - 1
    return res


def solution(N,M,A,B):
    answer = 0
    A.sort()
    B.sort()
    for a in A:
        answer += binary_search(B, a) + 1
    print(answer)

T = int(input())
for _ in range(T):
    N, M = map(int, input().split(" "))
    A = list(map(int, input().split(" ")))
    B = list(map(int, input().split(" ")))
    solution(N, M, A, B)

이분 탐색 라이브러리 사용

from bisect import bisect_left

def solution(N,M,A,B):
    answer = 0
    A.sort()
    B.sort()
    for a in A:
        answer += bisect_left(B, a)
    print(answer)

T = int(input())
for _ in range(T):
    N, M = map(int, input().split(" "))
    A = list(map(int, input().split(" ")))
    B = list(map(int, input().split(" ")))
    solution(N, M, A, B)