[Kotlin] 음양 더하기 (프로그래머스 알고리즘)

 

문제

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

 

 

---

 

접근

 

• 먼저 absoulutes[i] 값에 signs[i]의 부호값을 더한 배열을 만든다.
• 그리고 만들어진 배열 원소의 합을 구하고 출력하면 된다고 생각했다.

//오류 코드
// 1. absoulutes[i] 값에 signs[i]의 부호값을 더한 배열을 만듦
// 2. 해당 배열 원소의 합을 구하고 출력

class Solution {
    fun solution(absolutes: IntArray, signs: BooleanArray): Int {
        
        var sumArray = intArrayOf()
        for(i in 0..absolutes.size){
            if(signs[i] == true) sumArray += absolutes[i]
            else  sumArray += -absolutes[i]
        }
        return sumArray.sum()       
    }
}

• 오류: Exception in thread "main" java.lang.ArrayIndexOutOfBoundsException: Index 3 out of bounds for length 3
• absolutes.size 만큼 for문을 반복하면 마지막에 인덱스 최대값을 초과하므로, -1을 해주면 된다.

 

---

 

성공 코드

 

class Solution {
    fun solution(absolutes: IntArray, signs: BooleanArray): Int {
        
        var sumArray = intArrayOf()
        for(i in 0..absolutes.size-1){
            if(signs[i] == true) sumArray += absolutes[i]
            else  sumArray += -absolutes[i]
        }
        return sumArray.sum()       
    }
}

 

---

 

설명

 

1. 결과값을 넣을 배열을 만든다.
2. absolutes 배열의 길이 - 1만큼 for문을 반복해주면서, absolutes 각 원소에 부호를 붙여 결과 배열에 넣는다.
3. 결과 배열 원소의 합을 .sum()으로 출력한다.

 

---

 

다른 방법

 

class Solution {
    fun solution(absolutes: IntArray, signs: BooleanArray): Int {

        val array = IntArray(signs.size)
        repeat(signs.size) {
            if(signs[it]) array[it] = absolutes[it]
            else array[it] = -(absolutes[it])
        }
        return array.sum()
    }
}

• for문 대신 repeat를 이용한 풀이이다.

 

class Solution {
    fun solution(absolutes: IntArray, signs: BooleanArray) =
        absolutes.foldIndexed(0) { idx, acc, num -> acc + if (signs[idx]) num else -num }
}

• foldIndexed()라는 함수를 사용한 정석적인 풀이다.
• foldIndexed 함수
  - foldIndexed 함수는 컬렉션의 각 요소에 대해 지정된 초기값과 람다 함수를 사용하여 연산을 수행함
  - 람다 함수의 첫 번째 매개변수는 인덱스, 두 번째는 누산기(accumulator), 세 번째는 컬렉션의 현재 요소
• acc + if (signs[idx]) num else -num
  - 이 부분은 각 요소에 대한 연산을 수행
  - if 표현식을 사용하여 signs[idx]의 값에 따라 num을 더할지 빼줄지를 결정
  - signs[idx]가 true이면 num을 더함
  - signs[idx]가 false이면 num을 뺌
• 전체 표현식 설명
  - foldIndexed 함수는 주어진 초기값인 0부터 시작하여 배열 absolutes의 각 요소에 대해 람다 함수를 실행
  - 각 요소에 대해 연산을 수행한 결과를 누산기에 더하여 최종적인 결과를 반환
  - 최종적으로 반환되는 값은 모든 요소에 대해 acc + if (signs[idx]) num else -num 연산을 수행한 결과

 

💡 fold() 함수

val absolutes = intArrayOf(1, 2, 3, 4, 5, 6, 7, 8, 9, 10)

val sumNumber = absolutes.fold(100) { sum, element ->
    sum + element
}
println(sumNumber) // 출력값: 155
// 초기값 100에 absolutes 배열의 element 들의 합 55가 더해진 값 155가 출력

• 주어진 Int로만 구성된 Array 혹은 Collection 의 모든 element들을 연속적으로 불러와 계산
• fold 함수는 파라미터로 initial값(초기값)을 받음

 

💡 foldIndexed() 함수

val absolutes = intArrayOf(1, 2, 3, 4, 5, 6, 7, 8, 9, 10)
val signs = booleanArrayOf(false, false, true, true, false, false, true, true, true, true)

val sumWithIndex = absolutes.foldIndexed(0) {
    idx, sum, element ->
    println("index::: $idx \nsum::: $sum \nelement::: $element")
    sum + if (signs[idx]) element else -element
}

• 다른 건 fold()함수와 똑같고, index를 같이 가져온다는 것이 fold 함수와 다른 점