[Golang] 프로그래머스 Lv.1 음양 더하기

문제 설명

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

제한사항

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

입출력 예

absolutes	signs	result
[4,7,12]	[true,false,true]	9
[1,2,3]	[false,false,true]	0

입출력 예 설명

입출력 예 #1

• signs가 [true,false,true] 이므로, 실제 수들의 값은 각각 4, -7, 12입니다.
• 따라서 세 수의 합인 9를 return 해야 합니다.

입출력 예 #2

• signs가 [false,false,true] 이므로, 실제 수들의 값은 각각 -1, -2, 3입니다.
• 따라서 세 수의 합인 0을 return 해야 합니다.

풀이

func solution(absolutes []int, signs []bool) int {

	var total int
	for i, v := range absolutes {
		if !signs[i] {
			v *= -1
		}
		total += v
	}

	return total
}

• absolutes 를 순회하며, 현재 절댓값이 음수인지 양수인지 signs[i]로 확인
• 만약 부호가 음수일 경우, 현재 v 의 값이 -1 을 곱해줌
• 부호가 정해진 v 값을 total 에 더해줌

테스트코드

유닛 테스트

const TESTCOUNT = 2

var (
	absolutes = [TESTCOUNT][]int{
		{4, 7, 12},
		{1, 2, 3},
	}
	signs = [TESTCOUNT][]bool{
		{true, false, true},
		{false, false, true},
	}
	result = [TESTCOUNT]int{9, 0}
)

func TestSolution(t *testing.T) {
	for i := 0; i < TESTCOUNT; i++ {
		fmt.Printf("absolutes : %v\n", absolutes[i])
		fmt.Printf("signs : %v\n", signs[i])
		fmt.Printf("result : %v\n", result[i])
		fmt.Printf("solution result : %v\n", solution(absolutes[i], signs[i]))

	}
}

$ go test -v
=== RUN   TestSolution
absolutes : [4 7 12]
signs : [true false true]
result : 9
solution result : 9
absolutes : [1 2 3]
signs : [false false true]
result : 0
solution result : 0
--- PASS: TestSolution (0.00s)
PASS
ok      golang-coding-test/Lv1/sum_of_signed_integers   0.205s

Github 바로가기