算法基础 - 二叉查找树变成有序双向链表

自己今天真是笨死了，这么简单的题目，竟然在当场写代码的时候没有写出来。

这里写一下，其实这个题目非常简单，就是一个中序遍历，在中序遍历的时候，更改一下指针指向。自己在面试写的时候太紧张了，就蒙了。。。

下面贴一下代码实现：

TreeNode* inorderFlatTree(TreeNode * root){    stack<TreeNode *> st;    TreeNode * temp = root;    TreeNode * sw = NULL;    while (temp != NULL || !st.empty()) {        while (temp != NULL) {            st.push(temp);            temp = temp->left;        }        if (!st.empty()) {            if (sw != NULL) {                sw->right = st.top();                st.top()->left = sw;            }            cout<<st.top()->val<<endl;            temp = st.top()->right;            sw = st.top();            st.pop();        }    }    return sw;}

这样子返回的是链表的尾部，如果需要增序的话，用个while循环，把sw指向头结点就可以了。

while(sw != NULL){sw = sw->left;}

就这个样子。

递归实现

其实递归实现也非常简单，只用记录上一次的节点指针就行了。

void flatTree(TreeNode * root, TreeNode * head){    if (root == NULL) {        return;    }    flatTree(root->left, head);    if (head == NULL) {        head = root;    }else{        head->right = root;        root->left = head;        head = root;    }    flatTree(root->right, head);}

自己对代码掌控还是不够熟练的收放自如，还是需要不断的努力啊。