🗃️문제 설명
N개의 원소를 포함하고 있는 수열이 오름차순으로 정렬되어 있습니다.
이때 이 수열에서 x가 등장하는 횟수를 계산하세요.
예를 들어 수열 {1, 1, 2 ,2 ,2 ,2, 3}이 있을 때 x=2라면, 현재 수열에서 값이 2인 원소가 4개이므로 4를 출력합니다.
단, 이 문제는 시간 복잡도 O(logN)으로 알고리즘을 설계하지 않으면 ‘시간 초과’ 판정을 받습니다.
첫째 줄에 N과 x가 정수 형태로 공백으로 구분되어 입력됩니다. (1 ≤ N ≤ 1,000,000), (-10e9 ≤ x ≤ 10e9)
둘째 줄에 N개의 원소가 정수 형태로 공백으로 구분되어 입력됩니다. (-10e9 ≤ 각 원소의 ≤ 10e9)
수열의 원소 중에서 값이 x인 원소의 개수를 출력합니다. 단, 값이 x인 원소가 하나도 없다면 -1을 출력합니다.
🖥️코드
import sys
input = sys.stdin.readline
from bisect import bisect_left, bisect_right
N, x = map(int, input().split())
data = list(map(int, input().split()))
if bisect_left(data, x) == bisect_right(data, x):
print(-1)
else:
print(bisect_right(data, x) - bisect_left(data, x))🧠아이디어
알고리즘 유형 : 이진 탐색
O(log N)으로 설계하라는 말은 결국 이진 탐색을 하라는 의미
🔒문제 출처 및 참고 코드
이것이 코딩테스트다 with 파이썬 - 정렬된 배열에서 특정 수의 개수 구하기
모범 답안 - 정렬된 배열에서 특정 수의 개수 구하기
