力扣|将树展开成单链表结构(C++实现)

思路一:使用额外的数组,前向递归数组记录树的节点。然后再将左节点置nullptr,右节点依次更新。

class Solution {
public:
    vector<int> v;
    void flatten(TreeNode* root) {
        if(!root) return;
        preorder(root);
        TreeNode* pre = root;
        root->left = nullptr;
        root->right = nullptr;
        for(int i = 1; i < v.size(); i++){
            TreeNode* p = new TreeNode(v[i]);
            pre->right = p;
            pre = p;
        }
    }
    void preorder(TreeNode* root){
        if(!root) return;
        v.push_back(root->val);
        preorder(root->left);
        preorder(root->right);
    }
};

思路二:利用前驱节点:

树展开为链表实质是找到左子树的最右节点,然后将右子树接过去。最后实现左子树全部变成空。

class Solution {
public:
    vector<int> v;
    void flatten(TreeNode* root) {
     TreeNode* cur=root;
     while(cur!=nullptr)
   {
      if(cur->left!=nullptr)
      {
        auto next=cur->left;
        auto temp=next;
     while(temp->right!=nullptr)
          {temp=temp->right; }//找到左子树的最右节点 
      temp->right=cur->right;  //把当前节点的右子树接在左子树的最右一个点的下面
      cur->left=nullptr;   //当前节点的左子树设置成nullptr
      cur->right=next;  //将左子树挪到当前节点的右子树上
      }
      cur=cur->right;
   }
 
  }
};

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值