[JAVA]1111 IQ 테스트

 📚 문제  

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

 

🔍 아이디어  

절댓값 내의 범위가 -100, 100 사이라 하여 완전 탐색을 고려하였다.

1. 실수 포인트 : -100 ~ 100 사이에서 A가 반드시 나오지 않는다. B가 커버칠수도 있다 (-100 , 99 와 같은 케이스 )

input:
3
-100 -99 2

output:
10203

// 수열의 세 번째 수로 2가 아니라 2 이상 100 이하의 수 중 아무거나 사용하여도
// 모두 곱해지는 수의 절댓값이 100을 넘습니다.
// -100 * 101 + 10001 = -99, -99 * 101 + 10001 = 2

출처 : https://www.acmicpc.net/board/view/107405  

2. 실수 포인트 : 중복 값을 비교하는 부분에서  if (isAble && next != foundNum) { 

next의 값의 초기값을 잘 정해야 한다. 처음에 next로 -1을 지정해두니 -1이 다음값으로 나올 수 있는 경우가 무시된다. 

Integer.MAX_VALUE로 변경 후 통과하였다. 

 

📝 풀이 

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringBuilder sb = new StringBuilder();
        StringTokenizer st;

        int n = Integer.parseInt(br.readLine());
        int[] arr = new int[n];
        st = new StringTokenizer(br.readLine());
        for (int i = 0; i < n; i++) {
            arr[i] = Integer.parseInt(st.nextToken());
        }

        if (n == 2) {
            if (arr[0] == arr[1]) {
                System.out.println(arr[0]);
                return;
            }
        }

        if (n <= 2) { //1 또는 2
            System.out.println("A");
            return;
        }

        int foundCnt = 0;
        int next = Integer.MAX_VALUE;
        int tempB;
        for (int a = -200; a <= 200 && foundCnt < 2; a++) {
            boolean isAble = true;
            tempB = arr[1] - arr[0] * a;
            for (int i = 1; i < n; i++) {
                if (arr[i - 1] * a + tempB != arr[i]) {
                    isAble = false;
                    break;
                }
            }

            int foundNum = arr[n - 1] * a + tempB;
            if (isAble && next != foundNum) {
//                System.out.println("FOUND : " + foundNum);
//                System.out.println("ABLE: " + a + " " + tempB);
                next = foundNum;
                foundCnt++;
            }
        }
//        System.out.println(foundCnt);
        System.out.println(foundCnt == 0 ? "B" : (foundCnt >= 2 ? "A" : next));
    }
}