본문 바로가기

CodingTest/SW Expert Academy

[ SW Expert Academy ] 1208. Flatten

아직 시간복잡도에 대한 감이나 각 라이브러리에 대한 감이 안 잡힌 것 같다.

풀이 방법이나 문제 구현은 쉽게 했는데

시간복잡도에서 터질까봐 확신 없이 제출했다.

만약 실전이라면 히든테케 맞았나 안 맞았나 불안해서 전전긍긍할 듯.. 

 

1차 제출 [ 10 / 10 ] PASS

for tc in range(1,11):
    dump_limit = int(input())
    boxes = list(map(int, input().split()))

    while dump_limit > 0:
        maxIdx = boxes.index(max(boxes))
        minIdx = boxes.index(min(boxes))

        boxes[maxIdx] -= 1
        boxes[minIdx] += 1

        dump_limit -= 1

    print("#{} {}".format(tc, max(boxes)-min(boxes)))