Binary Tree Longest Consecutive Sequence II (Leetcode 549)

不得不說,Tree的題還算挺有意思的。首先給出two pass的做法: pass one找往下的sequence,存到map里;然后 pass two找往上的sequence,再更新全局變量.

class Solution {
public:

    int search_smaller(TreeNode *root){
        if(!root) return 0;
        int left = search_smaller(root->left);
        int right = search_smaller(root->right);
        int cur = 1;
        if(root->left && root->left->val == root->val - 1){
            cur = max(cur, left + 1); 
        }
        if(root->right && root->right->val == root->val - 1){
            cur = max(cur, right + 1);
        }
        mp[root] = cur;
        return cur;
    }
    
    void search_bigger(TreeNode *root){
        if(!root) return;
        max_count = max(max_count, mp[root]);
        if(root->left && root->left->val == root->val + 1){
            mp[root->left] = max(mp[root->left], mp[root] + 1);
        }
        if(root->right && root->right->val == root->val + 1){
            mp[root->right] = max(mp[root->right], mp[root] + 1);
        }
        search_bigger(root->left);
        search_bigger(root->right);
    }
    
    int longestConsecutive(TreeNode* root) {
        if(!root) return 0;
        search_smaller(root);
        search_bigger(root);
        return max_count;
    }
private:
    unordered_map<TreeNode*, int> mp;
    int max_count = 1;
};

網上果然有one pass的解法,就是把同一個節點的增序數目,和降序數目統計到一起,然后該節點序列為 增序 + 降序 - 1,即可

class Solution {
    
   struct ResultSet{
        int dec, inc;
        ResultSet(int d, int i) : dec(d), inc(i){}
   };
    
public:

    ResultSet search_util(TreeNode* root){
        if(!root) return ResultSet(0, 0);
        ResultSet left = search_util(root->left);
        ResultSet right = search_util(root->right);
        ResultSet ret(1, 1);
        if(root->left){
            if(root->left->val == root->val - 1){
                ret.inc = max(ret.inc, left.inc + 1);
            }else if(root->left->val == root->val + 1){
                ret.dec = max(ret.dec, left.dec + 1);
            }
        }
        if(root->right){
           if(root->right->val == root->val - 1){
                ret.inc = max(ret.inc, right.inc + 1);
            }else if(root->right->val == root->val + 1){
                ret.dec = max(ret.dec, right.dec + 1);
            } 
        }
        max_count = max(max_count, ret.inc + ret.dec - 1);
        return ret;
    }

    int longestConsecutive(TreeNode* root) {
        if(!root) return 0;
        search_util(root);
        return max_count;
    }
private:
    int max_count = 1;
};
最后編輯于
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市,隨后出現的幾起案子,更是在濱河造成了極大的恐慌,老刑警劉巖,帶你破解...
    沈念sama閱讀 228,156評論 6 531
  • 序言:濱河連續發生了三起死亡事件,死亡現場離奇詭異,居然都是意外死亡,警方通過查閱死者的電腦和手機,發現死者居然都...
    沈念sama閱讀 98,401評論 3 415
  • 文/潘曉璐 我一進店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人,你說我怎么就攤上這事。” “怎么了?”我有些...
    開封第一講書人閱讀 176,069評論 0 373
  • 文/不壞的土叔 我叫張陵,是天一觀的道長。 經常有香客問我,道長,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 62,873評論 1 309
  • 正文 為了忘掉前任,我火速辦了婚禮,結果婚禮上,老公的妹妹穿的比我還像新娘。我一直安慰自己,他們只是感情好,可當我...
    茶點故事閱讀 71,635評論 6 408
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著,像睡著了一般。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發上,一...
    開封第一講書人閱讀 55,128評論 1 323
  • 那天,我揣著相機與錄音,去河邊找鬼。 笑死,一個胖子當著我的面吹牛,可吹牛的內容都是我干的。 我是一名探鬼主播,決...
    沈念sama閱讀 43,203評論 3 441
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了?” 一聲冷哼從身側響起,我...
    開封第一講書人閱讀 42,365評論 0 288
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后,有當地人在樹林里發現了一具尸體,經...
    沈念sama閱讀 48,881評論 1 334
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內容為張勛視角 年9月15日...
    茶點故事閱讀 40,733評論 3 354
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發現自己被綠了。 大學時的朋友給我發了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 42,935評論 1 369
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖,靈堂內的尸體忽然破棺而出,到底是詐尸還是另有隱情,我是刑警寧澤,帶...
    沈念sama閱讀 38,475評論 5 358
  • 正文 年R本政府宣布,位于F島的核電站,受9級特大地震影響,放射性物質發生泄漏。R本人自食惡果不足惜,卻給世界環境...
    茶點故事閱讀 44,172評論 3 347
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧,春花似錦、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 34,582評論 0 26
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至,卻和暖如春,著一層夾襖步出監牢的瞬間,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 35,821評論 1 282
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人。 一個月前我還...
    沈念sama閱讀 51,595評論 3 390
  • 正文 我出身青樓,卻偏偏與公主長得像,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當晚...
    茶點故事閱讀 47,908評論 2 372

推薦閱讀更多精彩內容

  • 背景 一年多以前我在知乎上答了有關LeetCode的問題, 分享了一些自己做題目的經驗。 張土汪:刷leetcod...
    土汪閱讀 12,762評論 0 33
  • LeetCode 刷題隨手記 - 第一部分 前 256 題(非會員),僅算法題,的吐槽 https://leetc...
    蕾娜漢默閱讀 17,878評論 2 36
  • 198. House Robber【Easy DP】You are a professional robber p...
    GeniusYY閱讀 1,157評論 0 0
  • 總結類型: 完全子樹(#222) BST(左右子樹值的性質,注意不僅要滿足parent-child relatio...
    __小赤佬__閱讀 704評論 0 0
  • 呵呵,只是一句謝謝! 斷了我所有的幻想! 我是個明明就很喜歡鉆石, 卻 可以抱著石頭回家的人! 不想那么復雜! 既...
    笨太狼閱讀 239評論 0 0