백준 1731번: 추론

백준 온라인 저지(BOJ) 1731번 문제

https://www.acmicpc.net/problem/1731

1. 문제 요약

입력 받은 정수 배열이 등차 수열인지 등비 수열인지 판단하고 다음 수열의 원소를 출력 하는 문제

2. 알고리즘

입력 받은 정수 배열에서 두 번째 원소를 첫 번째 원소로 나누어서 나머지가 0 이면 등비 수열이고 나머지가 있다면 등차 수열로 처리 한다.

3. 코드

#include <iostream>
#include <algorithm> // min
#include <functional>
#include <math.h>
#include <string>
#include <string.h>
#include <vector>
#include <map>
#include <sstream>
#include <queue>
using namespace std;
 
int main() {
 
    std::ios::sync_with_stdio(false); cin.tie(0);
 
    int n; cin >> n;
 
    vector<long long int> arr(n);
    for(int i=0; i<n; i++) {
        cin >> arr[i];
    }
 
    if (arr[1] % arr[0] == 0) {
        // 등비 수열
        int cand = arr[1] / arr[0];
        cout << arr[arr.size()-1] * cand << endl;
    } else {
        // 등차 수열
        int cand = arr[1] - arr[0];
        cout << arr[arr.size()-1] + cand << endl;
    }
    return 0;
}