為什么我的方法打印一個空列表？

Question

我正在嘗試在ArrayList<ArrayList<Integer>> result逐級打印樹。 result每個列表均視為其自己的級別。

例：

         1                
        / \             
       2   3   
      / \ / \           
     4  5 6  7
==>  [1][2, 3][4, 5, 6, 7]

出於某種原因，我不斷返回一個空列表。 這是我的代碼：

public ArrayList<ArrayList<Integer>> printLevelByLevel(TreeNode root) {
  ArrayList<ArrayList<Integer>> result = new ArrayList<ArrayList<Integer>>();
  ArrayList<Integer> temp = new ArrayList<Integer>();
  if(root == null) return result;
  int levelCount = 0;
  Queue<TreeNode> q = new LinkedList<TreeNode>();
  q.add(root);
  while(true){
    levelCount = q.size();
    if(levelCount == 0) break;
    while(levelCount > 0){
        TreeNode curr = q.poll();
        temp.add(curr.data);
        if(curr.left != null){
            q.add(curr.left);
        }
        if(curr.right != null){
            q.add(curr.right);
        }
        levelCount--;
    } // end of inner while 
    result.add(temp);
    temp.clear();
  } // end of outter while loop 

 return result;
}

我的temp.clear（）對嗎？ 我嘗試將其放在不同的位置，但結果仍然相同。 我知道我可以用兩個做這個Queues ，但是我希望能夠有一個做到這一點Queue 。

謝謝。

Answer 1

將相同的ArrayList實例（由temp變量引用）多次添加到reuslt列表中是錯誤的，因為最后result將包含多個空列表（或更准確地說，是對同一空列表的多個引用）。

您應該在每次迭代中創建一個新實例，而不是通過temp清除單個實例引用：

while(true){
    levelCount = q.size();
    if(levelCount == 0) break;
    ArrayList<Integer> temp = new ArrayList<Integer>();
    while(levelCount > 0){
        TreeNode curr = q.poll();
        temp.add(curr.data);
        if(curr.left != null){
            q.add(curr.left);
        }
        if(curr.right != null){
            q.add(curr.right);
        }
        levelCount--;
    } // end of inner while 
    result.add(temp);
}