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

🐤 목표

앞으로 매일 꾸준히 코딩테스트를 진행하면서 단계를 높여가보자.

문제 설명

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

제한사항

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

입출력의 예

문제 풀이

class Solution {
    public int[] solution(int[] arr) {
        // answer 배열을 생성, 길이는 arr 배열의 길이와 동일
        int[] answer = new int[arr.length];
        
        // 반복문으로 arr 배열의 각 원소에 대해
        for (int i=0; i<arr.length; i++) {
            
            // 조건문으로 50미만이고, 홀수인 경우와 50보다 크거나 같고, 짝수인 경우, 그 외에는 그대로 answer 배열로 반환
            if(arr[i] < 50 && arr[i] % 2 != 0) {
                answer[i] = arr[i] * 2;
            } else if (arr[i] >= 50 && arr[i] % 2 == 0) {
                answer[i] = arr[i] / 2;
            } else {
               answer[i] = arr[i]; 
            }
        }
        return answer;
    }
}