프로그래머스 자바 Lv.0 조건에 맞게 수열 변환하기 3

문제

정수 배열 arr와 자연수 k가 주어집니다.

만약 k가 홀수라면 arr의 모든 원소에 k를 곱하고, k가 짝수라면 arr의 모든 원소에 k를 더합니다.

이러한 변환을 마친 후의 arr를 return 하는 solution 함수를 완성해 주세요.

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

해결과정

k의 홀, 짝을 판별해서 조건에 맞게 계산합니다.

짝수 판별 : i % 2 == 0

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