1. 程式人生 > 其它 >998 最大二叉樹 II

998 最大二叉樹 II

技術標籤:LeetCode

題目描述:
最大樹定義:一個樹,其中每個節點的值都大於其子樹中的任何其他值。
給出最大樹的根節點 root。
就像之前的問題那樣,給定的樹是從表 A(root = Construct(A))遞迴地使用下述 Construct(A) 例程構造的:
如果 A 為空,返回 null
否則,令 A[i] 作為 A 的最大元素。建立一個值為 A[i] 的根節點 root
root 的左子樹將被構建為 Construct([A[0], A[1], …, A[i-1]])
root 的右子樹將被構建為 Construct([A[i+1], A[i+2], …, A[A.length - 1]])

返回 root
請注意,我們沒有直接給定 A,只有一個根節點 root = Construct(A).
假設 B 是 A 的副本,並附加值 val。保證 B 中的值是不同的。
返回 Construct(B)。

示例 1:
在這裡插入圖片描述
輸入:root = [4,1,3,null,null,2], val = 5
輸出:[5,4,null,1,3,null,null,2]
解釋:A = [1,4,2,3], B = [1,4,2,3,5]

示例 2:
在這裡插入圖片描述
輸入:root = [5,2,4,null,1], val = 3
輸出:[5,2,4,null,1,null,3]
解釋:A = [2,1,5,4], B = [2,1,5,4,3]

示例 3:
在這裡插入圖片描述
輸入:root = [5,2,3,null,1], val = 4
輸出:[5,2,4,null,1,3]
解釋:A = [2,1,5,3], B = [2,1,5,3,4]

提示:
1 <= B.length <= 100

方法1:
主要思路:解題彙總連結
(1)直接處理給出的根節點為空或根節點的值小於附加值的情形;
(2)然後使用遞迴插入給定的值,由於附加值是放到陣列的左右邊的,根據題目中的構造樹的特定,該附加值最後一定作為原樹的某個右結點插入;
(3)則遞迴時,需要判讀當前結點的右結點是否為空,若為空,則直接將該值作為新的右結點插入;
(4)若不為空,則判讀當前結點的右結點的值是否小於給定的值,若小於,則將當前結點作為右結點插入,原來的右結點作為新的右結點的左結點插入;

(5)否則,繼續遞迴判斷當前結點的右結點;

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
	//遞迴插入
    void insert_node(TreeNode* root,int& val){
        if(root->right==nullptr){//說明當前結點的右結點為空,則直接插入給定的值
            root->right=new TreeNode(val);
            return;
        }
        if(root->right->val<val){//當前結點的右結點小於給定的值,則將插入該值到右結點,並將原來右結點作為新節點的左結點插入
            TreeNode* tmp=new TreeNode(val);
            tmp->left=root->right;
            root->right=tmp;
            return ;
        }
        insert_node(root->right,val);//否則,繼續遍歷右結點
    }
    TreeNode* insertIntoMaxTree(TreeNode* root, int val) {
    	//處理根節點為空
        if(root==nullptr){
            return new TreeNode(val);
        }
        //處理根節點的值小於給定的值
        if(root->val<val){
            TreeNode* tmp=new TreeNode(val);
            tmp->left=root;
            return tmp;
        }
        //遞迴插入結點值
        insert_node(root,val);
        return root;
    }
};