[Programmers] 조건에 맞게 수열 변환하기 3

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

프로그래머스

[ 문제 설명 ]

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

[ 제한사항 ]

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

[ 입출력 예 ]

arr	k	result
[1, 2, 3, 100, 99, 98]	3	[3, 6, 9, 300, 297, 294]
[1, 2, 3, 100, 99, 98]	2	[3, 4, 5, 102, 101, 100]

[ 풀이 ]

public class Sequence_conversion {
	public static int[] solution(int[] arr, int k) {
        int[] answer = new int[arr.length];
        
        if (k % 2 != 0 ) {
        	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;
    }
	public static void main(String[] args) {
		int[] arr = {1, 2, 3, 100, 99, 98};
		int k = 3;
		System.out.println(solution(arr, k));
	}
}