[백준/파이썬] 5567번 결혼식 풀이

---

• 백준 문제풀이 전체 목록 보기

---

문제 정보

• 문제 출처: 백준 온라인 저지
• 문제 링크: 5567번 결혼식
• 문제풀이 코드 GitHub 링크
• 제출 언어: Python 3

풀이

문제

상근이의 친구와 친구의 친구(2촌)까지 초대했을 때 총 초대 인원을 구하는 문제입니다.

코드

n = int(input())
m = int(input())

l = list()

for _ in range(m):
    l.append(tuple(map(int, input().split())))

invited = set()
friends = set()

for a, b in l:
    if(a == 1):
        friends.add(b)
        invited.add(b)
    elif(b == 1):
        friends.add(a)
        invited.add(a)
    
for a, b in l:
    if(a in friends and b != 1):
        invited.add(b)
    elif(b in friends and a != 1):
        invited.add(a)
        
print(len(invited))

설명

1번과 직접 연결된 친구를 먼저 구하고, 그 친구들과 연결된 사람을 추가해 집합 크기를 출력합니다.

---

• 백준 문제풀이 전체 목록 보기

---