조건에 맞게 수열 반환하기 3

문제 설명

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

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

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

제한사항

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

나의 코드

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

느낀 점

조건에 맞게 수열 변환하기 1번 문제와 같이 크게 어려움이 없었던 문제다. 오히려 2번이 제일 까다로웠음 ,,