Notice
Recent Posts
Recent Comments
Link
일 | 월 | 화 | 수 | 목 | 금 | 토 |
---|---|---|---|---|---|---|
1 | 2 | 3 | 4 | 5 | ||
6 | 7 | 8 | 9 | 10 | 11 | 12 |
13 | 14 | 15 | 16 | 17 | 18 | 19 |
20 | 21 | 22 | 23 | 24 | 25 | 26 |
27 | 28 | 29 | 30 | 31 |
Tags
- 슬라이딩 윈도우
- SSAFY
- 브루트포스
- 에라토스테네스의 체
- 애드 혹
- 2357
- 13164
- 정렬
- 이분 탐색
- 문자열
- 모던 JavaScript 튜토리얼
- 투 포인터
- 정수론
- 누적 합
- BFS
- boj
- 해시 테이블
- 싸피
- 트리
- Python
- DFS
- 세그먼트 트리
- 플로이드-워셜
- 수학
- DP
- 구현
- 그래프
- 그리디
- JavaScript
- 맵
Archives
- Today
- Total
흙금이네 블로그
[BOJ] 2176 - 합리적인 이동경로 (Python) 본문
아이디어
다익스트라로 2번 정점까지의 최단 거리를 구한 후, 트리에서의 동적 계획법으로 경로의 수를 구한다.
풀이
import sys
from heapq import heappush, heappop
sys.setrecursionlimit(int(1e5))
input = sys.stdin.readline
INF = int(1e7)
def solution():
def dfs(u):
if dp[u]:
return dp[u]
if u == 2:
return 1
for v, c in graph[u]:
if D[v] < D[u]:
dp[u] += dfs(v)
return dp[u]
N, M = map(int, input().split())
graph = [[] for _ in range(N+1)]
for _ in range(M):
A, B, C = map(int, input().split())
graph[A].append((B, C))
graph[B].append((A, C))
D = [INF]*(N+1)
D[2] = 0
heap = [(0, 2)]
while heap:
c, u = heappop(heap)
for v, _c in graph[u]:
if c+_c < D[v]:
D[v] = c+_c
heappush(heap, (c+_c, v))
dp = [0]*(N+1)
dfs(1)
print(dp[1])
solution()
'알고리즘' 카테고리의 다른 글
[BOJ] 15553 - 난로 (Python) (0) | 2023.04.15 |
---|---|
[BOJ] 24526 - 전화 돌리기 (Python) (0) | 2023.04.15 |
[BOJ] 2887 - 행성 터널 (Python) (0) | 2023.04.15 |
[BOJ] 1368 - 물대기 (Python) (0) | 2023.04.15 |
[BOJ] 10423 - 전기가 부족해 (Python) (0) | 2023.04.15 |
Comments