FB面经 Valid binary tree

给你一个array of treenodes, 检验是否是一个完整的二叉树

  static boolean valid(TreeNode[] arr) {
    Set<TreeNode> set = new HashSet<>();
    for (TreeNode root: arr) {
      if (set.contains(root.left)) {
        return false;
      }  
      if (set.contains(root.right)) {
        return false
      }
      if (root.left != null)
      set.add(root.left);
      if (root.right != null)
      set.add(root.right);
    }
    // 就剩下根不在set里面
    if (set.size() != arr.length - 1) {
      return false;
    }
    // 检验这些node 是否在tree的集合里
    int cnt = 0;
    for (TreeNode root: arr) {
      if (set.contains(root))
        cnt++;
    }
    return arr.length - cnt == 1;
  }
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。