[프로그래머스] Lv.0 - 조건에 맞게 수열 변환하기 2.py

문제

• 정수 배열 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

입출력 예시

• 입출력 예 1
  • 위 작업을 반복하면 다음과 같이 arr가 변합니다.

• 이후로 arr가 변하지 않으며, arr(5) = arr(6)이므로 5를 return 합니다.

코드

def solution(arr):
    answer = 0
    count = 0

    while(1):
        for idx, val in enumerate(arr):
            if (val >= 50) and (val % 2 == 0):
                val /= 2
                arr[idx] = val
                continue
            elif (val < 50) and (val % 2 == 1):
                val = val * 2 + 1
                arr[idx] = val
                continue
            else:
                count += 1
                continue
        if count == len(arr):
            break
        count = 0
        answer += 1
    return answer

출처: 프로그래머스 코딩 테스트 연습, https://school.programmers.co.kr/learn/challenges