[백준] 7795번: 먹을 것인가 먹힐 것인가 - 파이썬

📕 풀이

정렬 후 투포인터를 사용하여 a의 값 하나를 b의 값 전부 하나하나 비교해가면서 a의 값이 클때의 갯수를 카운트 해준다.

 

💻 코드

import sys
input = sys.stdin.readline

for i in range(int(input().rstrip())):
	# a와 b의 갯수를 입력받음
    a, b = map(int, input().rstrip().split()) 
    # a와 b 정보를 입력받음(생명체 A, B)
    a_num = list(map(int, input().rstrip().split())) 
    b_num = list(map(int, input().rstrip().split()))
    # 내림차순 정렬
    a_num.sort(reverse=True)
    b_num.sort(reverse=True)
    a_idx, b_idx = 0,0
    cnt = 0
    
    while a_idx < a and b_idx < b:
    	# 만약 a[i]가 b[i]보다 크다면
        if a_num[a_idx] > b_num[b_idx]:
        	# b[i:] 즉 현재부터 내림차순의 끝 길이를 카운트에 더함
            cnt += b-b_idx
            # a 생명체 값 변경
            a_idx += 1
        # a[i]가 작다면 
        else:
            # b를 하나 내린다.
            b_idx += 1
    print(cnt)

 

 

 

 

📎 문제 링크 : 먹을 것인가 먹힐 것인가

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