파이썬(Python) 백준 20551번: Sort 마스터 배지훈의 후계자

백준 20551번

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

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

 

정답코드

- alist: 정수 A를 받아 정렬한다.

- aidx: 정수 A의 위치를 빠르게 알기 위해, {A: 위치} 와 같은 형태의 딕셔너리를 만든다.

- 각 정수 D에 대해, alist의 key에 D가 있으면 대응하는 value를 출력하고, 아니면 -1을 출력한다.

import sys
input = sys.stdin.readline

n, m = map(int, input().split())

alist = sorted([int(input()) for _ in range(n)])

aidx = dict()
for i in range(n):
    if alist[i] not in aidx:
    	aidx[alist[i]] = i

for i in range(m):
    d = int(input())
    print(aidx[d] if d in aidx else -1)