[leetcode]Flatten Binary Tree to Linked List

本文介绍了一种将二叉树结构通过非递归前序遍历的方式转换为链表的方法。此方法适用于原地操作,无需额外的空间。通过对二叉树节点的巧妙调整,确保每个节点的右子节点指向原二叉树前序遍历中的下一个节点。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

Flatten Binary Tree to Linked List

 

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

For example,
Given

         1
        / \
       2   5
      / \   \
     3   4   6

The flattened tree should look like:
   1
    \
     2
      \
       3
        \
         4
          \
           5
            \
             6


Hints:

If you notice carefully in the flattened tree, each node's right child points to the next node of a pre-order traversal.

解题思路:非递归前序遍历的应用改进,左右不空时压右

/**
 * 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 treeToLine(TreeNode *root, stack<TreeNode*> &stn){
        if(root->left == NULL && root->right == NULL){
            if(stn.empty()) return;
            root->right = stn.top(); //栈中还有结点时出栈
            stn.pop();
        }else if(root->left && root->right == NULL){
            root->right = root->left;
        }else if(root->left && root->right){
            stn.push(root->right); //左右都不空时,右子树进栈
            root->right = root->left;
        }
        root->left = NULL;
        treeToLine(root->right, stn);
    }
    void flatten(TreeNode *root) {
        // IMPORTANT: Please reset any member data you declared, as
        // the same Solution instance will be reused for each test case.
        if(NULL == root) return;
        stack<TreeNode*> stn;
        treeToLine(root, stn);
    }
};


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值