[Lv.0] 조건에 맞게 수열 변환하기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];
        
        //홀수일 경우
        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;
    }
}

 

 

int[] answer = new int[arr.length];

- answer 배열을 빈 배열로 초기화하는 대신 결과를 담을 배열로 초기화해야함

 

- if 조건문으로 홀수/짝수 틀 먼저 주고 for문으로 배열 값 담기