Coding Test

[백준] Passport Control

상상쓰 2021. 7. 8. 11:40

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

 

16288번: Passport Control

입력은 표준입력을 사용한다. 첫 번째 줄에는 두 개의 정수 N 과 k 가 주어진다. N은 입국 승객의 수이며 k는 여권 심사 창구의 수이다. 단, 2 ≤ k ≤ N ≤ 100 이다. 그리고 두 번째 줄에는 승객이 입

www.acmicpc.net

 

1 부터 N 까지 k 개의 q 에 차례대로 진입하므로 다 진입했을 때의 q 를 보면 각각 오름차순으로 정렬이 되어야한다. 즉, PI 를 가지고 k 개의 q 로 분할했을 때 오름차순으로 나타낼 수 없다면 'NO' 를 반환하고, 나타낼 수 있다면 'YES' 를 반환한다.

 

import sys

N, k = map(int, sys.stdin.readline().split())
pi = list(map(int, sys.stdin.readline().split()))
Q = [0] * k
count = 0

for i in PI:
    for j in range(k):
        if Q[j] < i:
            Q[j] = i
            count += 1
            break

answer = 'YES' if count == N else 'NO'

print(answer)