FazBrowse GitHub Viewer | Trending | | Home

Tools: [Download Repo ZIP] [View Raw Code] [Original HTTPS Page]

LeetCode-Solutions/Python/invert-binary-tree.py at master · tehsints/LeetCode-Solutions · GitHub

83 lines (67 loc) · 1.96 KB

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

# Time: O(n)

# Space: O(h)

import collections

# BFS solution.

class Queue(object):

def __init__(self):

self.data = collections.deque()

def push(self, x):

self.data.append(x)

def peek(self):

return self.data[0]

def pop(self):

return self.data.popleft()

def size(self):

return len(self.data)

def empty(self):

return len(self.data) == 0

# Definition for a binary tree node.

# class TreeNode:

# def __init__(self, x):

# self.val = x

# self.left = None

# self.right = None

class Solution(object):

# @param {TreeNode} root

# @return {TreeNode}

def invertTree(self, root):

if root is not None:

nodes = Queue()

nodes.push(root)

while not nodes.empty():

node = nodes.pop()

node.left, node.right = node.right, node.left

if node.left is not None:

nodes.push(node.left)

if node.right is not None:

nodes.push(node.right)

return root

# Time: O(n)

# Space: O(h)

# Stack solution.

class Solution2(object):

# @param {TreeNode} root

# @return {TreeNode}

def invertTree(self, root):

if root is not None:

nodes = []

nodes.append(root)

while nodes:

node = nodes.pop()

node.left, node.right = node.right, node.left

if node.left is not None:

nodes.append(node.left)

if node.right is not None:

nodes.append(node.right)

return root

# Time: O(n)

# Space: O(h)

# DFS, Recursive solution.

class Solution3(object):

# @param {TreeNode} root

# @return {TreeNode}

def invertTree(self, root):

if root is not None:

root.left, root.right = self.invertTree(root.right), \

self.invertTree(root.left)

return root

Back | FazBrowse Home | New Git URL