腾讯笔试4.26 二维top k

public static void main(String[] args) {
        List<Integer> list = new ArrayList<>();
        List<Integer> lenlist = new ArrayList<>();
        Scanner sc = new Scanner(System.in);
        int N = sc.nextInt();
        int K = sc.nextInt();
        for (int i=0;i<N;i++){
            int len = sc.nextInt();
            lenlist.add(len);
            for (int j=0;j<len;j++){
                list.add(sc.nextInt());
            }
        }
        list.sort(new Comparator<Integer>() { @Override public int compare(Integer o1, Integer o2) {
                return o2.compareTo(o1);
            }
        });
        for (int i =0;i<K;i++)
            System.out.print(list.get(i)+" ");
        sc.close();
    }