1. 程式人生 > >LeetCode 897 129 98 遞增順序查詢樹 求根到葉子節點之和 驗證二叉樹 (樹,深度優先搜尋)

LeetCode 897 129 98 遞增順序查詢樹 求根到葉子節點之和 驗證二叉樹 (樹,深度優先搜尋)

1.遞增順序查詢樹

難度:簡單
給定一個樹,按中序遍歷重新排列樹,使樹中最左邊的結點現在是樹的根,並且每個結點沒有左子結點,只有一個右子結點。
示例 :
輸入:[5,3,6,2,4,null,8,1,null,null,null,7,9]

       5
      / \
    3    6
   / \    \
  2   4    8
 /        / \ 
1        7   9

輸出:[1,null,2,null,3,null,4,null,5,null,6,null,7,null,8,null,9]

1
  \
   2
    \
     3
      \
       4
        \
         5
          \
           6
            \
             7
              \
               8
                \
                 9  

提示:
給定樹中的結點數介於 1 和 100 之間。
每個結點都有一個從 0 到 1000 範圍內的唯一整數值。
思路:中序遍歷得到樹的元素(可用迭代也可用遞迴),儲存在陣列中,然後建立新的樹。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode* increasingBST(TreeNode* root) {
        vector<int> valSet;
        stack<TreeNode*> stack;
        while(root || !stack.empty()){  //迭代求得中序遍歷
            while(root){
                stack.push(root);
                root=root->left;
            }
            if(!stack.empty()){
                valSet.push_back(stack.top()->val);
                root=stack.top()->right;
                stack.pop();
            }
        }
        TreeNode* increBST=new TreeNode(valSet[0]);
        TreeNode* tempBST= increBST; 
        for(int i=1;i<valSet.size();i++){
            tempBST->right=new TreeNode(valSet[i]);
            tempBST=tempBST->right;
        }
        return increBST;
    }
};

改用遞迴,其實時間是差不多的,基本上一樣

/**
* Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode* increasingBST(TreeNode* root) {
        vector<int> tempVal;
        inorderTraversal(root,tempVal);
        TreeNode* increBST=new TreeNode(tempVal[0]); //注意這裡需要new一個空間
        TreeNode* tempBST= increBST; 
        for(int i=1;i<tempVal.size();i++){
            tempBST->right=new TreeNode(tempVal[i]); //new 一個空間
            tempBST=tempBST->right;
        }
        return increBST;
      }
        void inorderTraversal(TreeNode* root,vector<int> &tempVal){ //遞迴中序遍歷得到樹的元素
            if(root->left!=NULL){//遍歷左子樹
                inorderTraversal(root->left,tempVal);
            }
            tempVal.push_back(root->val);//記錄根節點
            if(root->right!=NULL){//遍歷右子樹
                inorderTraversal(root->right,tempVal);
            }
        }
};

2.求根到葉子節點之和

難度:中等
給定一個二叉樹,它的每個結點都存放一個 0-9 的數字,每條從根到葉子節點的路徑都代表一個數字。
例如,從根到葉子節點路徑 1->2->3 代表數字 123。
計算從根到葉子節點生成的所有數字之和。
說明: 葉子節點是指沒有子節點的節點。

示例 1:
輸入: [1,2,3]

    1
   / \
  2   3

輸出: 25
解釋:
從根到葉子節點路徑 1->2 代表數字 12.
從根到葉子節點路徑 1->3 代表數字 13.
因此,數字總和 = 12 + 13 = 25.

示例 2:
輸入: [4,9,0,5,1]

    4
   / \
  9   0
 / \
5   1

輸出: 1026
解釋:
從根到葉子節點路徑 4->9->5 代表數字 495.
從根到葉子節點路徑 4->9->1 代表數字 491.
從根到葉子節點路徑 4->0 代表數字 40.
因此,數字總和 = 495 + 491 + 40 = 1026.

思路:很典型的一道題,也就是將每條路徑上的元素轉成string型別然後串聯起來,到達葉子節點後轉換成int型別push_back到容器中,再求和就行了,主要用到了遞迴實現深度優先搜尋。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    int sumNumbers(TreeNode* root) {
        if(root==NULL) return 0;
        vector<int> savePath;
        string str="";
        getSubPath(root,savePath,str);
        int sumNumber=0;
        vector<int>::iterator iter = savePath.begin();
	    while (iter != savePath.end()){
		     sumNumber += *iter++;   
        }
        return sumNumber;
    }
    void getSubPath(TreeNode* root, vector<int> &savePath, string str){
        str+=to_string(root->val);
        if(root->left==NULL && root->right==NULL) {
            savePath.push_back(stoi(str)); 
        }
        if(root->left!=NULL){
            getSubPath(root->left,savePath,str); //遞迴
        }
        if(root->right!=NULL){
            getSubPath(root->right,savePath,str);
        }
    }
};

3.驗證二叉樹

難度:中等
給定一個二叉樹,判斷其是否是一個有效的二叉搜尋樹。

假設一個二叉搜尋樹具有如下特徵:
節點的左子樹只包含小於當前節點的數。
節點的右子樹只包含大於當前節點的數。
所有左子樹和右子樹自身必須也是二叉搜尋樹。

示例 1:
輸入:

    2
   / \
  1   3

輸出: true

示例 2:
輸入:

    5
   / \
  1   4
     / \
    3   6

輸出: false
解釋: 輸入為: [5,1,4,null,null,3,6]。
根節點的值為 5 ,但是其右子節點值為 4 。

思路:遞迴的函式用得很巧,參考網上大神的。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    bool isValidBST(TreeNode* root) {
        return isValidBST(root, LONG_MIN, LONG_MAX);
    }
    
    bool isValidBST(TreeNode *root, long mn, long mx) {  //這個函式用得很巧
        if (!root) return true;
        if (root->val <= mn || root->val >= mx) return false;
        return isValidBST(root->left, mn, root->val) && isValidBST(root->right, root->val, mx);
    }
};