【剑指Offer】36.二叉搜索树与双向链表

题目

输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。如下图所示

数据范围:输入二叉树的节点数 0≤n≤1000,二叉树中每个节点的值 0≤val≤1000

要求:O(1)(即在原树上操作),时间复杂度 O(n)

注意:

1.要求不能创建任何新的结点,只能调整树中结点指针的指向。当转化完成以后,树中节点的左指针需要指向前驱,树中节点的右指针需要指向后继

2.返回链表中的第一个节点的指针

3.函数返回的TreeNode,有左右指针,其实可以看成一个双向链表的数据结构

4.你不用输出双向链表,程序会根据你的返回值自动打印输出

输入描述:

二叉树的根节点

返回值描述:

双向链表的其中一个头节点。

示例1

输入:{10,6,14,4,8,12,16}

返回值:From left to right are:4,6,8,10,12,14,16;From right to left are:16,14,12,10,8,6,4;

说明:输入题面图中二叉树,输出的时候将双向链表的头节点返回即可。

示例2

输入:{5,4,#,3,#,2,#,1}

返回值:From left to right are:1,2,3,4,5;From right to left are:5,4,3,2,1;

说明:

复制代码
                    5
                  /
                4
              /
            3
          /
        2
      /
    1
树的形状如上图

解答

源代码

java 复制代码
import java.util.*;
/**
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;

    }

}
*/
public class Solution {
    public TreeNode pre = null;
    public TreeNode head = null;
    public TreeNode Convert(TreeNode pRootOfTree) {
        if (pRootOfTree == null) {
            return null;
        }

        Convert(pRootOfTree.left);
        if (pre == null){
            pre = pRootOfTree;
            head = pRootOfTree;
        } else {
            pre.right = pRootOfTree;
            pRootOfTree.left = pre;
            pre = pRootOfTree;
        }
        Convert(pRootOfTree.right);

        return head;
    }
}

总结

二叉搜索树按中序遍历就可以得到一个从小到大排列的序列,因此可以通过对二叉树进行中序遍历来得到双向链表。head保存头结点,pre保存上一个结点,连接好当前节点和pre后,将pre设置为当前结点。

相关推荐
xiaolingting1 小时前
Java 二叉树非递归遍历核心实现
java··二叉树非递归遍历
嘵奇1 小时前
深入解析 Java 8 Function 接口:函数式编程的核心工具
java·开发语言
GISer_Jing3 小时前
前端算法实战:大小堆原理与应用详解(React中优先队列实现|求前K个最大数/高频元素)
前端·算法·react.js
一路向北North3 小时前
IDEA加载项目时依赖无法更新
java·ide·intellij-idea
小森77673 小时前
(三)机器学习---线性回归及其Python实现
人工智能·python·算法·机器学习·回归·线性回归
振鹏Dong4 小时前
超大规模数据场景(思路)——面试高频算法题目
算法·面试
uhakadotcom4 小时前
Python 与 ClickHouse Connect 集成:基础知识和实践
算法·面试·github
uhakadotcom4 小时前
Python 量化计算入门:基础库和实用案例
后端·算法·面试
小萌新上大分4 小时前
SpringCloudGateWay
java·开发语言·后端·springcloud·springgateway·cloudalibaba·gateway网关
uhakadotcom4 小时前
使用 Python 与 BigQuery 进行交互:基础知识与实践
算法·面试