리트코드 풀이

leetcode 1791. Find Center of Star Graph (파이썬)

ag2개발자 2022. 2. 26. 18:48

별모양 그래프의 중심을 찾는 문제.

한 노드와 모든 노드가 연결돼 있으므로 그 노드를 찾으면 된다.

그 노드는 반드시 선분에 포함돼 있으므로 edges[0]과 edges[1]만 비교해 보더라도

엣지들에서 공통되는 노드가 중심 노드이다.

class Solution:
    def findCenter(self, edges: List[List[int]]) -> int:
        if edges[0][0]==edges[1][0] or edges[0][0]==edges[1][1]:
            return edges[0][0] 
        return edges[0][1]