[백준] 2841번 외계인의 기타연주 파이썬

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

2841번: 외계인의 기타 연주

[코드]

import sys
N, P = map(int, sys.stdin.readline().split())

lines = [[] for _ in range(6)]
answer = 0
for i in range(N):
    l, p = map(int, sys.stdin.readline().split())

    while lines[l-1]:
        if lines[l-1][-1] > p:
            lines[l-1].pop()
            answer += 1

        elif lines[l-1][-1] < p:
            lines[l-1].append(p)
            answer += 1
        else:
            break
    if not lines[l-1]:
        lines[l-1].append(p)
        answer += 1

print(answer)