[BOJ / Python] 13424 - 비밀 모임

2022. 2. 28. 07:40· Koala - 5기/코딩테스트 준비 스터디
목차
  1. 문제 링크
  2. 풀이
  3. 코드

문제 링크

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

 

13424번: 비밀 모임

입력 데이터는 표준 입력을 사용한다. 입력은 T개의 테스트 데이터로 구성된다. 입력의 첫 번째 줄에 테스트 케이스의 개수를 나타내는 자연수 T가 주어진다. 각 테스트 케이스의 첫째 줄에는 방

www.acmicpc.net


풀이

  • 다익스트라 알고리즘을 활용해 풀었습니다. (line 6~17)
  • 인접 행렬 adj 에 양방향 간선을 표시해줍니다. (line 26~27)
  • 반복문을 돌려 1부터 n 번 방까지 다익스트라를 구합니다. (line 30~32)
    • [해당 방(i) -> 각 친구들의 방 위치] 까지 이동할때의 최단 거리 합을 구합니다. (line 32)
    • ans[i]에는 i까지 친구들의 이동거리의 총합이 저장되어 있습니다.
  • 가장 작은 총합을 구해(min) 해당 방의 번호(index)를 출력합니다. (line 33)

코드

from heapq import *
import sys

input = sys.stdin.readline

def dijkstra(s):
    cost = [float('inf')] * (n+1)
    hq = [[0, s]]
    cost[s] = 0
    while hq:
        t,x = heappop(hq)
        if cost[x] != t: continue
        for nx, nt in adj[x]:
            if cost[nx] > t+nt:
                cost[nx] = t+nt
                heappush(hq, [cost[nx], nx])
    return cost

t = int(input())
for _ in range(t):
    n, m = map(int, input().split())
    ans = [float('inf')]*(n+1)
    adj = [[] for _ in range(n+1)]
    for _ in range(m):
        a, b, c = map(int, input().split())
        adj[a].append([b, c])
        adj[b].append([a, c])
    k = int(input())
    friend = [*map(int, input().split())]
    for i in range(1, n+1):
        dist = dijkstra(i)
        ans[i] = sum(dist[f] for f in friend)
    print(ans.index(min(ans)))
저작자표시 (새창열림)

'Koala - 5기 > 코딩테스트 준비 스터디' 카테고리의 다른 글

BOJ 1504(python) : 특정한 최단 경로  (0) 2022.02.28
[BOJ / Swift & Python] 23354 - 군탈체포조  (0) 2022.02.27
[BOJ / Python] 13911번: 집 구하기  (0) 2022.02.26
[BOJ/C++] 2615 - 오목  (0) 2022.02.25
[BOJ / JAVA] 1504 - 특정한 최단 경로  (0) 2022.02.21
  1. 문제 링크
  2. 풀이
  3. 코드
'Koala - 5기/코딩테스트 준비 스터디' 카테고리의 다른 글
  • BOJ 1504(python) : 특정한 최단 경로
  • [BOJ / Swift & Python] 23354 - 군탈체포조
  • [BOJ / Python] 13911번: 집 구하기
  • [BOJ/C++] 2615 - 오목
KauKoala
KauKoala
항공대 알고리즘 동아리 Koala 🥰
Koala항공대 알고리즘 동아리 Koala 🥰
KauKoala
Koala
KauKoala
전체
오늘
어제
  • 분류 전체보기 (1883)
    • 공지 게시판 (10)
    • 정보 게시판 (8)
    • Codeforce (15)
    • acm-icpc (6)
    • Koala - 1기 (16)
    • Koala - 2기 (111)
      • Programming Contest (1)
      • A반 (20)
      • B반 (39)
      • C반 (22)
      • 기초 강의 (18)
    • Koala - 3기 (10)
      • 기초 스터디 (7)
    • Koala - 4기 (67)
    • Koala - 5기 (144)
      • 기초 알고리즘 스터디 (75)
      • 코딩테스트 준비 스터디 (68)
    • Koala - 6기 (102)
      • 기초 알고리즘 스터디 (75)
      • 코딩테스트 준비 스터디 (25)
      • 모의 테스트 스터디 (1)
    • Koala - 7기 (167)
      • 기초 알고리즘 스터디 (97)
      • 코딩테스트 준비 스터디 (68)
      • 모의 테스트 스터디 (1)
    • Koala - 8기 (44)
      • 기초 알고리즘 스터디 (32)
      • 코딩테스트 준비 스터디 (10)
      • 코드포스 버츄얼 스터디 (0)
      • 프로그래머스 LV2 스터디 (0)
    • Koala - 9기 (205)
      • 기초 알고리즘 스터디 (138)
      • 코딩테스트 준비 스터디 (64)
      • 모의테스트 준비 스터디 (1)
    • Koala - 10기 (117)
      • 기초 알고리즘 스터디 (30)
      • 코딩테스트 준비 스터디 (86)
      • 모의테스트 준비 스터디 (1)
    • Koala - 11기 (151)
      • 기초 알고리즘 스터디 (46)
      • 코딩테스트 준비 스터디 (104)
      • 모의테스트 준비 스터디 (1)
    • Koala - 12기 (86)
      • 기초 알고리즘 스터디 (31)
      • 코딩테스트 준비 스터디 (55)
    • Koala - 13기 (119)
      • 기초 알고리즘 스터디 (52)
      • 코딩테스트 준비 스터디 (67)
    • Koala - 14기 (116)
      • 기초 알고리즘 스터디 (39)
      • 코딩테스트 준비 스터디 (77)
    • Koala - 15기 (138)
      • 기초 알고리즘 스터디 (73)
      • 코딩테스트 준비 스터디 (65)
    • Koala - 16기 (47)
      • 코딩테스트 기초 스터디 (16)
      • 코딩테스트 심화 스터디 (31)
    • Koala - 17기 (62)
      • 코딩테스트 기초 스터디 (15)
      • 코딩테스트 심화 스터디 (47)
    • Koala - 18기 (31)
      • 코딩테스트 기초 스터디 (11)
      • 코딩테스트 심화 스터디 (20)
    • Koala - 19기 (38)
      • 코딩테스트 기초 스터디 (7)
      • 코딩테스트 심화 스터디 (31)

블로그 메뉴

  • 홈
  • 태그
  • 방명록

공지사항

  • 🐨항공대 알고리즘 학회 Koala 3기 모집
  • 🐨항공대 알고리즘 학회 Koala 2기 모집
  • 소모임 소개

인기 글

태그

  • BFS
  • BOJ
  • 파이썬
  • C++
  • 백트래킹
  • dfs
  • dp
  • 백준

최근 댓글

최근 글

hELLO · Designed By 정상우.v4.3.0
KauKoala
[BOJ / Python] 13424 - 비밀 모임
상단으로

티스토리툴바

단축키

내 블로그

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

블로그 게시글

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

모든 영역

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

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