[프로그래머스] 다음에 올 숫자 (C++)

다음에 올 숫자 / Lv.0

---

문제  설명 )

등차수열 혹은 등비수열 common이 매개변수로 주어질 때, 마지막 원소 다음으로 올 숫자를 return 하도록 solution 함수를 완성해보세요.

 

 

 

제한 사항 )

• 2 < common의 길이 < 1,000
• -1,000 < common의 원소 < 2,000
  • common의 원소는 모두 정수입니다.
• 등차수열 혹은 등비수열이 아닌 경우는 없습니다.
• 등비수열인 경우 공비는 0이 아닌 정수입니다.

 

 

 

입출력 예 )

 

 

입출력 예 설명 )

입출력 예 #1

• [1, 2, 3, 4]는 공차가 1인 등차수열이므로 다음에 올 수는 5이다.

입출력 예 #2

• [2, 4, 8]은 공비가 2인 등비수열이므로 다음에 올 수는 16이다.

 

 

 

풀이)

#include <string>
#include <vector>
 
using namespace std;
 
int solution(vector<int> common) {
    int answer = 0;
 
    
    // 무조건 등차수열 or 등비수열만 주어진다.
    int temp;
    bool flag = 0;
    // 등차수열인 경우
    if(common[2] == common[1] + (common[1] - common[0])) 
    {
        flag = 0;
        temp = common[1] - common[0];
    }
    // 등비수열인 경우
    else 
    {
        flag = 1;
        temp = common[1] / common[0];
    }
        
    // 등비수열인 경우
    if (flag)
    {
        answer = common.back() * temp;
    } 
    // 등차수열인 경우
    else 
    {
        answer = common.back() + temp;
    }
    
    return answer;
}

 

---

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

프로그래머스