Golang Algorithm - 삼각형의 완성조건 (2) (프로그래머스)

 

삼각형의 완성조건 (2)
문제 설명
선분 세 개로 삼각형을 만들기 위해서는 다음과 같은 조건을 만족해야 합니다.

가장 긴 변의 길이는 다른 두 변의 길이의 합보다 작아야 합니다.
삼각형의 두 변의 길이가 담긴 배열 sides이 매개변수로 주어집니다. 나머지 한 변이 될 수 있는 정수의 개수를 return하도록 solution 함수를 완성해주세요.

제한사항
sides의 원소는 자연수입니다.
sides의 길이는 2입니다.
1 ≤ sides의 원소 ≤ 1,000

입출력 예
sides result
[1, 2] 1
[3, 6] 5
[11, 7] 13

입출력 예 설명
입출력 예 #1

두 변이 1, 2 인 경우 삼각형을 완성시키려면 나머지 한 변이 2여야 합니다. 따라서 1을 return합니다.

입출력 예 #2
가장 긴 변이 6인 경우
될 수 있는 나머지 한 변은 4, 5, 6 로 3개입니다.
나머지 한 변이 가장 긴 변인 경우
될 수 있는 한 변은 7, 8 로 2개입니다.
따라서 3 + 2 = 5를 return합니다.

입출력 예 #3
가장 긴 변이 11인 경우
될 수 있는 나머지 한 변은 5, 6, 7, 8, 9, 10, 11 로 7개입니다.
나머지 한 변이 가장 긴 변인 경우
될 수 있는 한 변은 12, 13, 14, 15, 16, 17 로 6개입니다.
따라서 7 + 6 = 13을 return합니다.

 

 

 

package main

import (
	"sort"
)

func sum(a int, b int) int {
	return a + b
}

func solution(sides []int) int {
	sort.Ints(sides)
	count := 0
	// 가장 긴 변의 길이는 나머지 두변의 길이의 합보다 작아야한다.
	init := sides[1] - sides[0] + 1
	// case a  : 가장 긴 값이 x일 때
	// 배열의 원소의 합 > x > 배열의 최댓값

	for i := sides[1] + 1; i < sum(sides[0], sides[1]); i++ {
		count += 1
	}

	// case b  : 가장 긴 변이 배열의 최댓값일 때
	// 배열의 최댓값 > x > 배열의 최댓값 - 배열의 최솟값
	for j := init; j <= sides[1]; j++ {
		count += 1
	}
	//	fmt.Println(count)
	return count
}

func main() {
	// sides := []int{3, 6}
	sides := []int{11, 7}
	solution(sides)
}