【LeetCode刷题（简单程度）】面试题 17.12. BiNode

二叉树数据结构TreeNode可用来表示单向链表（其中left置空，right为下一个链表节点）。实现一个方法，把二叉搜索树转换为单向链表，要求依然符合二叉搜索树的性质，转换操作应是原址的，也就是在原始的二叉搜索树上直接修改。

返回转换后的单向链表的头节点。

注意：本题相对原题稍作改动

示例：

输入： [4,2,5,1,3,null,6,0]
输出： [0,null,1,null,2,null,3,null,4,null,5,null,6]
提示：

节点数量不会超过 100000。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/binode-lcci
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路1：利用中序遍历（递归版）实现（DFS），其主要的操作就是visit操作中要完成单链表的实现，即将左子树置空，前一节点的右子树连接到当前节点，所以这里引入一个哨兵节点。

/**
 * 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 {
public:
    TreeNode* dmmyHead = new TreeNode();//哨兵节点
    TreeNode* pre = dmmyHead;
    TreeNode* convertBiNode(TreeNode* root) {
        dfs(root);
        return dmmyHead->right;
    }

    void dfs(TreeNode* root)
    {
        if(root)
        {
            dfs(root->left);
            visit(root);
            dfs(root->right);
        }
    }
    void visit(TreeNode* root)//主要在于中序遍历访问当前节点的时候完成链表的构造
    {
        root->left = NULL;
        pre->right = root;
        pre = root;
    }
};

思路2：大体思路一致， 只是将递归版转换成迭代版。

class Solution {
public:
    TreeNode* dmmyHead = new TreeNode();//哨兵节点
    TreeNode* pre = dmmyHead;
    TreeNode* convertBiNode(TreeNode* root) {
        dfs(root);
        return dmmyHead->right;
    }

    void dfs(TreeNode* root)
    {
        stack<TreeNode*> s;
        while(true)
        {
            goLeftLongBrach(root, s);
            if(s.empty())//说明访问完毕
                break;
            root = s.top();
            s.pop();
            visit(root);
            root = root->right;
        }
    }
    void visit(TreeNode* root)//主要在于中序遍历访问当前节点的时候完成链表的构造
    {
        root->left = NULL;
        pre->right = root;
        pre = root;
    }

    static void goLeftLongBrach(TreeNode* cur,stack<TreeNode*>& s)
    {
        while(cur)
        {
            s.push(cur);
            cur = cur->left;
        }
    }
};

迭代版：

/**
 * 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 {
public:
    TreeNode* dmmyHead = new TreeNode();//哨兵节点
    TreeNode* pre = dmmyHead;
    TreeNode* convertBiNode(TreeNode* root) {
        inOrder(root);
        return dmmyHead->right;
    }
private:
    void inOrder(TreeNode* root)
    {
        stack<TreeNode*> s;
        TreeNode* p = root;
        while(p != nullptr || !s.empty())
        {
            while(p!=nullptr)
            {
                s.push(p);
                p = p->left;//尽量往左
            }
            if(!s.empty())
            {
                p = s.top();
                visit(p);
                s.pop();
                p = p->right;//有右子树
            }
        }
    }

    void visit(TreeNode* root)//主要在于中序遍历访问当前节点的时候完成链表的构造
    {
        root->left = NULL;
        pre->right = root;
        pre = root;
    }
};