这一部分主要来梳理二叉树题目最简单最基础的部分,包括遍历,一些简单题目。
https://leetcode.cn/problems/binary-tree-preorder-traversal/description/
二叉树的遍历是入门。我们需要在程序一开始就创建一个空的数组,然后递归遍历左右节点,将节点放进这个数组内。下面直接给出代码:
func preorderTraversal(root *TreeNode) []int {
if root == nil{
return nil
}
res := make([]int,0)
res = append(res,root.Val)
res = append(res,preorderTraversal(root.Left)...)
res = append(res,preorderTraversal(root.Right)...)
return res
}
append方法后面的“...”说明:
使用append
函数和...
操作符将这个切片中的所有元素添加到res
中。...
操作符的作用是将一个切片的所有元素作为独立的参数传递给函数,而不是将整个切片作为一个单独的参数。这意味着如果preorderTraversal(root.Left)
返回一个切片[a, b, c]
,append(res, preorderTraversal(root.Left)...)
实际上是append(res, a, b, c)
。
https://leetcode.cn/problems/binary-tree-inorder-traversal/
和前序遍历不同,但也很类似,只是将append的操作放到了递归遍历左右子树的操作中间。
func inorderTraversal(root *TreeNode) []int {
if root == nil{
return nil
}
res := make([]int,0)
res = append(res,inorderTraversal(root.Left)...)
res = append(res,root.Val)
res = append(res,inorderTraversal(root.Right)...)
return res
}
https://leetcode.cn/problems/binary-tree-postorder-traversal/description/
func postorderTraversal(root *TreeNode) []int {
res := make([]int,0)
if root == nil{
return nil
}
res = append(res,postorderTraversal(root.Left)...)
res = append(res,postorderTraversal(root.Right)...)
res = append(res,root.Val)
return res
}
https://leetcode.cn/problems/same-tree/description/
一共有两棵二叉树,我们怎么判断他们是一样的?要是两棵树都是空的,那就是一样的,要是有一棵树是空的,另一棵树却不是,那就不是一样的,如果两棵树都不为空,但是对应节点的值不一样,那也不是一样的。然后我们递归遍历两棵树的左右子树就可以了。
func isSameTree(p *TreeNode, q *TreeNode) bool {
if q ==nil && p == nil{
return true
}
if q ==nil || p == nil{
return false
}
if p.Val != q.Val{
return false
}
return isSameTree(p.Left,q.Left) && isSameTree(p.Right,q.Right)
}
https://leetcode.cn/problems/symmetric-tree/description/
和上面一题的思路一样。不过我们要再写一个辅助函数。来将一个二叉树的左右节点传进去,检测左子树的左侧与右子树的右侧,以及左子树的右侧与右子树的左侧是否相等。
func isSymmetric(root *TreeNode) bool {
// 如果树为空,那么它是对称的。
if root == nil {
return true
}
// 开始从根节点的左右子树进行比较
return isMirror(root.Left, root.Right)
}
// 辅助函数,用于递归比较两个节点是否是镜像对称的
func isMirror(left, right *TreeNode) bool {
// 如果两个节点都为空,那么它们是对称的
if left == nil && right == nil {
return true
}
// 如果其中一个节点为空,另一个不为空,那么它们不对称
if left == nil || right == nil {
return false
}
// 检查当前两个节点的值是否相等,且左节点的左子树和右节点的右子树是否对称,
// 左节点的右子树和右节点的左子树是否对称
return left.Val == right.Val && isMirror(left.Left, right.Right) && isMirror(left.Right, right.Left)
}
https://leetcode.cn/problems/sum-of-left-leaves/
我们需要判断哪些节点是左叶子。然后直接递归。
func sumOfLeftLeaves(root *TreeNode) int {
if root == nil {
return 0
}
sum := 0
// 检查左节点是否为叶节点
if root.Left != nil && root.Left.Left == nil && root.Left.Right == nil {
sum += root.Left.Val
}
// 递归地累加左右子树的左叶节点的和
sum += sumOfLeftLeaves(root.Left)
sum += sumOfLeftLeaves(root.Right)
return sum
}
https://leetcode.cn/problems/sZ59z6/description/
这里我们创建一个"set",将节点的值放进这个set中,遍历完之后,直接返回set的长度。在Golang中没有set这个结构,所以我们用字典来模拟,一样。
func numColor(root *TreeNode) int {
set := make(map[int]bool)
treeset(root,set)
return len(set)
}
func treeset(root *TreeNode,set map[int]bool){
if root == nil{
return
}
set[root.Val] = true
treeset(root.Left,set)
treeset(root.Right,set)
}