티스토리 뷰
먹을 것인가 먹힐 것인가 성공다국어
7795번: 먹을 것인가 먹힐 것인가
심해에는 두 종류의 생명체 A와 B가 존재한다. A는 B를 먹는다. A는 자기보다 크기가 작은 먹이만 먹을 수 있다. 예를 들어, A의 크기가 {8, 1, 7, 3, 1}이고, B의 크기가 {3, 6, 1}인 경우에 A가 B를 먹을
www.acmicpc.net
문제
심해에는 두 종류의 생명체 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.
![](https://blog.kakaocdn.net/dn/bvnbzM/btszmAe4SM4/pPKYDXNUfYt2K4xOEoUTX1/img.png)
두 생명체 A와 B의 크기가 주어졌을 때, A의 크기가 B보다 큰 쌍이 몇 개나 있는지 구하는 프로그램을 작성하시오.
입력
첫째 줄에 테스트 케이스의 개수 T가 주어진다. 각 테스트 케이스의 첫째 줄에는 A의 수 N과 B의 수 M이 주어진다. 둘째 줄에는 A의 크기가 모두 주어지며, 셋째 줄에는 B의 크기가 모두 주어진다. 크기는 양의 정수이다. (1 ≤ N, M ≤ 20,000)
출력
각 테스트 케이스마다, A가 B보다 큰 쌍의 개수를 출력한다.
내 코드 (이분 탐색 ver.)
import sys
t=int(sys.stdin.readline())
for _ in range(t):
n,m=map(int,sys.stdin.readline().split())
arr1=list(map(int,sys.stdin.readline().split()))
arr2=list(map(int,sys.stdin.readline().split()))
#숫자들 정렬.
arr1.sort()
arr2.sort()
total=0
big=max(arr2)
for num in arr1:
if num>big: #arr2의 최대보다 크면,
total+=len(arr2) #arr2의 길이만큼 더하기.
continue
start=0
end=len(arr2)-1
answer=0
while start<=end: #arr1의 수가 arr2의 어느 부분에 위치하는지 찾기.
mid=(start+end)//2
if arr2[mid]>=num:
end=mid-1
answer=mid
else:
start=mid+1
total+=answer #arr1보다 한 단계만 큰 arr2 수의 위치가 answer에 있음. 이것을 더해주기.
print(total)
풀이 및 접근)
- arr1(A)에 있는 수들이 arr2(B)에서 어느 부분에 위치하는지를 찾아야겠다고 생각했다.
- 이를 선형탐색으로 일일이 찾는게 아니라 이분 탐색을 통해 위치를 빠르게 찾았다.
- 이분 탐색으로 arr1의 특정 수가 arr2의 특정 수보다 작아지기 전까지 위치를 이동시켰다. 그러다가 arr1이 작아지는 순간의 arr2 위치가 그 개수가 된다. 이를 total에 누적한다.
- arr2의 최대값보다 arr1의 특정 수가 크면 모든 arr2의 수들보다 큰 것이므로 arr2의 길이를 누적한다.
내 코드 (투 포인터 ver.)
import sys
t=int(sys.stdin.readline())
for _ in range(t):
n,m=map(int,sys.stdin.readline().split())
arr1=list(map(int,sys.stdin.readline().split()))
arr2=list(map(int,sys.stdin.readline().split()))
#정렬.
arr1.sort()
arr2.sort()
pos1=0 #arr1에서 움직일 포인터
pos2=0 #arr2에서 움직일 포인터
total=0
while pos1<len(arr1) and pos2<len(arr2): #두 배열이 모두 좌표 내에 있을 때 반복.
if arr1[pos1]>arr2[pos2]: #arr1의 수가 더 크면,
pos2+=1 #arr2의 포인터를 전진.
else: #arr1의 수가 더 작으면,
total+=pos2 #total에 arr2의 위치(이것이 곧 개수)를 누적하고,
pos1+=1 #arr1의 포인터를 전진
#arr1에 수가 더 남아있다는 뜻은, 모든 arr2의 수보다 남은 수들이 더 큰 것임.
for _ in range(len(arr1)-pos1):
total+=pos2
print(total)
풀이 및 접근)
- arr1의 포인터와 arr2의 포인터를 만들어서 전진 시켜나가며 비교하는 방식이다.
- 정렬을 시킨 후 진행하므로, 비교를 할 때 처음으로 돌아가지 않아도 된다.
- 신경 써야할 것은, arr2부분이 끝났는데 arr1이 남아있으면, 남은 arr1이 모든 arr2보다 큰 것이므로, 이를 처리해 주어야 한다.
'백준' 카테고리의 다른 글
백준 5525: IOIOI - 문자열(파이썬) (0) | 2023.11.07 |
---|---|
백준 1238: 파티 - 다익스트라(파이썬) (0) | 2023.11.07 |
백준 16401: 과자 나눠주기 - 이분 탐색(파이썬) (0) | 2023.10.27 |
백준 1072: 게임 - 이분 탐색, 수학(파이썬) (0) | 2023.10.27 |
백준 6236: 용돈 관리 - 이분 탐색(파이썬) (0) | 2023.10.27 |