[79] JAVA 배열의 길이에 따라 다른 연산하기

문제 설명

정수 배열 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) {
    	return IntStream.rangeClosed(0, arr.length-1)
        .map(
        i -> arr.length % 2 ? 
            (i % 2 == 0 ? arr[i] + a : arr[i])
            :
            (i % 2 != 0 ? arr[i] + a : arr[i])
        ).toArray();
    }
}

 

 

[클린 풀이]

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

반복하는 곳에 i의 초기값 결정 기준을 배열의 길이의 홀/짝 여부를 기준으로 1 혹은 0 값을 반환해준다.

해당 값을 통해 i를 2씩 증가시키면서 n값을 더해준다.

i가 0부터 시작하면 0 2 4 ...

i가 1부터 시작하면 1 3 5 ...

짝수 혹은 홀수에 대한 값에만 n값을 더하게 된다..

유레카...

 

 

[희표좌 풀이]

import java.util.stream.IntStream;

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

range를 통해 인덱스로 사용될 i값이 만들어졌다.

배열의 i번째 요소에는 데이터가 더해진다.

이 데이터는 i가 짝수거나 홀수일때 달라진다.

배열의 길이가 짝수일때 홀수 요소들에만 n을 더하고

배열의 길이가 홀수일때 짝수 요소들에만 n을 더한다.

i%2를 기준으로 짝수 홀수 여부는 0혹은 1값이 기준이 된다.

위 식은 먼저 배열의 길이의 짝수, 홀수 여부를 계산한다.

짝수일 경우 홀수에 해당하는 1값

홀수일 경우 짝수에 해당하는 0값을 반환하고.

반환된 값을 통해 index의 짝수 홀수 여부로 사용한다.

이렇게 되면 위의 조건명제에서 각자 반대되는 조건을 만족하게 된다.

최종적으로 모든 조건을 만족하면 n을 더하고 만족하지 못하면 0일 더한다.