백준 #5567 - 결혼식

 

 

문제

상근이는 자신의 결혼식에 학교 동기 중 자신의 친구와 친구의 친구를 초대하기로 했다. 상근이의 동기는 모두 N명이고, 이 학생들의 학번은 모두 1부터 N까지이다. 상근이의 학번은 1이다.

상근이는 동기들의 친구 관계를 모두 조사한 리스트를 가지고 있다. 이 리스트를 바탕으로 결혼식에 초대할 사람의 수를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 상근이의 동기의 수 n (2 ≤ n ≤ 500)이 주어진다. 둘째 줄에는 리스트의 길이 m (1 ≤ m ≤ 10000)이 주어진다. 다음 줄부터 m개 줄에는 친구 관계 ai bi가 주어진다. (1 ≤ ai < bi ≤ n) ai와 bi가 친구라는 뜻이며, bi와 ai도 친구관계이다. 

출력

첫째 줄에 상근이의 결혼식에 초대하는 동기의 수를 출력한다.

예제 입력 1

6
5
1 2
1 3
3 4
2 3
4 5

예제 출력 1

3

예제 입력 2

6
5
2 3
3 4
4 5
5 6
2 5

예제 출력 2

0

힌트

예제 1의 경우 2와 3은 상근이의 친구이다. 또, 3과 4는 친구이기 때문에, 4는 상근이의 친구의 친구이다. 5와 6은 친구도 아니고, 친구의 친구도 아니다. 따라서 2, 3, 4 3명의 친구를 결혼식에 초대한다.

 

 

---

 

 

info 리스트는 1의 친구를 저장하는 리스트이다.

info[3]이 1이라면 3번은 1의 친구임을 나타낸다.

 

friends 리스트에는 1과 직접적인 친구가 아닌 관계 정보를 저장한다.

 

if friend[0] == 1 or friend[1] == 1:
        if friend[0] == 1:
            info[friend[1]] = 1
        else:
            info[friend[0]] = 1
 else:
        friends.append(friend)

 

 

위 코드를 통해 info 리스트에 1의 친구들을 저장한다.

위 if문에 걸리지 않은 관계들은 1과 직접적인 친구가 아닌 관계이므로  friends 리스트에 추가한다.

 

friends2 리스트에는 friends 리스트의 관계들 중, 1과 연관이 있는 관계만 추출해 담는 리스트이다.

 

for i in friends:
    if info[i[0]] == 1 or info[i[1]] == 1:
        friends2.append(i)

 

 

이 코드를 통해 friends2 리스트에 1과 연관이 있는 관계를 담는다.

 

그 후 마지막으로, friends2의 관계들을info 리스트에 저장한 후, info 리스트의 1의 개수를 출력하면 된다.

 

그렇게 나온 최종 코드는 다음과 같다.

 

import sys
input = sys.stdin.readline

n = int(input()) # 동기 수
m = int(input()) # 리스트 길이

info = [0] * (n + 1)
friends = []

for i in range(m):
    friend = list(map(int, input().split()))
    if friend[0] == 1 or friend[1] == 1:
        if friend[0] == 1:
            info[friend[1]] = 1
        else:
            info[friend[0]] = 1
    else:
        friends.append(friend)

friends2 = []
for i in friends:
    if info[i[0]] == 1 or info[i[1]] == 1:
        friends2.append(i)

for i in friends2:
    info[i[0]] = 1 
    info[i[1]] = 1

print(info.count(1))