leetcode114- 二叉树展开为链表

/*
 * @lc app=leetcode.cn id=114 lang=cpp
 *
 * [114] 二叉树展开为链表
 */

// @lc code=start
/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:

    //题解1：前序遍历将二叉树转换为链表，
    // void flatten(TreeNode* root) {
    //     root=build(root);
    // }
    // TreeNode* build(TreeNode* root){
    //     if(root==nullptr) return nullptr;
    //     TreeNode* head=new TreeNode(0);
    //     stack<TreeNode*> st;
    //     st.push(root);
    //     while(!st.empty()){
    //        TreeNode* t=st.top();
    //         head->right=t;
    //         st.pop();
    //         head->left=nullptr;
    //         if(root->right) st.push(t->right);
    //         if(root->left) st.push(t->left);
    //         head=head->right;
    //     }
    //     return head->right;
    // }

    ////题解2：将右子树连接到左子树的最右边节点上，然后左子树移动到原来右子树的地方，最后将根节点的左子树赋为空
    void flatten(TreeNode* root) {
        while(root!=nullptr){
            if(root->left==nullptr)//左子树为空， 考虑下一个节点
                root=root->right;
            else{
                TreeNode* pre=root->left;
                while(pre->right!=nullptr)// 寻找左子树最右边的节点
                    pre=pre->right;
                pre->right=root->right;//右子树连接到左子树的最右边节点上
                root->right=root->left;// 更新根节点的右子树
                root->left=nullptr;//左子树赋为空
                root=root->right;// 考虑下一个节点
            }
        }
    }
};
// @lc code=end