[Level.0] 조건에 맞게 수열 변환하기(2)

문제 설명: 정수 배열 arr가 주어집니다. arr의 각 원소에 대해 값이 50보다 크거나 같은 짝수라면 2로 나누고, 50보다 작은 홀수라면 2를 곱하고 다시 1을 더합니다. 이러한 작업을 x번 반복한 결과인 배열을 arr(x)라고 표현했을 때, arr(x) = arr(x + 1)인 x가 항상 존재합니다. 이러한 x 중 가장 작은 값을 return 하는 solution 함수를 완성해 주세요.

• 단, 두 배열에 대한 "="는 두 배열의 크기가 서로 같으며, 같은 인덱스의 원소가 각각 서로 같음을 의미합니다.

제한사항

• 1 ≤ arr의 길이 ≤ 1,000,000
• 1 ≤ arr의 원소의 값 ≤ 100

문제 해결

def solution(arr):
    count = 0
    while True:
        changed = False
        for i in range(len(arr)):
            val = arr[i]
            if val >= 50 and val % 2 == 0:
                arr[i] = val // 2
                changed = True
            elif val < 50 and val % 2 == 1:
                arr[i] = val * 2 + 1
                changed = True
        if not changed:
            return count
        count += 1

 

주석: 조건에 부합하는 원소에 대해 변환 연산을 적용하고 변경 사실을 기록합니다.

for ar in arr:

    if ar >= 50 and ar%2 == 0:

        ar //= 2

        changed = True

윗 코드가 불가능한 이유는 for ar in arr에서 ar은 리스트의 원소를 복사해온 임시 변수일 뿐이다. 그래서 ar //= 2나 ar = ar * 2 + 1을 수행하면 임시 변수 ar의 값만 바뀔 뿐 실제 arr 안에 들어있는 값은 그대로 유지되기 때문에 changed는 True가 되어 count는 올라가지만 arr의 내용물은 변하지 않으므로 다음 루프에서도 똑같은 작업을 반복하며 무한 루프에 빠지게 됩니다.