1. 程式人生 > 實用技巧 >leetcode 6

leetcode 6

二叉搜尋樹中的兩個節點被錯誤地交換。

請在不改變其結構的情況下,恢復這棵樹。

示例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); } };

注意:二叉搜尋樹(二叉排序樹)的中序遍歷是從小到大的。因此只要找出中序遍歷中不滿足從小到大的兩個點,然後進行交換就可以得到相應的結果。