在列表列表中找到具有最小大小的列表

Question

我有一个列表列表，我想使用 java Stream 返回具有最小大小的列表。

这是我尝试过的：

import java.util.ArrayList;
import java.util.Arrays;
import java.util.stream.Collectors;
import java.util.stream.IntStream;

public class Example {
   public static void main( String[] args )  {
         ArrayList<Integer> list1 = new ArrayList<>();
         list1.add(0);list1.add(2);list1.add(3);

         ArrayList<Integer> list2 = new ArrayList<>();
         list2.add(0);list2.add(2);list2.add(3);list2.add(4);

         System.out.println( getTheMinList(list1,list2).size());    
   }

   public static ArrayList<Integer> getTheMinList(ArrayList<Integer>... lists) {
         return Arrays.stream(lists)
                      .map(list->list.size())
                      /*here i stoped*/
                ;
   }
}

程序应该打印3，因为第一个列表的大小最小。

请注意，我无法更改 getTheMinList() 的签名。谁能给我一个提示？

Answer 1

只是另一个想法......

正如@NullPointer 建议的那样，我会让该方法采用List<List<Integer>> 而不是可变参数。

鉴于此更改，您只需执行以下操作：

Collections.min(lists, Comparator.comparingInt(List::size)); 

简短、简单且易读对吧？

请注意，上述内容将在集合为空时抛出NoSuchElementException，这也可能是您想要的预期行为，但以防万一这不是您可能想要在调用之前检查是否为空的预期行为min 或使用@NullPointer 建议的流方法，如果 Stream#min 返回空 Optional，则使用 Optional API 返回替代值。

Answer 2

你可以用Stream.min比较List.size：

public static List<Integer> getTheMinList(List<Integer>... lists){
    return Arrays.stream(lists)
            .min(Comparator.comparingInt(List::size))
            .orElse(new ArrayList<>());
}

或者如果你可以摆脱 var-args 和未经检查的赋值可能会更好

private static List<Integer> getTheMinList(List<List<Integer>> lists){
    return lists.stream()
            .min(Comparator.comparingInt(List::size))
            .orElse(new ArrayList<>());
}
// invoked as
System.out.println(getTheMinList(List.of(list1, list2)).size());