牛客题霸 NC14 按之字形顺序打印二叉树

Posted 2021-09-04 Starzkg

https://www.nowcoder.com/practice/91b69814117f4e8097390d107d2efbe0

解决方案

Go

func Print(pRoot *TreeNode) [][]int {
	NC14dfs(pRoot, 0)
	return NC14ans
}

var NC14ans = make([][]int, 0)

func NC14dfs(pRoot *TreeNode, num int) {
	if pRoot == nil {
		return
	}
	if len(NC14ans) == num {
		NC14ans = append(NC14ans, []int{})
	}
	if num%2 == 1 {
		NC14ans[num] = append(NC14ans[num], pRoot.Val)
	} else {
		NC14ans[num] = append([]int{pRoot.Val}, NC14ans[num]...)
	}
	if pRoot.Right != nil {
		NC14dfs(pRoot.Right, num+1)
	}
	if pRoot.Left != nil {
		NC14dfs(pRoot.Left, num+1)
	}
}

参考文章