간만의 알고리즘 공부 - 13 (음양 더하기)

문제 설명
어떤 정수들이 있습니다. 이 정수들의 절댓값을 차례대로 담은 정수 배열 absolutes와 이 정수들의 부호를 차례대로 담은 불리언 배열 signs가 매개변수로 주어집니다. 실제 정수들의 합을 구하여 return 하도록 solution 함수를 완성해주세요.

제한사항
- absolutes의 길이는 1 이상 1,000 이하입니다.
    - absolutes의 모든 수는 각각 1 이상 1,000 이하입니다.
- signs의 길이는 absolutes의 길이와 같습니다.
    - signs[i] 가 참이면 absolutes[i] 의 실제 정수가 양수임을, 그렇지 않으면 음수임을 의미합니다.

 

문제를 읽었을때 해당 귤 크기별로 리스트를 만들 생각을 했었고 없는 크기의 귤 리스트는 제거하고 큰 크기별로 리스트를 가지고 어떻게 해야할지 고민하다가 어떻게 해야할지 몰라서 고민하다가 첨삭 받았다.

 

내가 푼 방법

 fun solution(absolutes: IntArray, signs: BooleanArray): Int {
    return absolutes.mapIndexed {index, i -> if (signs[index]) i else -i}.sum()
}

 

 

풀었을 때 1점만 받은것 보니 그닥 잘풀지 못한듯 합니다. 그래서 다른 사람의 풀이를 봤습니다.

 

 

다른 사람 코드

 fun solution(absolutes: IntArray, signs: BooleanArray): Int {
    return  absolutes.foldIndexed(0) {index, sum, element -> if (signs[index]) sum + element else sum - element }
}

 

 

계산에 따른 순서만 잘 이뤄지면 훨씬 빠르게 계산이 가능하다. 처음에 내가 만든 코드의 경우 모든 숫자를 양수, 음수로 만든 다음에 전부 더하기를 했는데 foldIndexed 를 사용해서 for() 문을 돌때마다 더하고 빼기를 하니 훨씬 빠르게 계산이 됐습니다.. 간단하거나 얼마 안되는 데이터의 사칙연산은 큰 차이가 없겠으나 반복적이고 많은 양의 데이터를 다룬다면 당연히 차이가 나겠구나 싶습니다.

 

 

fold(초기값) {acc, number -> /* 로직 */}, foldIndex(초기값) {index, acc, number - > /* 로직 */}
acc 현재까지의 합, number 현재 요소, index 현재 요소의 인덱스