프로그래머스 - 삼각형의 완성 - C++

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

프로그래머스

 

#include <string>
#include <vector>

using namespace std;

int solution(vector<int> sides) {
    int answer = 0;
    
    // 나머지가 가장 길 경우
    int small = (sides[0]>sides[1])?sides[1]:sides[0];
    int big = (sides[0]<sides[1])?sides[1]:sides[0];
    
    for(int i=big;i<small+big;i++)
    {
        answer++;
    }
    
    // 나머지가 가장 길지 않을 때
    for(int i=big-small+1;i<big;i++)
    {
        if(big<=small+i)answer++;
    }
    
    
    return answer;
}

 

설명

삼각형의 나머지 한 변이 될 수 있는 길이의 개수를 구하는 문제이다. 

두 가지의 경우로 나눠서 구해볼 수 있다.

1. 나머지 한 변이 가장 길 경우

2. 나머지 한 변이 가장 길지 않을 경우

 

첫번째 경우는 

// 나머지가 가장 길 경우
    int small = (sides[0]>sides[1])?sides[1]:sides[0];
    int big = (sides[0]<sides[1])?sides[1]:sides[0];
    
    for(int i=big;i<small+big;i++)
    {
        answer++;
    }

이렇게 구할 수 있다.

small, big 은 삼항연산자를 이용해서 주어진 것들을 나눠서 저장해놓은 것이다.

 

삼각형의 가장 긴 변은 나머지 두 변의 길이 합보다 작아야한다.를 이용하야 한다.

 

나머지가 가장 길어야 한다면 당연히 big 이상이어야 한다. big 부터인 이유는 2,2,1 도 성립하기 때문이다. 

또 두 변의 길이합인 small+big -1까지만 for문을 돌며 개수를 구하면 된다.

 

 

// 나머지가 가장 길지 않을 때
    for(int i=big-small+1;i<big;i++)
    {
        if(big<=small+i)answer++;
    }

나머지가 가장 길지 않으려면 big 이하여야 한다. 하지만 위에서 이미 동일한 두변이 있는 경우를 계산했으니

big -1 까지만 계산한다.

 

big - small + 1 부터 시작하는 것은 big - small 보다 작다면 big 을 제외한 두변의 길이의 합이 big을 못넘겨 삼각형의 조건이 성립하지 않기 때문이다. 

+1 은 big-small 부터라면 big과 small+i가 동일할때도 계산하기 때문에 말했듯이 위에서 이미 계산하여 +1한다.

 

따라서 big-small+1 부터 big-1 까지 계산하여 조건에 맞으면 더한다.