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

题目：

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

给定两棵二叉树的头结点A和B，请返回一个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);
    }