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

문제

https://school.programmers.co.kr/learn/courses/30/lessons/181835

프로그래머스

 

문제 설명

정수 배열 arr와 자연수 k가 주어집니다.
만약 k가 홀수라면 arr의 모든 원소에 k를 곱하고, k가 짝수라면 arr의 모든 원소에 k를 더합니다.
이러한 변환을 마친 후의 arr를 return 하는 solution 함수를 완성해 주세요.

 

풀이

answer 배열의 크기를 arr.length로 지정후 반복문을 순회하면서 조건에 따라 값을 지정해줬다.

 

코드

class Solution {
    public int[] solution(int[] arr, int k) {
        int[] answer = new int[arr.length];
        if (k % 2 == 1) {
            for (int i = 0; i < arr.length ; i++) {
                answer[i] = arr[i] *= k;
            }
        } else {
            for (int i = 0; i < arr.length ; i++) {
                answer[i] = arr[i] += k;
            }
        }
        return answer;
    }
}