787. Cheapest Flights Within K Stops

LeetCode 갤러리
마이너

자동 짤방 이미지

이미지가 없습니다.

자동 짤방으로 사용할 이미지를 등록해 주세요.

Eat, Sleep, LeetCode, Repeat

매니저

부재중입니다.(ddakgun)

부매니저

흑화뉴비(coderhs)

개설일

2022-12-26

개발뉴비

2023.01.26 23:06:06

조회 33 추천 0 댓글 0

class Solution:
    def findCheapestPrice(self, n: int, flights: List[List[int]], src: int, dst: int, k: int) -> int:
        edges = [[] for _ in range(n)]
        dist = [[10**7 for _ in range(k+2)] for _ in range(n)]
        for a, b, c in flights:
            edges[a].append([c,b])
        pq = []
        dist[src][-1] = 0
        heappush(pq, [0, src, 0])
        while pq:
            dis, cur, prev = heappop(pq)
            if dist[cur][prev-1] != dis:
                continue
            for price, next in edges[cur]:
                if prev > k or dist[next][prev] <= dist[cur][prev-1] + price:
                    continue
                dist[next][prev] = dist[cur][prev-1] + price
                heappush(pq, [dist[next][prev], next, prev+1])
        return min(dist[dst]) if min(dist[dst]) != 10**7 else -1

n이 작아서 대충 눈치를 챌 수 있었지만 생각보다 어려워서 고생했다..

다익스트라를 사용했고, 모든 k에 대해서 해당하는 거리를 다 할당했음. (시작점은 -1)

결국은 k 이하 조건을 어떻게 관리하냐가 핵심인데, 그냥 다 넣어줘도 될만한 크기..

while 두 칸 밑 코드가 있냐 없느냐가 실행속도에 꽤 영향을 미친다 (200ms vs 500ms)

우선순위 큐에서 뽑히기 전에 이미 최소거리가 갱신되어있는 경우에는 거른다.

고정닉 0

전체 댓글 0개

등록순 최신순 답글순

본문 보기

타인의 권리를 침해하거나 명예를 훼손하는 댓글은 운영원칙 및 관련 법률에 제재를 받을 수 있습니다.
Shift+Enter 키를 동시에 누르면 줄바꿈이 됩니다.

갤러리 리스트
번호	제목	글쓴이	작성일	조회	추천
설문	2025년 가장 기억에 남는 인터넷 이슈는?	운영자	25/12/22	-	-
333	3355. Zero Array Transformation I	흑화뉴비	05.20	50	0
332	2918. Minimum Equal Sum of Two Arrays Af	흑화뉴비	05.10	21	0
331	2033. Minimum Operations to Make a Uni-V	흑화뉴비	03.28	45	0
330	2401. Longest Nice Subarray	흑화뉴비	03.19	38	0
329	1079. Letter Tile Possibilities	흑화뉴비	02.18	44	0
328	1718. Construct the Lexicographically La	흑화뉴비	02.16	26	0
327	1910. Remove All Occurrences of a Substr	흑화뉴비	02.11	28	0
326	2381. Shifting Letters II	흑화뉴비	01.05	42	0
323	2471. Minimum Number of Operations to So	흑화뉴비	24.12.25	30	0
322	773. Sliding Puzzle	흑화뉴비	24.11.27	30	0
321	3243. Shortest Distance After Road Addit	흑화뉴비	24.11.27	23	0
320	1233. Remove Sub-Folders from the Filesy	흑화뉴비	24.10.29	29	0
319	2684. Maximum Number of Moves in a Grid	흑화뉴비	24.10.29	31	0
318	2501. Longest Square Streak in an Array	흑화뉴비	24.10.29	42	0
317	3043. Find the Length of the Longest Com	흑화뉴비	24.09.24	38	0
316	1514. Path with Maximum Probability	흑화뉴비	24.08.27	37	0
315	264. Ugly Number II	흑화뉴비	24.08.19	35	0
314	2285. Maximum Total Importance of Roads	흑화뉴비	24.06.28	47	0
313	1552. Magnetic Force Between Two Balls	흑화뉴비	24.06.20	37	0
312	1482. Minimum Number of Days to Make m B	흑화뉴비	24.06.19	40	0
311	523. Continuous Subarray Sum	흑화뉴비	24.06.09	41	0
310	리트코드 갤러리 5월 이벤트 우승자 공지 [1]	진척	24.06.08	64	0
309	1208. Get Equal Substrings Within Budget	흑화뉴비	24.05.28	38	0
308	1608. Special Array With X Elements Grea	흑화뉴비	24.05.28	30	0
307	552. Student Attendance Record II	흑화뉴비	24.05.28	31	0
306	2597. The Number of Beautiful Subsets [1]	흑화뉴비	24.05.24	55	0
305	78. Subsets	흑화뉴비	24.05.21	32	0
304	1863. Sum of All Subset XOR Totals	흑화뉴비	24.05.21	36	0
302	1325. Delete Leaves With a Given Value	흑화뉴비	24.05.18	35	0
301	2331. Evaluate Boolean Binary Tree	흑화뉴비	24.05.18	27	0
298	3075. Maximize Happiness of Selected Chi	흑화뉴비	24.05.12	25	0
297	786. K-th Smallest Prime Fraction	흑화뉴비	24.05.12	29	0
296	506. Relative Ranks	흑화뉴비	24.05.08	37	0
295	2816. Double a Number Represented as a L	흑화뉴비	24.05.08	24	0
294	2487. Remove Nodes From Linked List	흑화뉴비	24.05.06	25	0
293	79. Word Search	o오어o(14.37)	24.05.04	28	0
292	45. Jump Game II	o오어o(14.37)	24.05.04	35	0
291	881. Boats to Save People	o오어o(14.37)	24.05.04	29	0
290	2441. Largest Positive Integer That Exis	흑화뉴비	24.05.02	27	0
289	2997. Minimum Number of Operations to Ma	흑화뉴비	24.05.01	24	0
288	2000. Reverse Prefix of Word	흑화뉴비	24.05.01	34	0
287	리트코드 갤러리 5월 이벤트 공지 [3]	진척	24.04.28	96	0
282	백준 1260. DFS와 BFS	차근차근춘식이(211.234)	24.04.23	46	0
277	206. Reverse Linked List	흑화뉴비	24.03.21	47	0
276	1669. Merge In Between Linked Lists	흑화뉴비	24.03.20	30	0
275	57. Insert Interval	흑화뉴비	24.03.18	30	0
274	2485. Find the Pivot Integer	흑화뉴비	24.03.13	24	0
273	1750. Minimum Length of String After Del	흑화뉴비	24.03.06	26	0
272	451. Sort Characters By Frequency	흑화뉴비	24.02.07	40	0
271	49. Group Anagrams	흑화뉴비	24.02.06	34	0