PigyChan_LeetCode 654. 最大二叉树

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;
    }
};