114. Flatten Binary Tree to Linked List

来源:互联网 发布:单片机 fc音效 编辑:程序博客网 时间:2024/04/29 01:13

一次AC,前序遍历再重构

/** * 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 preOrder(TreeNode* root,queue<TreeNode*>& nodes)    {        if(root==NULL)            return;        else        {            nodes.push(root);            if(root->left!=NULL)                preOrder(root->left,nodes);            if(root->right!=NULL)                preOrder(root->right,nodes);            return;        }    }    void flatten(TreeNode* root) {        if(root==NULL)            return;        queue<TreeNode*> nodes;        preOrder(root,nodes);        TreeNode* preRoot=nodes.front();        nodes.pop();        while(!nodes.empty())        {            preRoot->left=NULL;            preRoot->right=nodes.front();            preRoot=nodes.front();            nodes.pop();        }        preRoot->left=NULL;        preRoot->right=NULL;        return;    }};
0 0
原创粉丝点击