[Beakjoon] 백준 16471번 Python
Beakjoon 16471, 작은 수 내기
solved.ac Silver4
Greedy 알고리즘
문제
여자친구와 함께 보드게임카페에 간 주언이는, 여러 보드게임을 하며 데이트를 즐겼다. 3시간 커플세트로 결제를 하려던 순간, 주언이는 가격표 옆에 쓰여 있는 새로운 이벤트를 보았다.
바로 “사장님과의 게임에서 이기면 무료, 지거나 비기면 5000원 추가 지불” 이벤트였다. 보드게임에 자신이 있는 주언이는 사장님에게 게임 룰을 물어보았고, 그 룰은 다음과 같았다.
- 각자 N장의 카드를 받는다. (N은 홀수다)
- 각자 카드를 1장씩 골라서 카드에 적힌 수의 크기를 비교한다. (각 카드에 적힌 수는 0이상, 100,000이하의 정수다)
- 더 작은 수가 적힌 카드를 낸 사람이 1점을 얻고, 승부에 사용된 카드는 버린다. (무승부의 경우, 둘 다 점수를 획득하지 못한다)
- 총 N번의 승부 후, (N+1)/2점 이상의 점수를 획득한 사람이 승리한다.
- (N+1)/2점 이상의 점수를 획득한 사람이 없을 경우, 승자가 없는 것으로 게임이 끝난다.
주언이는 자신이 이길 확률이 조금이라도 있을 경우 게임을 하고자 한다.
사장님이 받은 카드에 적힌 수들과, 주언이가 받은 카드에 적힌 수들이 주어질 때, 주언이가 게임을 해도 되는지 확인하자.
입력
N값이 첫 번째 줄에 입력된다. (1 ≤ N < 100,000, N은 홀수)
주언이가 받은 카드 N장에 적힌 수들이 두 번째 줄에 입력된다.
사장님이 받은 카드 N장에 적힌 수들이 세 번째 줄에 입력된다.
출력
주언이가 이길 확률이 조금이라도 있을 경우 “YES” 라고 출력하고,주언이가 이길 확률이 존재하지 않을 경우 “NO”라고 출력한다.
Solution
n = int(input())
owner = list(map(int, input().split()))
me = list(map(int, input().split()))
owner.sort(reverse=True)
me.sort()
count = 0
for i in range(n):
if owner[n -i- 1] < me[0]: #min(owner) < min(me)
count += 1
del(owner[n -i- 1], me[0])
elif owner[0] > me[0]:
count -= 1
del(owner[0], me[0])
if count > 0:
print("YES")
else:
print("NO")
풀이
먼저 사장님의 카드는 내림차순으로 정렬하고, 나의 카드는 내림 차순으로 정렬합니다.
이제 아래의 비교를 하면서 비교를 한 숫자들은 list에서 제외를 시켜줄겁니다.
사장님의 가장 큰 수가 나의 가장 작은 수 보다 작게 된다면 나의 점수를 1점 올려줍니다.
그렇지 않은 경우 사장님의 가장 작은 수가 나의 가장 큰 수 보다 크게 된다면 나의 점수를 1점 깎아줍니다.
이 작업을 반복하시면 내가 이길 수 있는지 확인할 수 있습니다.
시간 초과가 발생할 수 있기 때문에 주의하시길 바랍니다!