LeetCode 두 갈래 나무 뒤집기

5382 단어 LeetCode

제목


가오리가 두 갈래 나무를 뒤집다.간단하게 말하면 두 갈래 나무의 각 노드의 좌우 노드를 교환하는 것이다.

예제

input:
	     4
   	   /   \
      2     7
     / \   / \
    1   3 6   9
output:
	     4
       /   \
      7     2
     / \   / \
    9   6 3   1

해법


깊이 우선 순위 - 반복 구현(Python)

class Solution:
    def invertTree(self, root: TreeNode) -> TreeNode:
        if(root == None):
            return None
        tmp = root.left
        root.left = root.right
        root.right = tmp
        self.invertTree(root.left)
        self.invertTree(root.right)
        return root

기본 사고방식
  는 위에서 아래로 깊이를 우선적으로 훑어보는 방식으로 먼저 root == None의 기준 상황을 처리한 다음에 현재 노드의 좌우 노드를 교환한다.교환이 끝난 후에 각각 좌우 나무를 돌려라.
복잡도 분석
   시간적으로 각 노드에 접근하여 좌우 하위 노드를 뒤집어야 하기 때문에 복잡도는 O(N) O\left(N\right) O(N)입니다.공간적으로 귀속을 사용하기 때문에 최악은 O(N) O\left(N\right) O(N), 평균은 O(l o g N) O\left(logN\right) O(logN)입니다.

광도 우선 반복 - 반복 구현(Python)

class Solution:
    def invertTree(self, root: TreeNode) -> TreeNode:
        if(root == None):
            return None
        n_queue = [root]
        while(n_queue != []):
            node = n_queue.pop(0)
            tmp = node.left
            node.left = node.right
            node.right = tmp
            if node.left:
                n_queue.append(node.left)
            if node.right:
                n_queue.append(node.right)
        return root

기본 사고방식
  는 한 노드에 접근할 때마다 이 노드의 좌우 노드를 뒤집은 다음에 None가 아닌 좌우 노드를 모든 노드에 접근할 때까지 대기열에 눌러 넣는다.
복잡도 분석
   시간적으로도 모든 노드에 접근해야 하기 때문에 시간 복잡도는 O(N) O\left(N\right) O(N)입니다.공간적으로 극단적인 상황은 루트 노드만 있고 모든 노드가 대기열에 있으며 복잡도는 O(N) O\left(N\right) O(N)입니다.두 갈래 트리를 가득 채우는 데 필요한 대기열의 길이는 노드가 가장 많은 층의 노드 수와 같으며, 복잡도는 O(⌈N2⌉)O\left(\lceil\frac{N}{2}\rceil\right)O(⌈2N⌉)입니다.

좋은 웹페이지 즐겨찾기