面试题 17.12. BiNode
Difficulty: 简单
二叉树数据结构TreeNode
可用来表示单向链表(其中left
置空,right
为下一个链表节点)。实现一个方法,把二叉搜索树转换为单向链表,要求依然符合二叉搜索树的性质,转换操作应是原址的,也就是在原始的二叉搜索树上直接修改。
返回转换后的单向链表的头节点。
**注意:**本题相对原题稍作改动
示例:
输入: [4,2,5,1,3,null,6,0]
输出: [0,null,1,null,2,null,3,null,4,null,5,null,6]
提示:
- 节点数量不会超过 100000。
Solution
Language: 全部题目
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public TreeNode convertBiNode(TreeNode root) {
Deque<TreeNode> stack = new LinkedList<>();
while(root !=null) {
stack.push(root);
root = root.left;
}
// TreeNode head = new TreeNode(0);
TreeNode prev = stack.peek();
while(stack.isEmpty() ==false) {
TreeNode cur =stack.pop();
root = cur.right;
while (root != null) {
stack.push(root);
root = root.left;
}
// TreeNode cur = stack.pop();
cur.left = null;
if (stack.isEmpty() ==false) {
cur.right = stack.peek();
}
}
return prev;
}
}