본문 바로가기
알고리즘/[프로그래머스] JAVA

[코딩 기초 트레이닝] 조건에 맞게 수열 변환하기 3

by 코딩맛집 2024. 1. 5.

문제 :

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

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

 

해결 :

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

 

다른 사람 풀이 :

import java.util.*;

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