음양 더하기

jericho·2023년 12월 9일

PS(문제풀이)

목록 보기
2/12

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

자연수와 부호 배열을 주고 합을 반환한다.

 - 풀이 1
fun solution(absolutes: IntArray, signs: BooleanArray): Int {
    var answer = 0
    for (i in absolutes.indices) answer += absolutes[i] * if (signs[i]) 1 else -1
    return answer
}

누적합을 계산할 변수를 두고 부호를 확인해 1 혹은 -1을 곱해 더함.

 - 풀이 2
fun solution(absolutes: IntArray, signs: BooleanArray): Int =
    absolutes.indices.sumOf { if (signs[it]) absolutes[it] else -absolutes[it] }

sumOf 적용. 부호 확인해 값 그대로 + - 붙여 더함.

indicies 대신 withIndexed 적용하면 더 좋을듯?

0개의 댓글