美文网首页
LeetCode之Leaf-Similar Trees(Kotl

LeetCode之Leaf-Similar Trees(Kotl

作者: 糕冷羊 | 来源:发表于2019-03-10 22:09 被阅读0次

    问题:
    Consider all the leaves of a binary tree. From left to right order, the values of those leaves form a leaf value sequence.



    For example, in the given tree above, the leaf value sequence is (6, 7, 4, 9, 8).
    Two binary trees are considered leaf-similar if their leaf value sequence is the same.
    Return true if and only if the two given trees with head nodes root1 and root2 are leaf-similar.
    Note:
    Both of the given trees will have between 1 and 100 nodes.


    方法:
    先遍历第一棵树的所有叶子节点,然后遍历第二棵树的所有叶子节点,然后比较两个叶子集合是否完全相同,如果完全相同则为相似树,如果不同则不是。

    具体实现:

    class LeafSimilarTrees {
    
        class TreeNode(var `val`: Int = 0) {
            var left: TreeNode? = null
            var right: TreeNode? = null
        }
    
        fun leafSimilar(root1: TreeNode?, root2: TreeNode?): Boolean {
            val leftLeafs = getLeafs(root1)
            val rightLeafs = getLeafs(root2)
            if (leftLeafs.size != rightLeafs.size) {
                return false
            }
            for (index in leftLeafs.indices) {
                if (leftLeafs[index].`val` != rightLeafs[index].`val`) {
                    return false
                }
            }
            return true
        }
    
        private fun getLeafs(root: TreeNode?): List<TreeNode> {
            if (root == null) {
                return emptyList()
            }
            val result = mutableListOf<TreeNode>()
            traverseLeafs(root, result)
            return result
        }
    
        private fun traverseLeafs(root: TreeNode?, result: MutableList<TreeNode>) {
            if(root == null) {
                return
            }
            if (root.left != null) {
                traverseLeafs(root.left, result)
            }
            if (root.right != null) {
                traverseLeafs(root.right, result)
            }
            if (root.left == null && root.right == null) {
                result.add(root)
            }
        }
    }
    

    有问题随时沟通

    具体代码实现可以参考Github

    相关文章

      网友评论

          本文标题:LeetCode之Leaf-Similar Trees(Kotl

          本文链接:https://www.haomeiwen.com/subject/oyxmpqtx.html