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

🔎 문제 설명

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

• k가 홀수라면 arr의 모든 원소에 k를 곱합니다.
• k가 짝수라면 arr의 모든 원소에 k를 더합니다.

이후 변환된 배열을 반환하는 solution 함수를 작성하세요.

---

✨ 초기 풀이

처음에는 아래와 같이 for-each 반복문 안에서 조건문을 돌리며 구현했습니다:

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

처음에는 문제를 빠르게 푸는 데 집중해서 작성했습니다.
하지만 k는 반복문마다 바뀌는 값이 아니므로, 매번 if 조건을 검사할 필요가 없다는 점이 아쉬웠습니다.

---

🔧 개선된 풀이

k의 조건에 따라 코드를 두 갈래로 분리해서, 불필요한 조건문 반복을 제거해보았습니다:

import java.util.stream.IntStream;

class Solution {
    public int[] solution(int[] arr, int k) {
        return k % 2 == 0 ?
            IntStream.of(arr).map(n -> n + k).toArray() :
            IntStream.of(arr).map(n -> n * k).toArray();
    }
}

이렇게 하면 k가 홀수인지 짝수인지를 한 번만 검사하고, 이후 반복문에서는 계산만 수행하게 됩니다.

---

🧠 느낀 점

• if문 하나지만, 반복문 안에 매번 등장하면 성능 저하로 이어질 수 있습니다.
• 특히 입력 데이터가 커질수록 이런 작은 최적화가 중요한 습관이 됩니다.
• 문제 풀이에서는 조건이 고정값이라면 반복 전에 분기 처리하는 습관을 들이는 것이 좋습니다.

---

💡 보너스: Java 스트림으로 구현하기

Java의 Stream API를 사용하면 더 간결하게 작성할 수도 있습니다:

import java.util.stream.IntStream;

class Solution {
    public int[] solution(int[] arr, int k) {
        return k % 2 == 0 ?
            IntStream.of(arr).map(n -> n + k).toArray() :
            IntStream.of(arr).map(n -> n * k).toArray();
    }
}

다만 스트림은 가독성은 좋지만 성능 측면에서는 단순 반복보다 약간 느릴 수 있으니, 상황에 맞게 사용하는 것이 좋습니다.

---

✅ 정리

• 조건이 반복문 내부에서 불필요하게 반복되지 않도록 처리하는 것이 중요.
• Java의 기초 문법을 활용하면서도, 성능 최적화 습관을 기를 수 있는 좋은 문제였다.