[python] 정렬된 배열에서 특정 수의 개수 구하기

- N개의 원소를 포함하고 있는 수열이 오름차순으로 정렬

- x가 등장하는 횟수 계산

 

예) [1, 1, 2, 2, 2, 2, 3] 에서 x = 2 이면, 현재 수열에서 값이 2인 원소가 4개

 

from bisect import bisect_left, bisect_right

def count_by_range(array, left_value, right_value):
  right_index = bisect_right(array, right_value)
  left_index = bisect_left(array, left_value)
  return right_index - left_index

n, x = map(int, input().split())  # n, x 입력
array = list(map(int, input().split()))  # 데이터 입력

# 값이 [x, x] 범위에 있는 데이터의 개수 계산
count = count_by_range(array, x, x)

# 값이 x인 원소가 존재하지 않는다면
if count == 0:
  print(-1)
# 값이 x인 원소가 존재하면
else:
  print(count)