본문 바로가기
👩‍💻Developer/Leetcode

[Easy][Python] Find Center of Star Graph

by 블루누들 2022. 10. 22.

n nodes를 가진 undirected star graph 가 존재한다. star graph는 n-1 개의 edge가 있고 가운데에 center node가 있는 형태이다. 

 

이는 코드로 2D integer array로 주어지며 각 edges[i]는 [u_i, v_i] 형태로 두 node사이에 edge가 있음을 나타낸다. 이 graph의 center를 return 한다. 

 

EX1)

edges = [[1,2],[2,3],[4,2]]
#output :2

 

class Solution:
    def findCenter(self, edges: List[List[int]]) -> int:
        res = list(reduce(lambda i, j: i&j, (set(n) for n in edges))) #reduce를 통해서 모든 list segments에 lambda function적용
        #lambda function은 각 nested list에 set을 적용한 후 common element를 찾기
        return res[0]

댓글