교재 : 이것이 코딩 테스트다 with 파이썬
CHAPTER 3 그리디
실전문제 3-3 숫자 카드 게임 96p
여러 개의 숫자 카드 중에서 가장 높은 숫자가 쓰인 카드 한 장을 뽑는 게임
단, 게임의 룰을 지키며 뽑아야함
숫자가 쓰인 카드들이 N X M 형태로 놓여 있다. 이때 N은 행의 개수를 의미하고, M은 열의 개수를 의미
먼저 뽑고자 하는 카드가 포함된 행을 선택
선택된 행에 포함된 카드들 중 가장 숫자가 낮은 카드를 뽑음
처음에 카드를 골라낼 행을 선택할 때, 이후 해당 행에서 가장 숫자가 낮은 카드를 뽑아야 하는 것을 고려해 가장 높은 숫자를 뽑을 수 있도록 해야함
첫째 줄에 숫자 카드들이 놓인 행의 개수 N과 열의 개수 M이 공백을 기준으로 하여 각각 자연수로 주어짐
(1 <= N, M <= 100)
둘째 줄부터 N개의 줄에 걸쳐 각 카드에 적힌 숫자가 주어짐 각 숫자는 1이상 10000이하의 자연수
첫째 줄에 게임의 룰에 맞게 선택한 카드에 적힌 숫자를 출력
입력 예시
3 3
3 1 2
4 1 4
2 2 2
출력 예시
2
n, m, k = map(int, input().split())
arr = list(map(int, input().split()))
res=0
arr.sort(reverse = True)
while m!=0:
res+=arr[0]*k
m-=k
res+=arr[1]
m-=1
print(res)
책은...진짜 레전드다...ㅋㅋㅋㅋㅋㅋㅋㅋ 반복문 하나없이 구현... ㄸㄹㄹ... 천재냐구...? 사실 이게 뭔가 책에 있는 예제라 어디에 돌려볼 수 없는 게 좀 아쉽긴 한데 책에서 나온 안 좋은 예? 처럼 중첩이 되어있지는 않긴 한데 진자 책처럼 아예 반복문 안쓰는 방법으로 한 게 너무 대단해보이고.... 진짜 뭔가 앞 문제도 그렇고 저자가 참 나누기랑 나머지 이용해서 반복문 많이 안쓰도록 잘 하는 것 같아서 그리디 말고 다른 거 풀 때도 나도 이렇게 고민을 많이 해봐야겠다고 느꼈다.. 진자 체고당.....
# N, M, K를 공백을 기준으로 구분하여 입력 받기
n, m, k = map(int, input().split())
# N개의 수를 공백을 기준으로 구분하여 입력 받기
data = list(map(int, input().split()))
data.sort() # 입력 받은 수들 정렬하기
first = data[n - 1] # 가장 큰 수
second = data[n - 2] # 두 번째로 큰 수
# 가장 큰 수가 더해지는 횟수 계산
count = int(m / (k + 1)) * k
count += m % (k + 1)
result = 0
result += (count) * first # 가장 큰 수 더하기
result += (m - count) * second # 두 번째로 큰 수 더하기
print(result) # 최종 답안 출력
- 풀이시간 : 20분
이것이 코딩 테스트다 with 파이썬 - 나동빈 저