[백준/Python] 7795번: 먹을 것인가 먹힐 것인가

https://www.acmicpc.net/problem/7795

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

입출력 예제)

소스코드

from sys import stdin

for i in range(int(stdin.readline())):
  n, m = map(int, stdin.readline().split())
  a =list(map(int,stdin.readline().split()))
  b =list(map(int,stdin.readline().split()))
  a.sort()
  b.sort()
  ans=0
  count=0
  for j in range(n):
     while True:
            if count == m or a[j] <= b[count]:
                ans += count
                break
            else:
                count += 1
                
 
        
  print(ans)

문제풀이

 sort 명령어를 통하여 오름차순으로 정렬하였다. 그리고 if문에서 집합 A의 원소가 집합 B의 원소보다 큰 경우를 나타내었고, 또, count==m을 넣음으로써 b의 index 범위가 넘어가지 않도록 하였다. 

 유의할 점은 for문 안에 for문을 넣는 것이 아니라 while문을 넣었는데 이는 실행 시간 초과를 방지하기 위해서이다.