배열의 길이에 따라 다른 연산하기 [자바 삼항연산자에서 return문 사용 불가. /js는 가능.][함수의 선언 타입대로 return 값 반환 필수]

하얀성·2023년 7월 13일
0

정수 배열 arr과 정수 n이 매개변수로 주어집니다. arr의 길이가 홀수라면 arr의 모든 짝수 인덱스 위치에 n을 더한 배열을, arr의 길이가 짝수라면 arr의 모든 홀수 인덱스 위치에 n을 더한 배열을 return 하는 solution 함수를 작성해 주세요.

제한사항
1 ≤ arr의 길이 ≤ 1,000
1 ≤ arr의 원소 ≤ 1,000
1 ≤ n ≤ 1,000
입출력 예
arr n result
[49, 12, 100, 276, 33] 27 [76, 12, 127, 276, 60][444, 555, 666, 777] 100 [444, 655, 666, 877]
입출력 예 설명
입출력 예 #1

예제 1번의 arr의 길이는 5로 홀수입니다. 따라서 arr의 짝수 인덱스 0, 2, 4에 주어진 n 값인 27을 더하면 [76, 12, 127, 276, 60]이 됩니다. 따라서 [76, 12, 127, 276, 60]를 return 합니다.
입출력 예 #1

예제 2번의 arr의 길이는 4로 짝수입니다. 따라서 arr의 홀수 인덱스 1, 3에 주어진 n 값인 100을 더하면 [444, 655, 666, 877]이 됩니다. 따라서 [444, 655, 666, 877]를 return 합니다.


<제출답안>

import java.util.stream.IntStream;

class Solution {
    public int[] solution(int[] arr, int n) {

        if(arr.length % 2 == 0){
            return IntStream.range(0, arr.length).map(i -> (i % 2 == 1)? arr[i]+n : arr[i]).toArray();
        }else{
            return IntStream.range(0, arr.length).map(i -> (i % 2 ==0)? arr[i]+n : arr[i]).toArray();
                } 
    }
}

stream선언을 통해 배열을 바꾸는 메소드들을 사용할 수 있다.(filter, map, filter, reduce 등)

하지만 stream 사용 후에 값을 반환하려면 Main 함수의 선언 타입대로 바꿔주고서 반환해야한다. 배열이라면 toArray()메서드를 사용한다.


profile
기적을 한웅큼 품은 js러버

0개의 댓글