[이코테 이진 탐색] 정렬된 배열에서 특정 수의 개수 구하기

# 문제

N개의 원소를 포함하고 있는 수열이 오름차순으로 정렬되어 있습니다. 이때 이 수열에서 x가 등장하는 횟수를 계산하세요. 예를 들어 수열 {1, 1, 2, 2, 2, 2, 3}이 있을 때 x = 2라면, 현재 수열에서 값이 2인 원소가 4개이므로 4를 출력합니다.

단, 이 문제는 시간 복잡도 O(logN)으로 알고리즘을 설계하지 않으면 '시간 초과' 판정을 받습니다.

 

---

내 풀이

from bisect import bisect_left, bisect_right


def count_x(array, x):
    left_idx = bisect_left(array, x)  # 정렬된 배열에 x를 삽입할 가장 왼쪽 인덱스 찾기
    right_idx = bisect_right(array, x)  # 정렬된 배열에 x를 삽입할 가장 왼쪽 인덱스 찾기
    return right_idx - left_idx  # x의 개수


n, x = map(int, input().split())
numbers = list(map(int, input().split()))

result = count_x(numbers, x)
# result 값이 0인 경우, -1 리턴
if not result:
    print(-1)
else:
    print(result)