조건에 맞게 수열 변환하기 1

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

제한사항
1 ≤ arr의 길이 ≤ 1,000,000
1 ≤ arr의 원소의 값 ≤ 100

 

풀이

def solution(arr):
    result = []
    for i in range(len(arr)):
        if arr[i] < 50 and arr[i] % 2 == 1:
            result.append(arr[i] * 2)  # 홀수이고 50 미만인 경우 두 배
        elif arr[i] >= 50 and arr[i] % 2 == 0:
            result.append(arr[i] // 2)  # 짝수이고 50 이상인 경우 절반
        else:
            result.append(arr[i])  # 나머지 값은 그대로
    return result

복합조건이 필요한 문제이므로 and를 사용해서 추가적으로 조건을 중첩시키면 된다.