package programmers;
import java.util.stream.IntStream;
class 배열의_길이에_따라_다른_연산하기_Java_230901 {
/*
정수 배열 arr과 정수 n이 매개변수로 주어집니다.
arr의 길이가 홀수라면 arr의 모든 짝수 인덱스 위치에 n을 더한 배열을,
arr의 길이가 짝수라면 arr의 모든 홀수 인덱스 위치에 n을 더한 배열을 return 하는 solution 함수를 작성해 주세요.
*/
public int[] solution(int[] arr, int n) {
if (arr.length % 2 == 0){
for(int i = 0; i < arr.length; i++){
if (i % 2 != 0){
arr[i] += n;
}
}
}else{
for(int i = 0; i < arr.length; i++){
if (i % 2 != 1){
arr[i] += n;
}
}
}
return arr;
}
public int[] solution_best(int[] arr, int n) {
for(int i = arr.length % 2 == 0 ? 1 : 0; i < arr.length; i += 2) {
arr[i]+=n;
}
return arr;
}
public int[] solution_stream(int[] arr, int n) {
IntStream.iterate((arr.length + 1) % 2, i -> i < arr.length, i -> i + 2)
.forEach(i -> arr[i] += n);
return arr;
}
}
package programmers
import java.util.stream.IntStream
class 배열의_길이에_따라_다른_연산하기_Kotlin_230901 {
fun solution(arr: IntArray, n: Int): IntArray {
if (arr.size % 2 == 0) {
for (i in arr.indices) {
if (i % 2 != 0) {
arr[i] += n
}
}
} else {
for (i in arr.indices) {
if (i % 2 != 1) {
arr[i] += n
}
}
}
return arr
}
fun solution_short(arr: IntArray, n: Int): List<Int> {
return arr.indices.map { arr[it] + if (it % 2 == (if (arr.size % 2 == 0) 1 else 0)) n else 0 }
}
}
댓글
댓글 쓰기