[프로그래머스(Java)] 배열의 길이에 따라 다른 연산하기

 

Level.0 

 

문제

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

* 제한사항
- 1 ≤ arr의 길이 ≤ 1,000
- 1 ≤ arr의 원소 ≤ 1,000
- 1 ≤ n ≤ 1,000

 

 

풀이.1

import java.util.stream.*;

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

 

stream을 이용한 한줄코드.

속도도 느리고 보기에도 좋지 않다. 

한줄이란것 빼곤 장점이 없어서 아래와 같이 다시 풀이하였다. 

 

 

풀이.2 

class Solution {
    public int[] solution(int[] arr, int n) {
        int[] answer = arr.clone();
        
        for(int i=(arr.length+1)%2; i<arr.length; i+=2){
            answer[i] += n;
        }
        return answer;
        
    }
}

 

 

 

 

---

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

프로그래머스