654. 最大二叉树
难度中等
给定一个不含重复元素的整数数组。一个以此数组构建的最大二叉树定义如下:
(1)二叉树的根是数组中的最大元素。
(2)左子树是通过数组中最大值左边部分构造出的最大二叉树。
(3)右子树是通过数组中最大值右边部分构造出的最大二叉树。
通过给定的数组构建最大二叉树,并且输出这个树的根节点。
示例 :
提示:
(1)给定的数组的大小在 [1, 1000] 之间。
思路1.0:
(1)创建一个结构体存储数组arr中元素的值和下标,再将其按值大小降序排序,这个问题就变成了向一个二叉搜索树依次添加元素了。
(2)一开始在二叉树中存的都是下标,要再遍历一次换上值。
代码1.0:
struct TreeNode {
int val;
TreeNode *left;
TreeNode *right;
TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};
struct idxVal
{
int val;
int index;
idxVal(int x, int y) :val(x), index(y) {}
};
class Solution {
public:
TreeNode* insertValue(TreeNode* TNode, int value)
{
if (TNode != NULL)
{
if (value > TNode->val) {
TNode->right = insertValue(TNode->right, value);
}
else {
TNode->left = insertValue(TNode->left, value);
}
}
//找到正确位置,插入value
else {
TreeNode* newNode = new TreeNode(value);
return newNode;
/*cout << "Insert value" << value << endl;*/
}
return TNode;
}
void PreOrder(TreeNode* TNode,const vector<int>& nums)
{
if (TNode != NULL)
{
TNode->val = nums[TNode->val];
PreOrder(TNode->left,nums);
PreOrder(TNode->right, nums);
}
}
TreeNode* constructMaximumBinaryTree(vector<int>& nums) {
int len = nums.size();
//初始化vec
vector<idxVal> vec;
for (int i = 0; i < len; ++i)
{
vec.push_back(idxVal(nums[i], i));
}
sort(vec.begin(), vec.end(), [](idxVal a, idxVal b) {return a.val > b.val; });
//问题转变为向搜索二叉树插入元素
TreeNode* root = new TreeNode(vec[0].index);
for (int i = 1; i < len; ++i)
{
insertValue(root, vec[i].index);
}
//此时该二叉树中存的都是下标,我们再遍历一次这树,用值替换下标
PreOrder(root, nums);
return root;
}
};