255. Verify Preorder Sequence in Binary Search Tree (M)

Given an array of numbers, verify whether it is the correct preorder traversal sequence of a binary search tree.

You may assume each number in the sequence is unique.

Consider the following binary search tree:

 5
/ \

2 6
/
1 3
Example 1:

Input: [5,2,6,1,3]
Output: false
Example 2:

Input: [5,2,1,3,6]
Output: true
Follow up:
Could you do it using only constant space complexity?


我的答案:TLE了

首先看题就懵了,没看懂什么意思。还是不熟悉BST是什么含义啊。

class Solution {
public:
    bool verifyPreorder(vector<int>& preorder) { 
        // edge case, empty input
        return helper(preorder, 0, preorder.size());
    }
    
    bool helper(vector<int>& preorder, int left, int right) {
        if (right-left <= 1)
            return true;
        int root_val = preorder[left];
        int trans_point = right; 
        // init, what if no right branch? -> trans = right -> empty -> true
        for (int i=left+1; i<right; ++i) {
            if (preorder[i] > root_val and trans_point == right) { // unique val
                trans_point = i;
            }
            if (i > trans_point and preorder[i] < root_val)
                return false;
        }
        return helper(preorder, left+1, trans_point) and helper(preorder, trans_point, right);
    }
};

写的第一版一个问题是,for循环的第一个if,没考虑到trans_point第一次变值之后就不用变了


小修补,提升速度的办法:剪枝
参考https://www.cnblogs.com/grandyang/p/5327635.html的方法3

class Solution {
public:
    bool verifyPreorder(vector<int>& preorder) { 
        return helper(preorder, 0, preorder.size(), INT_MIN, INT_MAX);
    }
    
    bool helper(vector<int>& preorder, int left, int right, int lower, int upper) {
        if (right-left < 1)
            return true;
        int root_val = preorder[left];
        if (root_val > upper or root_val < lower)
            return false;
        
        int i=left+1;
        for (; i<right; ++i) {
            if (preorder[i] > root_val) { // unique val
                break;
            }
        }
        
        return helper(preorder, left+1, i, lower, root_val) and helper(preorder, i, right, root_val, upper);
    }
};

Runtime: 852 ms, faster than 16.61% of C++ online submissions for Verify Preorder Sequence in Binary Search Tree.
Memory Usage: 22.9 MB, less than 26.44% of C++ online submissions for Verify Preorder Sequence in Binary Search Tree.

我也尝试写过有lower upper的程序,但是没有提前剪枝,反而更慢了。这里用lower upper和root_val比,替代剪枝掉的判断是否比trans_point小的功能。不过还是很慢


方法1用stack,https://www.cnblogs.com/grandyang/p/5327635.html

我的写法:

class Solution {
public:
    bool verifyPreorder(vector<int>& preorder) { 
        if (preorder.empty())
            return true;
        
        stack<int> s({preorder[0]});
        int ptr = 1;
        int low = INT_MIN;
        
        while (ptr < preorder.size() and !s.empty()) {
            while (ptr < preorder.size() and preorder[ptr] < s.top()) {
                s.push(preorder[ptr++]);
                if (s.top() < low)
                    return false;
            }
            while (ptr < preorder.size() and !s.empty() and s.top() < preorder[ptr]) {
                low = s.top();
                s.pop();
            }
            if (ptr < preorder.size()) {
                s.push(preorder[ptr++]);                
                if (s.top() < low)
                    return false;
            }
        }
        
        return true;
    }
};

Runtime: 64 ms, faster than 68.46% of C++ online submissions for Verify Preorder Sequence in Binary Search Tree.
Memory Usage: 23.1 MB, less than 12.42% of C++ online submissions for Verify Preorder Sequence in Binary Search Tree.

可以看到中间写起来非常麻烦,稍不小心就漏一个条件,链接里面的写法就比较简洁了。while循环得很注意各种边界条件,然后一个while循环里面不同步骤,得仔细看是否前面会影响后面的边界。所以还是用for循环写一次

class Solution {
public:
    bool verifyPreorder(vector<int>& preorder) { 
        if (preorder.empty())
            return true;
        
        stack<int> s;
        int low = INT_MIN;
        
        for (const auto& val : preorder) {
            if (val < low)
                return false;
            
            while (!s.empty() and s.top() < val) {
                low = s.top();
                s.pop();
            }
            
            s.push(val);
        }
        
        return true;
    }
};

Runtime: 64 ms, faster than 68.46% of C++ online submissions for Verify Preorder Sequence in Binary Search Tree.
Memory Usage: 23 MB, less than 17.11% of C++ online submissions for Verify Preorder Sequence in Binary Search Tree.

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 195,898评论 5 462
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 82,401评论 2 373
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 143,058评论 0 325
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 52,539评论 1 267
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 61,382评论 5 358
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 46,319评论 1 273
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 36,706评论 3 386
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 35,370评论 0 254
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 39,664评论 1 294
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 34,715评论 2 312
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 36,476评论 1 326
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 32,326评论 3 313
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 37,730评论 3 299
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,003评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 30,275评论 1 251
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 41,683评论 2 342
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 40,877评论 2 335

推荐阅读更多精彩内容