[백준 1111] IQ Test C++

 

www.acmicpc.net/problem/1111

1111번: IQ Test

 

 

처음에 x가 0이 되는 경우를 깜박해서 ZeroDivisionError 가 발생했었다

같은 수가 연속으로 나올경우 x값이 0이 된다!

 

모든 점이 한 직선위에 있음을 알아차리고, 직선의 기울기와 y절편을 구한다

구한 기울기와 y절편이 모든 좌표 쌍에 적용된다면 a, b는 구해진 것임

 

그리고 N==1, N==2일때의 기저조건 예외처리를 해주고 마무리를 했다

#include <iostream>
 
using namespace std;
 
int N, cnt = 0;
int arr[51];
 
 
void test(int a, int b){
    for(int i = 0; i < N-1; i++){
        if(arr[i] * a + b != arr[i+1]) return;
    }
    cnt = 1;
}
 
// 모든 점이 한 직선위에 있음!!
void make_ab(){
    int a = 0, b = 0, x, y;
    
    x = arr[1] - arr[0];
    y = arr[2] - arr[1];
    if(x != 0) a = y / x;
    b = arr[1] - arr[0] * a;
    
    test(a, b);
    
    if(cnt) cout << arr[N-1] * a + b;
    else cout << 'B';
}
 
int main(){
    cin >> N;
    for(int i = 0; i < N; i++){
        cin >> arr[i];
    }
    
    if(N == 1) cout << 'A';
    else if(N == 2){
        if(arr[1] == arr[0]) cout << arr[0];
        else cout << 'A';
    }
    else make_ab();
    
    return 0;
}