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]])
請注意,我們沒有直接給定 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)若不為空,則判讀當前結點的右結點的值是否小於給定的值,若小於,則將當前結點作為右結點插入,原來的右結點作為新的右結點的左結點插入;
/**
* 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;
}
};