{"componentChunkName":"component---src-templates-blog-post-js","path":"/Algorithm/BOJ-1504-Python/","result":{"data":{"site":{"siteMetadata":{"title":"hyesungoh","author":"hyesungoh","siteUrl":"https://www.hyesungoh.xyz","comment":{"disqusShortName":"","utterances":"hyesungoh/blog"},"sponsor":{"buyMeACoffeeId":""}}},"markdownRemark":{"id":"dfe96fd4-3103-5afe-a311-8c75ae68f250","excerpt":"정점의 개수 n, 간선의 개수 e를 입력받은 후 e의 개수만큼 “출발지 도착지 가중치”를 입력받는다. 그 후 정점의 두개 v1, v2를 입력받은 후 정점 1부터 n까지 v1, v2를 반드시 통과한 최단거리를 출력하는 문제. 첫 접근은 deque의 appendleft를 이용하여 다음 방문지가 v1 혹은 v2일 때 우선순위를 높이는 방식이였으나 실패하게 되었다. 접근을 바꾸어 시작, v1, v2에서 출발하는 최단거리 리스트를 저장하고 s > v1 > v2 > n과 s > v2 > v1 > n…","html":"<p>정점의 개수 n, 간선의 개수 e를 입력받은 후 e의 개수만큼 “출발지 도착지 가중치”를 입력받는다. 그 후 정점의 두개 v1, v2를 입력받은 후 정점 1부터 n까지 v1, v2를 반드시 통과한 최단거리를 출력하는 문제. 첫 접근은 deque의 appendleft를 이용하여 다음 방문지가 v1 혹은 v2일 때 우선순위를 높이는 방식이였으나 실패하게 되었다. 접근을 바꾸어 시작, v1, v2에서 출발하는 최단거리 리스트를 저장하고 s > v1 > v2 > n과 s > v2 > v1 > n을 계산 및 비교하여 출력하여 풀었다. 간선을 저장할 때 유무를 찾아 추가하는 방식에서 런타임 에러가 일어나, 수정하여 풀었다.</p>\n<div class=\"gatsby-highlight\" data-language=\"python\"><pre class=\"language-python\"><code class=\"language-python\"><span class=\"token comment\"># import heapq</span>\n<span class=\"token comment\"># import sys</span>\n<span class=\"token comment\"># input = sys.stdin.readline</span>\n<span class=\"token comment\"># INF = sys.maxsize</span>\n<span class=\"token comment\">#</span>\n<span class=\"token comment\"># def dijkstra(start):</span>\n<span class=\"token comment\">#     dist = [INF for _ in range(n+1)]</span>\n<span class=\"token comment\">#     dist[start] = 0</span>\n<span class=\"token comment\">#     q = []</span>\n<span class=\"token comment\">#     heapq.heappush(q, [0, start])</span>\n<span class=\"token comment\">#     while q:</span>\n<span class=\"token comment\">#         weight, here = heapq.heappop(q)</span>\n<span class=\"token comment\">#</span>\n<span class=\"token comment\">#         for there, w in graph[here]:</span>\n<span class=\"token comment\">#             next_weight = weight + w</span>\n<span class=\"token comment\">#             if dist[there] > next_weight:</span>\n<span class=\"token comment\">#                 dist[there] = next_weight</span>\n<span class=\"token comment\">#                 heapq.heappush(q, [next_weight, there])</span>\n<span class=\"token comment\">#     return dist</span>\n<span class=\"token comment\">#</span>\n<span class=\"token comment\">#</span>\n<span class=\"token comment\"># n, e = map(int, input().split())</span>\n<span class=\"token comment\"># graph = {}</span>\n<span class=\"token comment\">#</span>\n<span class=\"token comment\"># for _ in range(e):</span>\n<span class=\"token comment\">#     u, v, w = map(int, input().split())</span>\n<span class=\"token comment\">#     if u not in graph:</span>\n<span class=\"token comment\">#         graph[u] = [[v, w]]</span>\n<span class=\"token comment\">#     else:</span>\n<span class=\"token comment\">#         graph[u].append([v, w])</span>\n<span class=\"token comment\">#     if v not in graph:</span>\n<span class=\"token comment\">#         graph[v] = [[u, w]]</span>\n<span class=\"token comment\">#     else:</span>\n<span class=\"token comment\">#         graph[v].append([u, w])</span>\n<span class=\"token comment\">#</span>\n<span class=\"token comment\"># v1, v2 = map(int, input().split())</span>\n<span class=\"token comment\"># s = dijkstra(1)</span>\n<span class=\"token comment\"># s_v1 = dijkstra(v1)</span>\n<span class=\"token comment\"># s_v2 = dijkstra(v2)</span>\n<span class=\"token comment\">#</span>\n<span class=\"token comment\"># ans = min(s[v1] + s_v1[v2] + s_v2[n],</span>\n<span class=\"token comment\">#           s[v2] + s_v2[v1] + s_v1[n])</span>\n<span class=\"token comment\"># print(ans if ans &lt; INF else -1)</span>\n\n<span class=\"token keyword\">import</span> heapq\n<span class=\"token keyword\">import</span> sys\n<span class=\"token builtin\">input</span> <span class=\"token operator\">=</span> sys<span class=\"token punctuation\">.</span>stdin<span class=\"token punctuation\">.</span>readline\nINF <span class=\"token operator\">=</span> sys<span class=\"token punctuation\">.</span>maxsize\n\n<span class=\"token keyword\">def</span> <span class=\"token function\">dijkstra</span><span class=\"token punctuation\">(</span>start<span class=\"token punctuation\">)</span><span class=\"token punctuation\">:</span>\n    dist <span class=\"token operator\">=</span> <span class=\"token punctuation\">[</span>INF <span class=\"token keyword\">for</span> _ <span class=\"token keyword\">in</span> <span class=\"token builtin\">range</span><span class=\"token punctuation\">(</span>n<span class=\"token operator\">+</span><span class=\"token number\">1</span><span class=\"token punctuation\">)</span><span class=\"token punctuation\">]</span>\n    dist<span class=\"token punctuation\">[</span>start<span class=\"token punctuation\">]</span> <span class=\"token operator\">=</span> <span class=\"token number\">0</span>\n    q <span class=\"token operator\">=</span> <span class=\"token punctuation\">[</span><span class=\"token punctuation\">]</span>\n    heapq<span class=\"token punctuation\">.</span>heappush<span class=\"token punctuation\">(</span>q<span class=\"token punctuation\">,</span> <span class=\"token punctuation\">[</span><span class=\"token number\">0</span><span class=\"token punctuation\">,</span> start<span class=\"token punctuation\">]</span><span class=\"token punctuation\">)</span>\n    <span class=\"token keyword\">while</span> q<span class=\"token punctuation\">:</span>\n        weight<span class=\"token punctuation\">,</span> here <span class=\"token operator\">=</span> heapq<span class=\"token punctuation\">.</span>heappop<span class=\"token punctuation\">(</span>q<span class=\"token punctuation\">)</span>\n\n        <span class=\"token keyword\">for</span> there<span class=\"token punctuation\">,</span> w <span class=\"token keyword\">in</span> graph<span class=\"token punctuation\">[</span>here<span class=\"token punctuation\">]</span><span class=\"token punctuation\">:</span>\n            next_weight <span class=\"token operator\">=</span> weight <span class=\"token operator\">+</span> w\n            <span class=\"token keyword\">if</span> dist<span class=\"token punctuation\">[</span>there<span class=\"token punctuation\">]</span> <span class=\"token operator\">></span> next_weight<span class=\"token punctuation\">:</span>\n                dist<span class=\"token punctuation\">[</span>there<span class=\"token punctuation\">]</span> <span class=\"token operator\">=</span> next_weight\n                heapq<span class=\"token punctuation\">.</span>heappush<span class=\"token punctuation\">(</span>q<span class=\"token punctuation\">,</span> <span class=\"token punctuation\">[</span>next_weight<span class=\"token punctuation\">,</span> there<span class=\"token punctuation\">]</span><span class=\"token punctuation\">)</span>\n    <span class=\"token keyword\">return</span> dist\n\n\nn<span class=\"token punctuation\">,</span> e <span class=\"token operator\">=</span> <span class=\"token builtin\">map</span><span class=\"token punctuation\">(</span><span class=\"token builtin\">int</span><span class=\"token punctuation\">,</span> <span class=\"token builtin\">input</span><span class=\"token punctuation\">(</span><span class=\"token punctuation\">)</span><span class=\"token punctuation\">.</span>split<span class=\"token punctuation\">(</span><span class=\"token punctuation\">)</span><span class=\"token punctuation\">)</span>\ngraph <span class=\"token operator\">=</span> <span class=\"token punctuation\">[</span><span class=\"token punctuation\">[</span><span class=\"token punctuation\">]</span> <span class=\"token keyword\">for</span> _ <span class=\"token keyword\">in</span> <span class=\"token builtin\">range</span><span class=\"token punctuation\">(</span>n<span class=\"token operator\">+</span><span class=\"token number\">1</span><span class=\"token punctuation\">)</span><span class=\"token punctuation\">]</span>\n\n<span class=\"token keyword\">for</span> _ <span class=\"token keyword\">in</span> <span class=\"token builtin\">range</span><span class=\"token punctuation\">(</span>e<span class=\"token punctuation\">)</span><span class=\"token punctuation\">:</span>\n    u<span class=\"token punctuation\">,</span> v<span class=\"token punctuation\">,</span> w <span class=\"token operator\">=</span> <span class=\"token builtin\">map</span><span class=\"token punctuation\">(</span><span class=\"token builtin\">int</span><span class=\"token punctuation\">,</span> <span class=\"token builtin\">input</span><span class=\"token punctuation\">(</span><span class=\"token punctuation\">)</span><span class=\"token punctuation\">.</span>split<span class=\"token punctuation\">(</span><span class=\"token punctuation\">)</span><span class=\"token punctuation\">)</span>\n    graph<span class=\"token punctuation\">[</span>u<span class=\"token punctuation\">]</span><span class=\"token punctuation\">.</span>append<span class=\"token punctuation\">(</span><span class=\"token punctuation\">[</span>v<span class=\"token punctuation\">,</span> w<span class=\"token punctuation\">]</span><span class=\"token punctuation\">)</span>\n    graph<span class=\"token punctuation\">[</span>v<span class=\"token punctuation\">]</span><span class=\"token punctuation\">.</span>append<span class=\"token punctuation\">(</span><span class=\"token punctuation\">[</span>u<span class=\"token punctuation\">,</span> w<span class=\"token punctuation\">]</span><span class=\"token punctuation\">)</span>\n\nv1<span class=\"token punctuation\">,</span> v2 <span class=\"token operator\">=</span> <span class=\"token builtin\">map</span><span class=\"token punctuation\">(</span><span class=\"token builtin\">int</span><span class=\"token punctuation\">,</span> <span class=\"token builtin\">input</span><span class=\"token punctuation\">(</span><span class=\"token punctuation\">)</span><span class=\"token punctuation\">.</span>split<span class=\"token punctuation\">(</span><span class=\"token punctuation\">)</span><span class=\"token punctuation\">)</span>\ns <span class=\"token operator\">=</span> dijkstra<span class=\"token punctuation\">(</span><span class=\"token number\">1</span><span class=\"token punctuation\">)</span>\ns_v1 <span class=\"token operator\">=</span> dijkstra<span class=\"token punctuation\">(</span>v1<span class=\"token punctuation\">)</span>\ns_v2 <span class=\"token operator\">=</span> dijkstra<span class=\"token punctuation\">(</span>v2<span class=\"token punctuation\">)</span>\n\n <span class=\"token comment\"># s > v1 > v2 > n과 s > v2 > v1 > n 을 비교</span>\nans <span class=\"token operator\">=</span> <span class=\"token builtin\">min</span><span class=\"token punctuation\">(</span>s<span class=\"token punctuation\">[</span>v1<span class=\"token punctuation\">]</span> <span class=\"token operator\">+</span> s_v1<span class=\"token punctuation\">[</span>v2<span class=\"token punctuation\">]</span> <span class=\"token operator\">+</span> s_v2<span class=\"token punctuation\">[</span>n<span class=\"token punctuation\">]</span><span class=\"token punctuation\">,</span>\n          s<span class=\"token punctuation\">[</span>v2<span class=\"token punctuation\">]</span> <span class=\"token operator\">+</span> s_v2<span class=\"token punctuation\">[</span>v1<span class=\"token punctuation\">]</span> <span class=\"token operator\">+</span> s_v1<span class=\"token punctuation\">[</span>n<span class=\"token punctuation\">]</span><span class=\"token punctuation\">)</span>\n<span class=\"token keyword\">print</span><span class=\"token punctuation\">(</span>ans <span class=\"token keyword\">if</span> ans <span class=\"token operator\">&lt;</span> INF <span class=\"token keyword\">else</span> <span class=\"token operator\">-</span><span class=\"token number\">1</span><span class=\"token punctuation\">)</span></code></pre></div>","frontmatter":{"title":"BOJ-1504 - Python","date":"November 05, 2020"}}},"pageContext":{"slug":"/Algorithm/BOJ-1504-Python/","previous":{"fields":{"slug":"/Algorithm/BOJ-16953-Python/"},"frontmatter":{"title":"BOJ-16953 - Python","category":"Algorithm","draft":false}},"next":{"fields":{"slug":"/Algorithm/BOJ-15663-Python/"},"frontmatter":{"title":"BOJ-15663 - Python","category":"Algorithm","draft":false}}}},"staticQueryHashes":["3128451518","3290666882"]}