Leetcode - Verify Preorder Serialization of a Binary Tree

My code:

public class Solution {
    public boolean isValidSerialization(String preorder) {
        String[] nodes = preorder.split(",");
        int diff = 1;
        for (int i = 0; i < nodes.length; i++) {
            diff--;
            if (diff < 0) {
                return false;
            }
            if (!nodes[i].equals("#")) {
                diff += 2;
            }
        }
        
        return diff == 0;
    }
}

reference:
https://discuss.leetcode.com/topic/35976/7-lines-easy-java-solution/5

自己写的stack解法超时。待会再解释。
然后我看了解法,被 dietpepsi的这个解法惊讶到了。
其实原理很简单,用到的数据结构思想也很简单,我们都会,但真的想不到。主要还是学的太死,入度,出度,从来只是知道而从来没用过。
这道题目的思想就是,从根节点开始pre-order,一开始入度是1,
然后,每到一个节点,
如果是null, diff 减去一个入度,然后加上两个出度。
如果是null, diff减去一个入度就够了
这样的话,在整个过程中,diff 始终 >=0, 如果出现 < 0, 那么一定错了。
最后,diff 一定等于0,因为对于这样的丰满的树,他的
入度 = 出度

然后衍生思考,post-order 该如何验证。

Their code:

public boolean isValidSerialization_PostOrder(String postorder) {
    String[] nodes = postorder.split(",");
    int diff = 1;
    for (String node : nodes) {
        diff--;
        if (!node.equals("#")) diff += 2;
        // Post-Order traversal fail criteria
        if (diff > 0) return false;
    }
    return diff == 0;
}

还是很巧妙地。

然后讲下 stack 的做法:

My code:

public class Solution {
    public boolean isValidSerialization(String preorder) {
        String[] nodes = preorder.split(",");
        Stack<String> st = new Stack<String>();
        for (int i = 0; i < nodes.length; i++) {
            String val = nodes[i];
            while (val.equals("#") && !st.isEmpty() && st.peek().equals("#")) {
                st.pop();
                if (st.isEmpty()) {
                    return false;
                }
                st.pop();
            }
            st.push(val);
        }
        
        return st.size() == 1 && st.peek().equals("#");
    }
}

reference:
https://discuss.leetcode.com/topic/35973/java-intuitive-22ms-solution-with-stack

其实我一开始也是用 stack 做的。但是我的stack只允许插入节点,不允许插入 #。
这种解法,允许插入 #,并且有些情况下, 如果栈顶是 #, 当前字符串也是 #,那么就需要把 # 弹出栈。
其实我还是不明白为什么我的解法会超时,他的不会。

Anyway, Good luck, Richardo! — 09/16/2016

    原文作者:Richardo92
    原文地址: https://www.jianshu.com/p/29363bba21f2#comments
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞