[이분탐색] 난이도1.5, Amazon 인터뷰 '고정점 찾기' (Python)

<문제>

고정점이란, 수열의 원소 중에서 그 값이 인덱스와 동일한 원소를 의미합니다.
예를 들어 수열 a={-15,-4,2,8,13}이 있을 때 a [2]=2이므로, 고정점은 2가 됩니다.

하나의 수열이 N개의 서로 다른 원소를 포함하고 있으며, 모든 원소가 오름차순으로 정렬되어 있습니다. 이때 이 수열에서 고정점이 있다면, 고정점을 출력하는 프로그램을 작성하세요. 고정점은 최대 1개만 존재합니다. 만약 고정점이 없다면, - 1을 출력합니다.
단, 이 문제는 시간 복잡도 O(logN)으로 알고리즘을 설계하지 않으면 '시간 초과' 판정을 받는다.

 

<입력 조건>

• 첫째 줄에 N이 입력됩니다. (1 <= N <= 1,000,000)
• 둘째 줄에 N개의 원소가 정수 형태로 공백으로 구분되어 입력됩니다. (-10e9 <= 각 원소의 값 <= 10e9)

<출력 조건>

• 고정점을 출력한다, 고정점이 없다면 -1을 출력합니다.

---

<문제 풀이>

1. 탐색해야 할 값의 개수가 백만개이고, 시간 복잡도 O(logN)으로 알고리즘을 설계해야 하므로 이분 탐색 활용

<코드>

'''입력 예시
5
-15 -6 1 3 7

7
-15 -4 2 8 9 13 15

7
-15 -4 3 8 9 13 15
'''

import sys
n = int(input())
array = list(map(int,sys.stdin.readline().rstrip().split()))

def solution(start, end, array):
    while(start<=end):
        mid = (start + end) // 2
        if mid==array[mid]:
            return mid
        elif mid>array[mid]:
            start = mid+1
        elif mid<array[mid]:
            end = mid-1
    return -1

print(solution(0,n-1,array)) #또는
print(solution(0,n,array))

#220323
from bisect import bisect_left

n = int(input())
nlist = list(map(int,input().split()))
answer = -1

for i in range(n):
    if nlist[i]==bisect_left(nlist,nlist[i]):
        answer = nlist[i]
        
if answer==-1:
    print(-1)
else:
    print(answer)