标签:right TreeNode val int 二叉 99 root LeetCode left
题目描述链接:https://leetcode-cn.com/problems/recover-binary-search-tree/
解题思路:中序遍历二叉树,找出不符合条件的两个节点,然后重新遍历进行恢复
LeetCode C++求解代码:
/** * 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 recoverTree(TreeNode* root) { stack<TreeNode *>S; TreeNode* p=root; TreeNode* q; vector<int>res; while(!S.empty()||p){ if(p){ S.push(p); p=p->left; } else{ q=S.top(); S.pop(); res.push_back(q->val); p=q->right; } } int x=-1,y=-1; for(int i=0;i<res.size()-1;++i){ if(res[i+1]<res[i]){ y=res[i+1]; if(x==-1){ x=res[i]; } else{ break; } } } recover(root,2,x,y); } void recover(TreeNode *root,int count,int x,int y){ if(root){ if(root->val==x){ root->val=y; count--; } else if(root->val==y){ root->val=x; count--; } if(count==0){ return ; } recover(root->left,count,x,y); recover(root->right,count,x,y); } } };
时间复杂度:O(n) 空间复杂度O(n)
标签:right,TreeNode,val,int,二叉,99,root,LeetCode,left 来源: https://www.cnblogs.com/zzw-/p/13460240.html
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。