[백준] 30502번: 미역은 식물 아닌데요 - 파이썬

• 백준 링크: 30502번: 미역은 식물 아닌데요
• solved.ac 난이도: 브론즈 I
• 시간 제한: 1 초
• 메모리 제한: 1024 MB

문제

세종이는 미역을 식물로 분류해야 한다고 생각하고 있다. 식물은 광합성을 하고 운동성이 없으며, 광합성을 하고 운동성이 없으면 식물이므로 미역 역시 식물의 일종으로 분류되어야 한다는 것이 세종이의 주장이다.

입력

첫째 줄에 생물의 종류 수와 실험의 수를 나타내는 두 정수 $N$과 $M$이 공백으로 구분되어 주어진다. $(1\leq N\leq 1, 000;$ $1\leq M\leq 2\times N)$

출력

첫째 줄에 식물의 수로 가능한 것의 최솟값과 최댓값을 공백으로 구분해 출력한다.

코드

def main():
    import sys
    input = sys.stdin.read
    data = input().splitlines()

    N, M = map(int, data[0].split())
    arr = [[-1, -1] for _ in range(N + 1)]

    for i in range(1, M + 1):
        a, b, c = data[i].split()
        a = int(a)
        c = int(c)

        if b == 'P':
            arr[a][0] = c
        else:
            arr[a][1] = c

    must = 0
    able = 0
    for i in range(1, N + 1):
        if arr[i][0] == 1 and arr[i][1] == 0:
            must += 1
        if arr[i][0] != 0 and arr[i][1] != 1:
            able += 1

    print(must, able)

if __name__ == "__main__":
    main()