백준 SILVER 3 [2236번] 칩 만들기 {언어 : Python}

문제 링크

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

2236번: 칩 만들기

정답 코드

import sys
from collections import deque
input = sys.stdin.readline

N, K = map(int, input().split())
arr = list(map(int, input().split()))
sortedArr = []
for idx in range(N):
    sortedArr.append((arr[idx], idx))
sortedArr = sorted(sortedArr, key=lambda x: (x[0], x[1]))
result = [0 for _ in range(N)]
resultIndexes = []
q = deque()
while sortedArr:
    if len(resultIndexes) == K:
        break
    q.append(sortedArr.pop())
    current = q[0]
    result[current[1]] = current[1] + 1
    resultIndexes.append(current[1]+1)
    q.popleft()

sortedResultIndexes = sorted(resultIndexes)
for sri in sortedResultIndexes:
    print(sri)
for _ in range(K-N):
    print(0)
for r in result:
    print(r)

풀이 방법

• 거창하게 생각해서 요리조리 풀었는데 결론은 정렬해서 가장 큰 값을 하나씩 추가하는 것.
• 전원과 연결되지 않은 부분에 0 출력인가 그래서 마지막부분에 K-N만큼 0을 출력한다.
  • 이 부분 안하면 틀림

느낀점

• 이상한 문제다.
• 정렬을 하고 출력하든 안하고 출력하든 둘 다 맞는다.
• 예시의 답과 달라도 정답이다.
• 더 간단하게 풀 수도 있는데 바꾸기 귀찮다.