[백준]#1059 수2

문제

Lucky Set이란 정수의 집합이다.

구간 [A,B]란 A보다 크거나 같고, B보다 작거나 같은 모든 정수가 있는 구간이다. 이때, A와 B는 모두 양수이고, B는 A보다 크다.

구간 [A,B]가 Unlucky가 되기 위해선 구간에 속한 모든 정수가 Lucky Set에 없어야 한다.

Lucky Set과 N이 주어질 때, N을 포함하는 Unlucky 구간의 개수를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 Lucky Set에 포함된 숫자의 개수 L이 주어진다. 둘째 줄에는 L개의 수가 주어진다. 이 수는 1,000보다 작거나 같은 자연수이고, L은 50보다 작거나 같은 자연수이다. 그리고 중복되지 않는다. 마지막 줄에는 N이 주어진다. N은 Lucky Set에서 가장 큰 수보다 작거나 같은 자연수이다.

출력

첫째 줄에 문제의 정답을 출력한다.

예제 입력 1

4
1 7 14 10
2

예제 출력 1

4

풀이

이 문제는 수학을 잘하면 금방 풀 수 있는 문제이다. 수학적 접근이 핵심인 문제.

import java.util.ArrayList;
import java.util.Scanner;

public class Main {

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        ArrayList<Integer> list = new ArrayList<>();
        int L = sc.nextInt();

        for(int i=0; i<L; i++) {
            list.add(sc.nextInt());
        }
        int N = sc.nextInt();
        list.sort(null);

        if(!list.contains(N)) {
            list.add(N);
            list.sort(null);

            if(list.get(0)==N) {
                int x = list.get(list.indexOf(N)+1);
                System.out.println((x-N)*N-1);
            }

            else {
                int upper = list.get(list.indexOf(N)+1)-N;
                int lower = N-list.get(list.indexOf(N)-1);
                System.out.println(upper*lower-1);
            }
        }
        else
            System.out.println(0);
    }
}