[알고리즘 일기] 28. 먹을 것인가 먹힐 것인가

 

문제 28. (2021-05-28)

 

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

7795(Baekjoon). 먹을 것인가 먹힐 것인가

심해에는 두 종류의 생명체 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보다 큰 쌍의 개수를 출력한다.

 

 

❧ 테스트 셋

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

 

❧ 출력 예시

7
1

 

---

❧ 정답

def binary_search(element, some_list):
    start, end = 0, len(some_list)-1
    result = -1
    while start <= end:
        middle = (start + end) // 2
        if some_list[middle] < element:
            result = middle
            start = middle + 1
        else:
            end = middle - 1
    return result

repeat = int(input())
for i in range(repeat):
  count = 0
  a, b = map(int, input().split())
  groupA = list(map(int,input().split()))
  groupB = list(map(int,input().split()))
  groupA.sort()
  groupB.sort()
  for l in groupA:
    count += (binary_search(l, groupB) + 1)
  print(count)

 

1️⃣ 이분 탐색 진행

2️⃣ binary_search 함수의 반환값이 인덱스 이므로, count 맨 뒤에 +1을 해줌

3️⃣ 반환값이 없으면 -1(오류) 리턴시키고 +1을 하여 count += 0을 해줌

 

 

 

 

728x90