每天一道剑指offer-二叉搜索树与双向链表
2018-12-17 本文已影响0人
程序员乔戈里
前言
今天的题目
每天的题目见github(看最新的日期):
https://github.com/gzc426
具体的题目可以去牛客网对应专题去找。
昨天的题解
题目
每天一道剑指offer-二叉搜索树与双向链表
来源:
https://www.nowcoder.com/practice/947f6eb80d944a84850b0538bf0ec3a5?tpId=13&tqId=11179&tPage=2&rp=2&ru=/ta/coding-interviews&qru=/ta/coding-interviews/question-ranking
题目详述
输入一棵二叉搜索树,将该二叉搜索树转换成一个排序的双向链表。要求不能创建任何新的结点,只能调整树中结点指针的指向。
题目详解
思路
-
先中序遍历二叉搜索树,这样二叉搜索树就按照val值的大小从小到大排好序了,存放在数组中
-
然后要转换为双向链表,由于数组中的存放的树的节点已经按照键值从小到大排好序了,那么就对于每个节点的左子树指向数组的上一个节点,右子树指向数组的下一个节点,这样就完成了变成双向链表。
代码
public class Solution {
public TreeNode Convert(TreeNode pRootOfTree) {
if(pRootOfTree == null || (pRootOfTree.left == null && pRootOfTree.right == null))
return pRootOfTree;
ArrayList<TreeNode> nodeList = new ArrayList<>();
BuildArrayList(pRootOfTree,nodeList);//这个函数执行后,数组中每个元素按照大小前后排序
for(int i=0;i<nodeList.size();i++)
{
if(i == 0)
{//数组的第一个节点处理,只有右子树指向下一个节点
nodeList.get(0).right = nodeList.get(1);
}
else if(i == nodeList.size()-1)
{//数组的最后一个节点,只有左子树指向前一个节点
nodeList.get(i).left = nodeList.get(i-1);
}
else{//数组中的中间节点,左子树指向上一个节点,右子树指向数组的下一个节点
nodeList.get(i).left = nodeList.get(i-1);
nodeList.get(i).right = nodeList.get(i+1);
}
}
return nodeList.get(0);
}
public void BuildArrayList(TreeNode root,ArrayList<TreeNode> nodeList)
{//二叉搜索的中序遍历,并把每个节点存入数组中
if(root == null)
return;
if(root.left != null)//左子树
BuildArrayList(root.left,nodeList);
if(root != null)//根节点
nodeList.add(root);
if(root.right != null)//右子树
BuildArrayList(root.right,nodeList);
}
}
代码截图(避免乱码)
结束语
作者乔戈里亲历2019秋招,哈工大计算机本硕,百度java工程师,欢迎大家关注我的微信公众号:程序员乔戈里,公众号有3T编程资源,以及我和我朋友(百度C++工程师)在秋招期间整理的近200M的面试必考的java与C++面经,并有每天一道leetcode打卡群与技术交流群,欢迎关注。