1 분 소요

문제 설명

프로그래머스 181882번 조건에 맞게 수열 변환하기 1

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

제한 사항

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

입출력 예시

arr result
[1, 2, 3, 100, 99, 98] [2, 2, 6, 50, 99, 49]

입출력 예 설명 1

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

코드 구현

Python

def solution(arr):
    for idx in range(len(arr)):
        if arr[idx] % 2 == 0 and arr[idx] >= 50:
            arr[idx] /= 2
        elif arr[idx] % 2 == 1 and arr[idx] < 50:
            arr[idx] *= 2
            
    return arr

Java

class Solution {
    public int[] solution(int[] arr) {
        for (int i = 0; i < arr.length; i += 1) {
            if (arr[i] % 2 == 0 && arr[i] >= 50) {
                arr[i] /= 2;
            } else if (arr[i] % 2 == 1 && arr[i] < 50) {
                arr[i] *= 2;
            }
        }

        return arr;
    }
}