[프로그래머스] JAVA 0단계 조건에 맞게 수열 변경하기 3

조건에 맞게 수열 변경하기 3

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

 

풀이

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;
    }
}

 

결과

테스트 1
입력값 〉	[1, 2, 3, 100, 99, 98], 3
기댓값 〉	[3, 6, 9, 300, 297, 294]
실행 결과 〉	테스트를 통과하였습니다.

테스트 2
입력값 〉	[1, 2, 3, 100, 99, 98], 2
기댓값 〉	[3, 4, 5, 102, 101, 100]
실행 결과 〉	테스트를 통과하였습니다.