[알고리즘] BOJ 7795 먹을 것인가 먹힐 것인가 #Python

[BOJ] 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보다 큰 쌍의 개수를 출력한다.

---

📍 풀이

💡 고찰

• 두 종류의 생명체 A, B를 크기 순으로 정렬한 후 이진 탐색(Binary Search) 을 적용하면 A의 크기가 B보다 큰 쌍이 몇 개나 있는지 구할 수 있다.
• 파이썬 라이브러리 중에 이진 탐색을 위한 bisect 가 존재한다.
  → 📒 bisect — 배열 이진 분할 알고리즘
• bisect_left 함수는 주어진 리스트에서 자신보다 크거나 같은 값의 왼쪽 index 값을 반환한다.
• bisect_left 함수를 활용하여 A의 크기가 B보다 큰 쌍이 몇 개나 있는지 구할 수 있다.

📌 문제 풀이

✏️ 1. 이진 탐색을 이용한 A의 크기가 B보다 큰 [A, B] 쌍의 개수 탐색

answer = 0
for a in A:
    answer += bisect_left(B,a)

✍ 코드

from sys import stdin
from bisect import bisect_left

def solution(A,B):
    answer = 0
    # 이진 탐색 적용
    for a in A:
        answer += bisect_left(B,a)
    return answer

# input
T = int(stdin.readline())
for _ in range(T):
    N, M = map(int,stdin.readline().split())
    A = sorted(list(map(int,stdin.readline().split())))
    B = sorted(list(map(int,stdin.readline().split())))

    # result
    result = solution(A,B)
    print(result)