码迷,mamicode.com
首页 > 其他好文 > 详细

LeetCode (15) Flatten Binary Tree to Linked List

时间:2015-04-18 23:48:59      阅读:213      评论:0      收藏:0      [点我收藏+]

标签:c++   leetcode   算法   二叉树   

题目描述

Given a binary tree, flatten it to a linked list in-place.

For example,
Given
技术分享
The flattened tree should look like:
技术分享

本题也是考察二叉树和指针操作的题目。题目要求将一棵二叉树拉平为一个链表 。链表通过树节点的右子树相连,且展开的顺序为原来树的前序遍历。

实现思路:

  • 若节点n存在左子树left,则将左子树连接为n的右子树
  • 若n存在右子树(n->right存在),则将n->right链接为n->left的最右下方。相当于把n节点 的整个左子树插入了n与n->right之间。
  • 对上述处理后的n->right重复上述操作,直到达到叶节点。

需要注意的是,对处理后的左子树指针需要设置为NULL。

/**
 * Definition for binary tree
 * 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)
            return;
        if (root->left != NULL)
        {
            TreeNode *left = root->left;
            while(left->right)
            {
                left = left->right;
            }
            left->right = root->right;
            root->right = root->left;
            root->left = NULL;

        }

        flatten(root->right);
    }
};

LeetCode (15) Flatten Binary Tree to Linked List

标签:c++   leetcode   算法   二叉树   

原文地址:http://blog.csdn.net/angelazy/article/details/45117275

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!