프로그래머스 - 다음에 올 숫자 - C++

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

프로그래머스

 

#include <string>
#include <vector>

using namespace std;

int solution(vector<int> common) {
    int answer = 0;
    bool ch=false;
    int d;
    
    if(common[1]-common[0]==common[2]-common[1])
    {
        ch=false;
        d=common[1]-common[0];
    }
    else 
    {
        ch=true;
        d=common[1]/common[0];
    }
    
    answer=common[common.size()-1];
    
    if(ch)
        answer*=d;
    else
        answer+=d;
    
    return answer;
}

 

설명

공차수열인지 공비수열인지를 알아내서 다음에 올 숫자가 무엇인지 계산하는 문제이다. 

 

최소 3의 길이가 주어진다고 나와있으니 

if(common[1]-common[0]==common[2]-common[1])
    {
        ch=false;
        d=common[1]-common[0];
    }
    else 
    {
        ch=true;
        d=common[1]/common[0];
    }

위의 조건문으로 공차수열인지 공비수열인지를 확인한다. 

공차수열은 각 원소의 차이가 동일한 수열이니 if조건이 참이된다. 

 

ch 불리언변수로 표시하고 d 변수에 공차를 저장한다. 

마찬가지로 if 조건이 거짓이면 공비수열이란 것으로 

ch 변수에 true로 표시하고 d에 공비를 저장한다. 

 

 answer=common[common.size()-1];
    
    if(ch)
        answer*=d;
    else
        answer+=d;

가장 마지막 원소에 표시했던 ch 불리언 변수를 이용해 공비수열이면 d를 곱하고

공차수열이면 d를 더한다.

 

그리고 답을 리턴한다.