[LeetCode] 654. Maximum Binary Tree

  1. Given an integer array with no duplicates. A maximum tree building on this array is defined as follow:

    The root is the maximum number in the array.
    The left subtree is the maximum tree constructed from left part subarray divided by the maximum number.
    The right subtree is the maximum tree constructed from right part subarray divided by the maximum number.
    Construct the maximum tree by the given array and output the root node of this tree.

    Example 1:

    Input: [3,2,1,6,0,5]
    Output: return the tree root node representing the following tree:
    
          6
        /   \
       3     5
        \    / 
         2  0   
           \
            1
    

    Note:
    The size of the given array will be in the range [1,1000].


    題目不算難,可以很容易的想到解決思路:

    1. 創建一個根節點,其值為數組的最大值
    2. 創建根節點的左節點,左節點的值為數組左半部分的最大值
    3. 創建根節點的右節點,右節點的值為數組右半部分的最大值
    4. 遞歸以上操作

    于是我用最直接的代碼寫出了以下解決方法

    Solution:

    class TreeNode {
     int val;
     TreeNode left;
     TreeNode right;
     TreeNode(int x) {
         val = x;
     }
    }
    public class Solution {
     public static TreeNode constructMaximumBinaryTree(int[] nums) {
         if(nums.length==0) return null;
         int max=getMaxNumber(nums);
         TreeNode root=new TreeNode(max);
         root.left=constructMaximumBinaryTree(getArrayLeftPart(max, nums));
         root.right=constructMaximumBinaryTree(getArrayRightPart(max, nums));
         return root;
     }
     public static int getMaxNumber(int[] nums) {
         int[] tempArray = nums.clone();
         Arrays.sort(tempArray);
         return tempArray[tempArray.length - 1];
     }
     public static int[] getArrayLeftPart(int spilt, int[] nums) {
         List<Integer> list = new ArrayList<Integer>();
         for (int i = 0; i < nums.length; i++) {
             if (nums[i] == spilt) {
                 break;
             }
             list.add(nums[i]);
         }
         int[] result = new int[list.size()];
         for (int k = 0; k < list.size(); k++) {
             result[k] = list.get(k);
         }
         return result;
     }
     public static int[] getArrayRightPart(int spilt, int[] nums) {
         List<Integer> list = new ArrayList<Integer>();
         for (int i = nums.length - 1; i > 0; i--) {
             if (nums[i] == spilt) {
                 break;
             }
             list.add(nums[i]);
         }
         int[] result = new int[list.size()];
         for (int k = list.size() - 1; k >= 0; k--) {
             result[list.size() - k - 1] = list.get(k);
         }
         return result;
     }
    }
    

    這個解法給了AC,但是效率很低,Leetcode給出的運行時間是433ms。對于性能差我能想到的原因有:

    1. 頻繁的創建數組,獲取數組最大值需要克隆數組,獲取數組的左半部分也需要從list轉為數組,這些都會影響程序的性能。
    2. 頻繁遍歷數組,這個不用我多說,看代碼就可以。

    性能優化

    上面的方法使用在遞歸時傳的是值(value),這樣的遞歸需要傳遞的參數個數比較少,但對資源的利用率不高。所以我們改為傳遞索引(index)的方式,提高對數組的利用率。這樣在數據傳遞的時候只需要傳遞數組的索引值,整個遞歸過程只需要對一個數組進行操作。

     public static TreeNode constructMaximumBinaryTree(int[] nums){
          return constructor(nums,0,nums.length-1);
      }
      //給出的method只能傳遞一個參數,所以用一個新的method
      public static TreeNode constructor(int[] nums,int left,int right){
          if (left>right)
              return null;
          int max=MaxArrayIndex(nums, left, right);
          TreeNode root=new TreeNode(nums[max]);
          root.left=constructor(nums,left,max-1);
          root.right=constructor(nums, max+1, right);
          return root;
      }
      //獲取數組中最大值的索引
      public static int MaxArrayIndex(int[] nums,int left,int right){
          int point=nums[left];
          int max_index=left;
          for(int i=left+1;i<=right;i++){
              if(nums[i]>point){
                  point=nums[i];
                  max_index=i;
              }else continue;
          }
          return max_index;
      }
    

改進后運行時間降到了13ms,超過了66.8%的提交。

邊界問題

邊界問題似乎不是什么大問題,但它卻會在一定程度上阻礙你解題的速度。

首先我們得先了解哪里會出現數組越界的情況。容易想到越界肯定容易在索引發生變化的地方出現,也就是在遞歸那一步的時候。由于在遞歸時我們改變了數組的索引,所以我們必須在函數的第一步判斷索引是不是有效的。在本例中你需要保證:

  1. MaxArrayIndex 函數的索引保持的合法范圍內。
  2. constructMaximumBinaryTree 的參數保持在合法范圍內,本例中顯然沒有問題。
  3. 遞歸函數constructor 中合理的索引校驗

第一點保證了遞歸內部獲取的索引一定是合法的。第二點保證了,調用遞歸時索引的值是合法的。第三點保證了子遞歸出現索引問題時能夠及時阻斷這個子遞歸。

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

推薦閱讀更多精彩內容