单值二叉树(深度优先搜索 && 遍历)

最新推荐文章于 2024-08-09 20:36:57 发布

E.L.E

最新推荐文章于 2024-08-09 20:36:57 发布

阅读量192

点赞数 1

分类专栏：算法文章标签：二叉树 java 数据结构算法 leetcode

本文链接：https://blog.csdn.net/weixin_48557384/article/details/109696102

版权

算法专栏收录该内容

38 篇文章 0 订阅

订阅专栏

这篇博客介绍了如何使用深度优先搜索（DFS）判断一个二叉树是否为单值二叉树。方法一是通过非递归的列表遍历实现，方法二是通过递归的方式分别检查左右子树。在Java和C++中给出了详细的代码实现，并解释了相关算法的运行原理和注意事项。

摘要由CSDN通过智能技术生成

单值二叉树(深度优先搜索 && 遍历)

题目描述

如果二叉树每个节点都具有相同的值，那么该二叉树就是单值二叉树。

只有给定的树是单值二叉树时，才返回 true；否则返回 false。

输入：[1,1,1,1,1,null,1]
输出：true

输入：[2,2,2,5,2]
输出：false

方法一（深度优先遍历）

题解(JAVA)

class Solution {
    List<Integer> vals;
    public boolean isUnivalTree(TreeNode root) {
        vals = new ArrayList();
        dfs(root);
        for (int v: vals)
            if (v != vals.get(0))
                return false;
        return true;
    }

    public void dfs(TreeNode node) {
        if (node != null) {
            vals.add(node.val);
            dfs(node.left);
            dfs(node.right);
        }
    }
}

详解

1>.首先定义泛型为Integer类型的列表 vals，因为我们对列表主要的操作为向尾部添加数据，所以用ArrayList的效率更高。
2>.之后我们需要定义一个方法用来遍历存储树的节点到 vals 中
3>.之后遍历比较列表中的元素与第一个元素（二叉树的根节点）的值,如果存在不相等的元素，则return false,比较所有的元素都相等，则return true

Tips1

ArrayList:根据索引查询数据的效率高，向尾部添加数据的效率高，但是删除和插入效率低，因为插入和删除需要移动数组，假如删除第一个数据，需要将第二个以后的数据向前移动一位
LinkList:添加和删除的效率高，只需要改变指针的指向即可，但是访问主句的效率较低，需要对链表进行遍历。

方法二（递归）

题解1(JAVA)

class Solution {
    public boolean isUnivalTree(TreeNode root) {
        boolean left_correct = (root.left == null ||
                (root.val == root.left.val && isUnivalTree(root.left)));
        boolean right_correct = (root.right == null ||
                (root.val == root.right.val && isUnivalTree(root.right)));
        return left_correct && right_correct;
    }
}

图示1（递归）

在这里插入图片描述

Tips2

‘||’该双目运算符，当返回类型为boolean时，前一个值为false时，需要运行‘||’后面的代码，只有当前后两个值都确定时才可得到最终的值
当前一个值为true时，不用继续运行后面的代码，可直接得到最终值为true

题解2(C++)

class Solution {
public:
    bool isUnivalTree(TreeNode* root) {
        if(root == NULL) {
            return true;
        }
        if(root->left && root->val != root->left->val) {
            return false;
        }
        if(root->right && root->val != root->right->val) {
            return false;
        }
        return isUnivalTree(root->left) && isUnivalTree(root->right);
    }
};