[BOJ | Python] 1059번: 좋은 구간

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

 

문제

정수 집합 S가 주어졌을때, 다음 조건을 만족하는 구간 [A, B]를 좋은 구간이라고 한다.

• A와 B는 양의 정수이고, A < B를 만족한다.
• A ≤ x ≤ B를 만족하는 모든 정수 x가 집합 S에 속하지 않는다.

집합 S와 n이 주어졌을 때, n을 포함하는 좋은 구간의 개수를 구해보자.

 

입력

첫째 줄에 집합 S의 크기 L이 주어진다. 둘째 줄에는 집합에 포함된 정수가 주어진다. 셋째 줄에는 n이 주어진다.

 

 

출력

첫째 줄에 n을 포함하는 좋은 구간의 개수를 출력한다.

 

제한

• 1 ≤ L ≤ 50
• 집합 S에는 중복되는 정수가 없다.
• 집합 S에 포함된 모든 정수는 1보다 크거나 같고, 1,000보다 작거나 같다.
• 1 ≤ n ≤ (집합 S에서 가장 큰 정수)

 

풀이

처음엔 S의 가장 작은 원소보다 n이 작으면 무조건 답이 0이라고 생각해서 오답이었는데, 

질문게시판의 힘을 빌려 해결했다...ㅎ

L = int(input())  # S의 길이
S = sorted(list(map(int, input().split())))
n = int(input())  # n을 포함하면서 S의 요소는 포함되지 않음 -> 좋은 구간

# n이 집합 S에 이미 포함되어 있으면 좋은 구간이 없다.
if n in S:
    print(0)
    exit()

# n보다 작은 가장 큰 수와 n보다 큰 가장 작은 수 찾기
left = 0
right = S[0]

for s in S:
    if s < n:
        left = s
    elif s > n:
        right = s
        break

# 좋은 구간의 경우의 수 계산
ans = (n - left - 1) * (right - n) + (right - n - 1)
print(ans)