245. 子树

题目

有两个不同大小的二叉树: T1 有上百万的节点; T2 有好几百的节点。请设计一种算法,判定 T2 是否为 T1的子树。
若 T1 中存在从节点 n 开始的子树与 T2 相同,我们称 T2 是 T1 的子树。也就是说,如果在 T1 节点 n 处将树砍断,砍断的部分将与 T2 完全相同。
http://www.lintcode.com/zh-cn/problem/subtree/

分析

其实拆分一下,就是两个步骤:

  1. 搜索T1中是否存在与T2根节点值相同的节点,如果有,作为候选节点
  2. 考察T1中,以候选节点为根节点,与T2进行比较,从根节点到各个子节点是否值都相等,若全部相等,则T2是T1的子树,否则不是。

T1中有可能存在若干个候选节点,需要逐一考察它们。

代码


/**
 * Definition of TreeNode:
 * public class TreeNode {
 *     public int val;
 *     public TreeNode left, right;
 *     public TreeNode(int val) {
 *         this.val = val;
 *         this.left = this.right = null;
 *     }
 * }
 */


public class Solution {
    
    void findNode(TreeNode tree, int target, List result){
        if(tree == null) return;
        
        if(tree.val == target){
            result.add(tree);
        }
        this.findNode(tree.left, target, result);
        this.findNode(tree.right, target, result);
    }
    
    List findTargetNode(TreeNode tree, int target){
        List result = new ArrayList();
        
        findNode(tree, target, result);
        
        return result;
    }
    
    boolean compare(TreeNode a, TreeNode b){
        if ( a == null && b == null) return true;
        if( a != null && b != null){
            if( a.val == b.val){
                return this.compare(a.left, b.left) && this.compare(a.right, b.right);
            }else{
                return false;
            }
        }
        return false;
    }
    /*
     * @param T1: The roots of binary tree T1.
     * @param T2: The roots of binary tree T2.
     * @return: True if T2 is a subtree of T1, or false.
     */
    public boolean isSubtree(TreeNode T1, TreeNode T2) {
        // write your code here
        if(T2 == null) return true;
        List candicates = this.findTargetNode(T1, T2.val);
        if(candicates.size() < 1) return false;
        
        for(int i=0; i < candicates.size(); i++){
            if ( this.compare(candicates.get(i), T2) ){
                return true;
            }
        }
        return false;
    }
}
Show Comments

Get the latest posts delivered right to your inbox.