FilesExpand file tree

master

/

114_flatten.go

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

54 lines (46 loc) · 821 Bytes

master

/

114_flatten.go

Copy path

Top

File metadata and controls

• Code
• Blame

54 lines (46 loc) · 821 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

package main

/**

* Definition for a binary tree node.

*/

type TreeNode struct {

Val int

Left *TreeNode

Right *TreeNode

}

func flatten(root *TreeNode) {

if root == nil {

return

}

result := root

left := root.Left

right := root.Right

root.Left = nil

root.Right = left

if left != nil {

result = flattenDps(left)

}

if result != nil && right != nil {

result.Right = right

result = flattenDps(right)

}

}

func flattenDps(node *TreeNode) *TreeNode {

if node == nil {

return node

}

result := node

right := node.Right

if node.Left != nil {

node.Right = node.Left

node.Left = nil

result = node.Right

result = flattenDps(node.Right)

fmt.Println(result.Val)

}

if right != nil {

result.Right = right

result = flattenDps(node.Right)

fmt.Println(result.Val)

}

return result

}