标签:right TreeNode val nums null root leetcode
二叉搜索树中的两个节点被错误地交换。
请在不改变其结构的情况下,恢复这棵树。
示例 1:
输入: [1,3,null,null,2]
1
/
3
\
2
输出: [3,1,null,null,2]
3
/
1
\
2
示例 2:
输入: [3,1,4,null,null,2]
3
/ \
1 4
/
2
输出: [2,1,4,null,null,3]
2
/ \
1 4
/
3
/** * 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 read_tree(vector<int> &nums,TreeNode* root) { if(root ==NULL) return; read_tree(nums,root->left); nums.push_back(root->val); read_tree(nums,root->right); } pair<int,int> find_swap(vector<int> &nums) { int n=nums.size(); int x=-1,y=-1; for(int i=0;i<n-1;i++) { if(nums[i+1]<nums[i]) { y=nums[i+1]; if(x==-1) { x=nums[i]; } else break; } } return {x,y}; } void change(pair<int,int> &s,TreeNode* root,int remain) { if(root!=nullptr) { if(root->val==s.first || root->val==s.second) { root->val = root->val==s.first ? s.second:s.first; if(--remain == 0) return; } change(s,root->left,remain); change(s,root->right,remain); } } void recoverTree(TreeNode* root) { vector <int> nums; read_tree(nums,root); pair<int,int> s=find_swap(nums); change(s,root,2); } };
注意:二叉搜索树(二叉排序树)的中序遍历是从小到大的。因此只要找出中序遍历中不满足从小到大的两个点,然后进行交换就可以得到相应的结果。
标签:right,TreeNode,val,nums,null,root,leetcode 来源: https://www.cnblogs.com/zmachine/p/13458446.html
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。