[JAVA 문제 풀이] 207. 조건에 맞게 수열 변환하기 3

프로그래머스 (181835)

Stupefyee

Apr 17, 2025

Contents

내가 작성한 코드 1 (반복문)내가 작성한 코드 2 (스트림)

school.programmers.co.kr

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

내가 작성한 코드 1 (반복문)


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

        return arr;
    }
}

내가 작성한 코드 2 (스트림)


import java.util.*;

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