백준 2606 바이러스

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

2606번: 바이러스

n = int(input())  # 컴퓨터 수
pair = int(input())  # 컴퓨터 쌍의 수
p = []
for _ in range(pair):                   # 정렬
    a, b = input().split()
    if a < b:
        p.append([int(a), int(b)])
    else:
        p.append([int(b), int(a)])
p.sort()

if 1 != p[0][0]:
    print(0)

else:
    virus = [1]*n
    result = [1]
    cnt = 0
    while 1:
        if result == []:
            break
        x = result[0]    # 쌍에 x 가 존재하는 지 비교
        virus[x-1] = 0   # x 번 컴퓨터는 바이러스 전염됨

        for i in range(len(p)):
            if p[i] != 0 :        # 비교하지 않은 쌍인 경우
                if p[i][0] == x:
                    result.append(p[i][1])
                    p[i] = 0

                elif p[i][1] == x:
                    result.append(p[i][0])
                    p[i] = 0
      
        del result[0]
    print(virus.count(0)-1)    # 1번 빼고 전염된 개수 출력