判断是否存在拓扑结构相同的子树(C++版)

来源:互联网 发布:linux创建多个定时器 编辑:程序博客网 时间:2024/05/28 15:06

题目:

对于两棵彼此独立的二叉树A和B,请编写一个高效算法,检查A中是否存在一棵子树与B树的拓扑结构完全相同。

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

实现思路:

将两颗树按照相同的遍历方式序列化为字符串 之后问题就转化为字符串匹配问题 大大降低时间复杂度【O(n+m)】。

注:

1. 常规的解决思路是树的遍历 逐个节点比较 这样的时间复杂度为O(n*m)。

2. n m分别为两颗树的节点总数


贴代码:

// 序列化树    void serialTree(TreeNode *T, string &s){if(T != NULL)         {                          s += T->val;              serialTree(T->left, s);              serialTree(T->right, s);          }          else          {              s += "#";          }}        // 求next数组    void nextArr(string str, vector<int> &v){if (str.length() == 1){v[0] = -1;return;}v[0] = -1;v[1] = 0;int pos = 2, cnt = 0;while(pos < v.size()){if (str[pos - 1] == str[cnt]){v[pos++] = ++cnt;}else if(cnt > 0){cnt = v[cnt];}else{v[pos++] = 0;}}}// KMP匹配算法int getIndex(string str, string mode){if (str.empty() || mode.empty() || mode.length() < 1 || str.length() < mode.length()){return -1;}vector<int> next(mode.length());nextArr(mode, next);int idx = 0, modeIdx = 0;while(idx < str.length() && modeIdx < mode.length()){if (str[idx] == mode[modeIdx]){idx++;modeIdx++;}else if (next[modeIdx] == -1){idx++;}else{modeIdx = next[modeIdx];}}return ((modeIdx == mode.length()) ? (idx - modeIdx) : -1);}    bool chkIdentical(TreeNode* A, TreeNode* B) {        string str1, str2;        serialTree(A, str1);        serialTree(B, str2);              return (getIndex(str1, str2) != -1);    }


0 0