摺疊2D ArrayList

Question

我有一個對象，其中包含一個動態分配的2D ArrayList。

它有一個getter：ArrayList<Integer> getList()

我想一個函數來摺疊2D ArrayList成一維ArrayList。確實有Java的這種事還是我堅持做這樣的：

public ArrayList<Integer> getList(){ 
    ArrayList<Integer> result = new ArrayList<Integer>(); 

    for(int i = 0; i < list_.size(); i++){ 
     for(int j = 0; j < list_.get(i).size(); j++){ 
      result.add(list_.get(i).get(j)); 
     } 
    } 
    return result; 
} 

Answer 1

沒有Java 8，你必須使用循環。使用addAll(..)確實擺脫其中之一。

public ArrayList<Integer> getList(){ 
    ArrayList<Integer> result = new ArrayList<Integer>(); 

    for(ArrayList<Integer> i : list_){ 
     result.addAll(i); 
    } 
    return result; 
} 

此外，通用類型如ArrayList必須是對象類型，因此Integer，而不是int。

Answer 2

與Java 8：

return _list.stream() 
      .flatMap(List::stream) 
      .collect(Collectors.toList()); 

的Java 8之前，需要一個循環。