CS/알고리즘 문제 풀이

[백준] 5567 - 결혼 [Python(파이썬)]

2021. 6. 23. 23:40
목차
  1. 문제
  2. 풀이
  3. 코드

문제

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

 

5567번: 결혼식

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

www.acmicpc.net

 

풀이

분류는 구현으로 되어 있었는데, 그래프 탐색으로도 풀 수 있는 문제였다. 문제의 힌트가 중요한데, 친구의 친구까지만 결혼식에 초대가 가능하다는 점이다. 따라서 1번 노드(상근이)를 시작으로 너비 우선 탐색을 진행하면서 거리(dist)가 2 이하인 노드들만 개수를 세주면 된다.

 

코드

import sys
from collections import defaultdict, deque

def bfs(start):
    cnt = 0
    visited = [0 for _ in range(n+1)]
    visited[start] = 1
    queue = deque([[start, 0]])
    while queue:
        u, dist = queue.popleft()
        if dist <= 2: # 친구의 친구(dist==2)까지 초대할 수 있다
            cnt += 1
        for v in g[u]:
            if not visited[v]:
                visited[v] = 1
                queue.append([v, dist+1])
    return cnt-1 # cnt에 자기 자신까지 포함되어 있으므로 1을 빼준다

n = int(sys.stdin.readline().strip())
m = int(sys.stdin.readline().strip())
g = defaultdict(list)
for _ in range(m):
    a, b = map(int, sys.stdin.readline().split())
    g[a].append(b)
    g[b].append(a)

print(bfs(1))
저작자표시 비영리 (새창열림)

'CS > 알고리즘 문제 풀이' 카테고리의 다른 글

[백준] 1946 - 신입 사원 [Python(파이썬)]  (0) 2021.06.28
[백준] 15686 - 치킨 배달 [Python(파이썬)]  (0) 2021.06.24
[백준] 1325 - 효율적인 해킹 [Python(파이썬)]  (4) 2021.06.22
[백준] 1759 - 암호 만들기 [Python(파이썬)]  (0) 2021.06.10
[백준] 1697 - 숨바꼭질 [Python(파이썬)]  (0) 2021.05.29
  1. 문제
  2. 풀이
  3. 코드
'CS/알고리즘 문제 풀이' 카테고리의 다른 글
  • [백준] 1946 - 신입 사원 [Python(파이썬)]
  • [백준] 15686 - 치킨 배달 [Python(파이썬)]
  • [백준] 1325 - 효율적인 해킹 [Python(파이썬)]
  • [백준] 1759 - 암호 만들기 [Python(파이썬)]
코택
코택
코택
TaxFree
코택
전체
오늘
어제
  • 분류 전체보기 (369)
    • Spring (29)
      • Spring (18)
      • 스프링 핵심 원리 - 고급편 (11)
    • Spring Batch (4)
    • JPA (4)
    • CS (89)
      • 자료구조 (2)
      • 네트워크 (5)
      • 운영체제 (1)
      • 데이터베이스 (4)
      • SQL (7)
      • 알고리즘 이론 (4)
      • 알고리즘 문제 풀이 (66)
    • 웹 (28)
      • React.js (4)
      • Next.js (1)
      • Node.js (14)
      • FastAPI (4)
      • Django (5)
    • 프로그래밍 언어 (45)
      • Python (5)
      • Java + Kotlin (29)
      • JavaScript + TypeScript (11)
    • 테스트코드 (26)
      • ATDD, 클린 코드 with Spring (4)
      • 이규원의 현실 세상의 TDD: 안정감을 주는 코드.. (20)
    • 인프라 (6)
      • AWS (2)
      • Kubernetes (4)
    • 트러블슈팅 (25)
    • 책 (89)
      • Effective Java (54)
      • Effective Kotlin (14)
      • 도메인 주도 개발 시작하기: DDD 핵심 개념 정.. (11)
      • 웹 프로그래머를 위한 데이터베이스를 지탱하는 기술 (6)
      • 도메인 주도 설계 첫걸음 (4)
    • Git (10)
    • 회고 (5)
    • etc (8)

블로그 메뉴

  • 홈
  • 방명록
  • 관리
  • GitHub
  • LinkedIn

공지사항

  • 스킨 관련

인기 글

태그

  • mysql
  • dp
  • 브루트포스
  • 그래프
  • http
  • 그래프 탐색
  • 백준
  • BOJ
  • atdd
  • Shortest Path
  • Git
  • fastapi
  • 장고
  • 파이썬
  • 깊이 우선 탐색

최근 댓글

최근 글

hELLO · Designed By 정상우.
코택
[백준] 5567 - 결혼 [Python(파이썬)]
상단으로

티스토리툴바

단축키

내 블로그

내 블로그 - 관리자 홈 전환
Q
Q
새 글 쓰기
W
W

블로그 게시글

글 수정 (권한 있는 경우)
E
E
댓글 영역으로 이동
C
C

모든 영역

이 페이지의 URL 복사
S
S
맨 위로 이동
T
T
티스토리 홈 이동
H
H
단축키 안내
Shift + /
⇧ + /

* 단축키는 한글/영문 대소문자로 이용 가능하며, 티스토리 기본 도메인에서만 동작합니다.