代码随想录算法训练营第36期DAY25

DAY25

669修剪二叉搜索树

1. /**
2.  * Definition for a binary tree node.
3.  * struct TreeNode {
4.  *     int val;
5.  *     TreeNode *left;
6.  *     TreeNode *right;
7.  *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
8.  *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
9.  *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
10.  * };
11.  */
12. class Solution {
13. public:
14.     TreeNode* fdr(TreeNode* root,int low,int high)
15.     {
16.         if(root==nullptr) return nullptr;
17.         if(root->val<low){
18.             TreeNode* right=fdr(root->right,low,high);
19.             return right;
20.         }
21.         if(root->val>high){
22.             TreeNode* left=fdr(root->left,low,high);
23.             return left;
24.         }
25.         root->left=fdr(root->left,low,high);
26.         root->right=fdr(root->right,low,high);
27.         return root;
28.     }
29.     TreeNode* trimBST(TreeNode* root, int low, int high) {
30.         return fdr(root,low,high);
31.     }
32. };

108将有序数组转为二叉搜索树

有序数组构造二叉树：从数组中间位置取值作为节点元素，就是平衡的了。

根据数组构造二叉树，其本质是寻找分割点，然后递归左区间和右区间。

二分法mid=(left+right)/2可能会爆INT，写成mid=left+(right-left)/2【相当于：偶数取左边】;

1. /**
2.  * Definition for a binary tree node.
3.  * struct TreeNode {
4.  *     int val;
5.  *     TreeNode *left;
6.  *     TreeNode *right;
7.  *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
8.  *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
9.  *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
10.  * };
11.  */
12. class Solution {
13. public:
14.   TreeNode* getr(vector<int> &nums,int left,int right)
15.     {
16.         if(left>right){
17.             return nullptr;
18.         }
19.         int mid=left+(right-left)/2;
20.         TreeNode* root=new TreeNode(nums[mid]);
21.         root->left=getr(nums,left,mid-1);
22.         root->right=getr(nums,mid+1,right);
23.         return root;
24.     }
25.     TreeNode* sortedArrayToBST(vector<int>& nums) {
26.         return getr(nums,0,nums.size()-1);
27.     }
28. };

538把二叉搜索树转换为累加树

1. /**
2.  * Definition for a binary tree node.
3.  * struct TreeNode {
4.  *     int val;
5.  *     TreeNode *left;
6.  *     TreeNode *right;
7.  *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
8.  *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
9.  *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
10.  * };
11.  */
12. class Solution {
13. private:
14.     int pre=0;
15.     void GS(TreeNode* root)
16.     {
17.         if(root==nullptr) return;
18.         GS(root->right);
19.         root->val+=pre;
20.         pre=root->val;
21.         GS(root->left);
22.     }
23. public:
24.     TreeNode* convertBST(TreeNode* root) {
25.         GS(root);
26.         return root;
27.     }
28. };