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

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

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

下面贴一下代码实现：

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

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