114.二叉树展开为链表

题目描述：

给定一个二叉树，原地将它展开为链表。

例如，给定二叉树

将其展开为：

思想：

1. 将原来的右子树接到左子树的最右边节点
2. 将左子树插入到根节点右子树的地方
3. 考虑新的右子树的根节点，一直重复上边的过程，直到新的右子树根节点为 null

代码：

/**
 * 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:
    void flatten(TreeNode* root) {
        if(root==nullptr)   //不要忘记判断root为空的情况
            return ;
        while(root){
            if(root->left!=nullptr){
                TreeNode* p = root->left;
                while(p->right!=nullptr)    //找到root左子树的最右结点
                    p = p->right;
                p->right = root->right;    //将root的右子树移到root左子树最右结点下
                root->right = root->left;  //将root的左子树赋值为右子树
                root->left = nullptr;   //不要忘记把root左子树赋值指针赋值为空指针
            }
            root = root->right;
        }
        return ;
    }
};