1129. Shortest Path with Alternating Col

LeetCode 갤러리
마이너

자동 짤방 이미지

이미지가 없습니다.

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

Eat, Sleep, LeetCode, Repeat

매니저

부재중입니다.(ddakgun)

부매니저

흑화뉴비(coderhs)

개설일

2022-12-26

개발뉴비

2023.02.11 23:43:57

조회 52 추천 0 댓글 1

class Solution:
    def shortestAlternatingPaths(self, n: int, redEdges: List[List[int]], blueEdges: List[List[int]]) -> List[int]:
        rg = [[] for _ in range(n)]
        bg = [[] for _ in range(n)]
        d = [[-1, -1] for _ in range(n)]
        for s, e in redEdges: rg[s].append(e)
        for s, e in blueEdges : bg[s].append(e)
        queue = deque([(0, 'r'), (0, 'b')])
        d[0] = [0, 0]
        while queue:
            cur, prev = queue.popleft()
            if prev == 'r':
                nxt = d[cur][0]
                for next in bg[cur]:
                    if d[next][1] == -1:
                        d[next][1] = nxt + 1
                        queue.append((next, 'b'))
            if prev == 'b':
                nxt = d[cur][1]
                for next in rg[cur]:
                    if d[next][0] == -1:
                        d[next][0] = nxt + 1
                        queue.append((next, 'r'))
        ans = [-1 for _ in range(n)]
        for i in range(n):
            if -1 in d[i]:
                ans[i] = sum(d[i]) + 1
            else :
                ans[i] = min(d[i])
        return ans

평범한 BFS 문제인데 생각보다 귀찮았당; 이틀 연속 BFS네

고정닉 0

전체 댓글 0개

등록순 최신순 답글순

본문 보기

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

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