787. Cheapest Flights Within K Stops

LeetCode 갤러리
마이너

자동 짤방 이미지

이미지가 없습니다.

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

Eat, Sleep, LeetCode, Repeat

매니저

부재중입니다.(ddakgun)

부매니저

흑화뉴비(coderhs)

개설일

2022-12-26

개발뉴비

2023.01.26 23:06:06

조회 32 추천 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	-	-
156	2306. Naming a company (설명X) [1]	ㅇㅇ(218.234)	23.02.09	32	0
155	45. Jump Game II	이벤트(1.226)	23.02.08	21	0
154	45. Jump Game II	chromate00	23.02.08	22	0
153	45. Jump Game II	개발뉴비	23.02.08	28	0
151	904. Fruit Into Baskets	개발뉴비	23.02.08	33	0
148	904. Fruit Into Baskets	이벤트(1.226)	23.02.07	30	0
147	904. Fruit Into Baskets	chromate00	23.02.07	23	0
146	1470. Shuffle the Array	이벤트(1.226)	23.02.07	21	0
145	1470. Shuffle the Array	개발뉴비	23.02.06	31	0
143	1470. Shuffle the Array	chromate00	23.02.06	23	0
142	438. Find All Anagrams in a String	개발뉴비	23.02.05	32	0
141	438. Find All Anagrams in a String	chromate00	23.02.05	18	0
139	438. Find All Anagrams in a String	이벤트(1.226)	23.02.05	23	0
138	567. Permutation in String	개발뉴비	23.02.05	59	0
137	567. Permutation in String	이벤트(1.226)	23.02.04	23	0
136	567. Permutation in String [1]	chromate00	23.02.04	47	0
134	6. Zigzag Conversion.	chromate00	23.02.03	24	0
133	6. Zigzag Conversion	이벤트(1.226)	23.02.03	20	0
131	6. Zigzag Conversion	개발뉴비	23.02.03	30	0
130	953. Verifying an Alien Dictionary [1]	이벤트(1.226)	23.02.02	36	0
128	953. Verifying an Alien Dictionary	개발뉴비	23.02.02	42	0
127	953. Verifying an Alien Dictionary	chromate00	23.02.02	25	0
126	1071. Greatest Common Divisor of Strings	개발뉴비	23.02.02	27	0
125	1071. Greatest Common Divisor of Strings	이벤트(1.226)	23.02.01	37	0
123	1071. Greatest Common Divisor of Strings	chromate00	23.02.01	29	0
122	LeetCode 오늘의 문제 redirect 서비스	ㅇㅇ(1.229)	23.02.01	39	1
121	1626. Best Team With No Conflicts	이벤트(1.226)	23.01.31	25	0
119	1626. Best Team With No Conflicts [2]	chromate00	23.01.31	52	0
118	1137. N-th Tribonacci Number	개발뉴비	23.01.30	36	0
117	1137. N-th Tribonacci Number	이벤트(1.226)	23.01.30	19	0
116	1137. N-th Tribonacci Number	chromate00	23.01.30	22	0
113	352. Data Stream as Disjoint Intervals	개발뉴비	23.01.29	46	0
112	구글러에게 들은 코딩인터뷰 팁	ㅇㅇ(203.226)	23.01.28	63	0
111	352. Data Stream as Disjoint Intervals [1]	이벤트(219.251)	23.01.28	39	0
110	352. Data Stream as Disjoint Intervals [1]	chromate00	23.01.28	50	0
108	472. Concatenated Words [1]	이벤트(219.251)	23.01.28	37	0
106	472. Concatenated Words [1]	개발뉴비	23.01.28	55	0
104	하드라니 오늘 풀 수 있을까	ㅇㅇ(118.235)	23.01.27	43	0
103	472. Concatenated Words	chromate00	23.01.27	43	0
	787. Cheapest Flights Within K Stops	개발뉴비	23.01.26	32	0
100	787. Cheapest Flights Within K Stops	이벤트(219.251)	23.01.26	34	0
97	787. Cheapest Flights Within K Stops	chromate00	23.01.26	41	0
96	2359. Find Closest Node to Given Two Nod	이벤트(219.251)	23.01.26	39	0
95	2359. Find Closest Node to Given Two ~ [1]	개발뉴비	23.01.26	36	0
92	이벤트 너무 어려운데 힌트 봐도 되나요 [1]	이벤트(39.7)	23.01.25	74	0
91	2359. Find Closest Node to Given Two Nod	chromate00	23.01.25	38	0
90	53. Maximum Subarray	크아아아앙	23.01.25	32	0
89	45. Jump Game II	크아아아앙	23.01.25	26	0
88	55. Jump Game	크아아아앙	23.01.25	40	0
87	909. Snake and Ladders	chromate00	23.01.24	70	0