现在的位置: 首页 > 综合 > 正文

LeetCode – Flatten Binary Tree to Linked List (Java)

2018年04月09日 ⁄ 综合 ⁄ 共 583字 ⁄ 字号 评论关闭

这题参考了Code Ganker大牛思路

http://blog.csdn.net/linhuanmars/article/details/23717703

设置pre结点,每次pre左结点设置为空,右结点设置为当前结点。有一点特别要注意的是Java是pass by value,所以需要用一个单元素的TreeNode数组来储存并且改值。

  1. ArrayList<TreeNode> pre = new ArrayList<TreeNode>();
  2. public void flatten(TreeNode root) {
     
      
  3.     pre.add(null);  
  4.     helper(root);  
  5. }  
  6. private void helper(TreeNode root)  
  7. {  
  8.     if(root == null)  
  9.         return;  
  10.     TreeNode right = root.right;  
  11.     if(pre.get(0)!=null)  
  12.     {  
  13.         pre.get(0).left = null;  
  14.         pre.get(0).right = root;  
  15.     }  
  16.     pre.set(0,root);  
  17.     helper(root.left);  
  18.     helper(right);  
  19. }

抱歉!评论已关闭.