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

문제

• 정수 배열 arr가 주어집니다. arr의 각 원소에 대해 값이 50보다 크거나 같은 짝수라면 2로 나누고, 50보다 작은 홀수라면 2를 곱합니다. 그 결과인 정수 배열을 return 하는 solution 함수를 완성해 주세요.

제한사항

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

입출력 예시

• 입출력 예 1
  • 1, 3은 50 미만의 홀수 이므로 2를 곱하고, 100, 98은 50 이상의 짝수이므로 2로 나눕니다.
  • 나머지 값들은 변경 조건에 해당하지 않으므로 바꾸지 않습니다.
  • 따라서 [2, 2, 6, 50, 99, 49]를 return 합니다.

코드

def solution(arr):
    answer = []
    for idx, val in enumerate(arr):
        if (val >= 50) and (val % 2 == 0):
            val = val / 2
            arr[idx] = val
            continue
        if (val < 50) and (val % 2 == 1):
            val *= 2
            arr[idx] = val
            continue
    answer = arr
    return answer

 

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