[PROGRAMMERS] 조건에 맞게 수열 변환하기 2 (LV. 0) - Python

[ 문제 ]

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

 

 

[ 입출력 예 ]

arr	result
[1, 2, 3, 100, 99, 98]	5

 

< 입출력 예 #1 >

• 위 작업을 반복하면 다음과 같이 arr가 변합니다.
• 이후로 arr가 변하지 않으며, arr(5) = arr(6)이므로 5를 return 합니다.

반복 횟수	arr
0	[1, 2, 3, 100, 99, 98]
1	[3, 2, 7, 50, 99, 49]
2	[7, 2, 15, 25, 99, 99]
3	[15, 2, 31, 51, 99, 99]
4	[31, 2, 63, 51, 99, 99]
5	[63, 2, 63, 51, 99, 99]
6	[63, 2, 63, 51, 99, 99]

 

 

 

[ 해결 방법 ]

1. 반복 횟수를 입력할 answer 변수와 연산된 값을 저장할 after_calc 리스트를 생성
2. 무한루프로 반복을 진행
3. 무한루프 내부에서 arr 리스트로 반복을 진행
4. 50 이상의 짝수일 경우 after_calc에 i / 2 값을 추가
5. 50 미만의 홀수일 경우 after_calc에 i * 2 + 1 값을 추가
6. 만약 arr과 after_calc가 같을 경우 무한루프 종료
7. 종료하지 못할 경우 answer에 횟수를 추가하고 after_calc를 arr로 변경하고 after_calc를 빈 리스트로 변경
8. 모든 반복이 종료된 후 answer 반환

 

 

 

[ solution.py ]

def solution(arr):
    answer = 0    
    after_calc = []
    
    while 1:
        for i in arr:
            if i >= 50 and i % 2 == 0:
                after_calc.append(i / 2)
            elif i < 50 and i % 2 != 0:
                after_calc.append(i * 2 + 1)
            else:
                after_calc.append(i)
                
        if arr == after_calc:
            break
            
        answer += 1
        arr = after_calc
        after_calc = []
    
    return answer

 

 

 

> 🌻피드백은 언제나 환영입니다🌻 <