拓扑结构相同子树

https://www.nowcoder.com/study/vod/1/3/2
对于两棵彼此独立的二叉树A和B,请编写一个高效算法,检查A中是否存在一棵子树与B树的拓扑结构完全相同。

给定两棵二叉树的头结点A和B,请返回一个bool值,代表A中是否存在一棵同构于B的子树。

1, 普通二叉树解法:遍历+匹配,B要么和A完全一样,要么是A的左子树的子树要么是A的右子树的子树。需要考察A中以每个节点开头的子树是否和B的头结点相同,如果相同则继续匹配孩子节点。如果不相同则继续考察A的左子树和右子树。时间复杂度是O(N*M),N是A的节点数M是B的节点数。

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};*/

class IdenticalTree {
public:
    bool isSame(TreeNode* A,TreeNode* B){
        if(B == NULL && A == NULL)
            return true;
        if(A == NULL && B != NULL)
            return false;
        if(A != NULL && B == NULL)
            return false;
        if(A->val != B->val)
            return false;
        return isSame(A->left, B->left)&& isSame(A->right, B->right);

    }
    bool chkIdentical(TreeNode* A, TreeNode* B) {
        // write code here
        //只要B是空的就返回true
        if(B==NULL) return true;
        bool flag = false;
        if(A)
            if(A->val == B->val){
                flag = isSame(A,B);
            }
            if(!flag) flag = chkIdentical(A->left,B);
            if(!flag) flag = chkIdentical(A->right,B);
        }
        return flag;
    }
};

2, 转化为字符串匹配问题,按照相同顺序遍历两个二叉树得到两个字符串,然后用字符串匹配算法KMP去判断第一个字符串中是否包含第二个字符串,如果包含则返回true否则false。在将数转化成字符串时,如果当前节点为空就用”#!”代替,访问完一个节点后添加一个”#”标志,以区分12和1 2这种情况。

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};*/

class IdenticalTree {
public:

    vector<int> get_next(string &t){
        vector<int> next(t.length());
        int i = -1;//前缀
        int j = 0;//后缀
        next[0] = -1;//初始化第一个字符的模式值
        while(j < t.length()-1){//注意是t.length()-1,后缀表示当前字符的前一个
            if(i==-1 || t[i] == t[j]){
                i++;
                j++;
                next[j] = i;
            }else{
                //前缀回溯
                i = next[i];
            }
        }
        return next;
    }

    bool kmp(string &s,string &t){
        int i=0,j=0;
        vector<int> next = get_next(t);
        while(s[i] && t[j]){
            if(s[i] == t[j]){
                i++;
                j++;
            }else if(j == 0){//j回溯到0位置,说明都不匹配,i+1继续下一轮
                i++;
            }else{
                j = next[j];
            }
        }
        if(!t[j]) return true;
        return false;
    }

    void treeToString(TreeNode* t,string &str){
        if(t==NULL){
            str+="#!";
            return;
        }
        str+=t->val;
        str+="#";

        treeToString(t->left,str);
        treeToString(t->right,str);
    }

    bool chkIdentical(TreeNode* A, TreeNode* B) {
        // write code here
        string str_a;
        string str_b;

        treeToString(A,str_a);
        treeToString(B,str_b);

        return kmp(str_a,str_b);
    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值