[백준] 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

제출 답안

• res = -1인 이유는 mid가 0일 경우 cnt를 더할 때 0으로 더해진다.
  • 또한 만약에 해당되는 경우가 없으면 -1을 반환하여 +1을 했을 때 cnt에 0을 더하기 위해서다.

'''
1. 아이디어
- a와 b를 순서대로 정렬한다.
- b의 중간 인덱스를 찾아서 a보다 크면 인덱스 왼쪽으로
- a보다 작으면 인덱스 오른쪽으로 한 칸씩 이동
2. 시간 복잡도
- 
3. 변수
- num []
'''

import sys
input = sys.stdin.readline

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


t = int(input())

for _ in range(t):
    n , m = map(int, input().split())
    a = sorted(list(map(int, input().split())))
    b = sorted(list(map(int, input().split())))
    cnt = 0
    for i in a:
        cnt += binary_search(b, i) + 1
    print(cnt)