저울 (python)

해법을 이해하기 다소 어려운 문제, 정렬 사용

👀 깃헙 소스

문제설명

하나의 양팔 저울을 이용하여 물건의 무게를 측정하려고 합니다. 이 저울의 양팔의 끝에는 물건이나 추를 올려놓는 접시가 달려 있고, 양팔의 길이는 같습니다. 또한, 저울의 한쪽에는 저울추들만 놓을 수 있고, 다른 쪽에는 무게를 측정하려는 물건만 올려놓을 수 있습니다.

저울추가 담긴 배열 weight가 매개변수로 주어질 때, 이 추들로 측정할 수 없는 양의 정수 무게 중 최솟값을 return 하도록 solution 함수를 작성해주세요.

예를 들어, 무게가 각각 [3, 1, 6, 2, 7, 30, 1]인 7개의 저울추를 주어졌을 때, 이 추들로 측정할 수 없는 양의 정수 무게 중 최솟값은 21입니다.

입출력 예

weight 			return 
[3, 1, 6, 2, 7, 30, 1] 	21

솔루션

작은 추 순서대로 정렬한 후
nxt에 측정 범위 내에 있지 않는 제일 작은 값을 담아 둔다.
그 다음 추가 nxt값 보다 크면 측정 불가능하므로 그 값을 리턴해 준다.

코드

# 파이썬
def solution(weight):
    weight.sort() # 작은 추 순서대로 정렬
    # nxt에 측정 범위 내에 있지 않는 제일 작은 값을 담아 둠 초기값:1
    nxt = 1
    for w in weight:
        # 그 다음 추가 nxt값 보다 크면 측정 불가능, 리턴
        if w > nxt:
            break;
        nxt += w
    return nxt