题目地址:
https://leetcode.com/problems/verify-preorder-serialization-of-a-binary-tree/
给定一棵二叉树的前序遍历,问其是否合法。该前序遍历是含null节点的,null以'#'
表示。其余节点以数字表示,以','
隔开。
直接模拟以二叉树前序遍历建树的过程就行了。先以","
做split,这样就得到了每个节点元素。接着开始模拟建树,并用一个下标来跟踪用到了哪个节点。如果该建立节点但是无元素可用了,则返回false;如果是null,则建出null节点,后移指针并返回true,否则说明是一个非null节点,建立之指针后移,然后递归建立左子树和右子树,如果其一建立失败就返回false,否则返回true。代码如下:
class Solution {
public:
bool isValidSerialization(string t) {
istringstream ss(t);
vector<string> v;
for (string s; getline(ss, s, ',');) v.push_back(s);
int idx = 0;
return dfs(idx, v) && idx == v.size();
}
bool dfs(int &idx, vector<string> &v) {
if (idx == v.size()) return false;
if (v[idx] == "#") {
idx++;
return true;
}
idx++;
return dfs(idx, v) && dfs(idx, v);
}
};
时间复杂度 O ( n ) O(n) O(n), n n n是字符串长度,空间 O ( h ) O(h) O(h), h h h是树高。