力扣 331. 验证二叉树的前序序列化 思维

https://leetcode-cn.com/problems/verify-preorder-serialization-of-a-binary-tree/
在这里插入图片描述
思路:最直观的方法就是把树重构出来,然后进行判断。这里说一下不需要重构的方法,考虑用一栈维护当前所有已知节点的还需要填充的儿子数量,那么如果遇到了 # \# #,直接把栈顶值减一即可;如果遇到了非空节点,仍要把栈顶值减一,然后再压入一个2(新节点还有两个儿子节点待补充);如果栈顶为空,那么直接丢掉即可。在字符串还未遍历完时栈已经为空,说明无解(未用完给定序列已经构造出了一棵二叉树);若字符串遍历完时栈不为空,说明无解(用完给定序列还未构造出二叉树)。为了处理只有1个节点的情况,我们可以在初始时压入一个1。现在在这个思想上更进一步,我们其实不需要栈,用一个整数即可维护所有状态(相当于栈的总和),那么套用之前的思路把逻辑部分替换一下即可。

class Solution {
    
    
public:
    bool isValidSerialization(string preorder) {
    
    
        int i=0,ans=1,n=preorder.size();
        while(i<n)
        {
    
    
            if(!ans)
                return 0;
            if(preorder[i]==',')
                ++i;
            else if(preorder[i]=='#')
                --ans,++i;
            else
            {
    
    
                while(preorder[i]>='0'&&preorder[i]<='9')
                    ++i;
                ++ans;
            }
        }
        return !ans;
    }
};

猜你喜欢

转载自blog.csdn.net/xiji333/article/details/114684382