剑指offer 二叉搜索树与双向链表 C++

题目描述

输入一棵二叉搜索树，将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点，只能调整树中结点指针的指向。

思路

本题和leetcode14有相似之处。
采用树的中序历遍的思想。

解答

/*
struct TreeNode {
	int val;
	struct TreeNode *left;
	struct TreeNode *right;
	TreeNode(int x) :
			val(x), left(NULL), right(NULL) {
	}
};*/
class Solution {
public:
    TreeNode* Convert(TreeNode* pRootOfTree)
    {
        if(NULL==pRootOfTree) return pRootOfTree;
        if(pRootOfTree->left==NULL && pRootOfTree->right==NULL) return pRootOfTree;
        TreeNode* left=Convert(pRootOfTree->left);
        TreeNode* p=left;
        while(p && p->right) p=p->right;
        if(left)
        {
            p->right=pRootOfTree;
            pRootOfTree->left=p;
        }
        TreeNode* right=Convert(pRootOfTree->right);
        if(right)
        {
            pRootOfTree->right=right;
            right->left=pRootOfTree;
        }
        return left? left:pRootOfTree;
    }
};