面试题 17.12. BiNode
二叉树数据结构TreeNode可用来表示单向链表(其中left置空,right为下一个链表节点)。实现一个方法,把二叉搜索树转换为单向链表,要求依然符合二叉搜索树的性质,转换操作应是原址的,也就是在原始的二叉搜索树上直接修改。
返回转换后的单向链表的头节点。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/binode-lcci
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
题解:使用中序遍历,记录前一个结点,遍历到每个节点时,将left置空,将前一个节点的right指向当前节点。二叉搜索树中序遍历最后一个节点的right必然为空
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
TreeNode ans;
TreeNode *pre;
public:
Solution() {
pre = &ans;
}
~Solution() {
}
void inOrder(TreeNode* root) {
if (root == NULL) {
return;
}
inOrder(root->left);
root->left = NULL;
if (pre->right != root) {
pre->right = root;
}
pre = root;
inOrder(root->right);
}
TreeNode* convertBiNode(TreeNode* root) {
inOrder(root);
TreeNode *head = ans.right;
ans.right = NULL;
pre = NULL;
return head;
}
};