leecode第一百一十四题（二叉树展开为链表）

Posted 2022-05-02 cjt-blog

/**
 * 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==NULL)//记住！root可能为[]
            return;
        
        stack<TreeNode*> temp;
        TreeNode* first=root;//记住上一个节点
        if(root->right!=NULL)
            temp.push(root->right);
        if(root->left!=NULL)
            temp.push(root->left);
        
        while(!temp.empty())
        
            TreeNode* node=temp.top();//前序遍历
            temp.pop();
            if(node->right!=NULL)
                temp.push(node->right);
            if(node->left!=NULL)
                temp.push(node->left);
            
            first->right=node;//同时原地记录顺序
            first->left=NULL;
            first=node;
        
    
;

分析：

思路很简单，就是前序遍历里加一点东西就好。