[백준] 20551 Sort마스터 배지훈의 후계자 python

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

20551번: Sort 마스터 배지훈의 후계자

이진탐색 문제이고 python 함수 bisect를 사용해서 풀었다.

# import sys; input = sys.stdin.readline
from bisect import bisect_left

def index(a, x):
    i = bisect_left(a, x)
    if i != len(a) and a[i] == x:
        return i
    return -1

def main():
    N, M = map(int, input().split())
    arr = sorted([int(input()) for _ in range(N)])
    answer = []
    for _ in range(M):
        q = int(input())
        answer.append(index(arr, q))
    print('\n'.join(map(str, answer)))


if __name__ == "__main__":
    main()

다른 문제를 풀려했는데 계속 틀려서...