在一组arraylist

Question

我有一个存储在另一个arraylist中的整数列表。

List<List<Integer>> h = new ArrayList<List<Integer>>();
ArrayList<Integer> temp = new ArrayList<Integer>();
temp.add(x1);
temp.add(x2);
temp.add(y);
h.add(temp);

我想根据该数组的第一个数字对其进行排序。

input
(2 4 3)
(1 3 2)
(3 4 5)

expected output
(1 3 2)
(2 4 3)
(3 4 5)

我尝试过使用 collections.sort ，但似乎无效。 有没有其他方法可以尝试对其进行排序？

Answer 1

您希望自定义比较器比较整数列表的列表。

public class ListComparator implements Comparator<List<Integer>> {
    @Override
    public int compare(List<Integer> list1, List<Integer> list2) {
        int value1 = list1.get(0);
        int value2 = list2.get(0);
        return Integer.compare(value1, value2);
    }
}

然后将其作为第二个参数传递给sort。

    Collections.sort(demoList, new ListComparator());

编辑 - 这是一个处理正在排序的列表中的空列表和空列表的版本。

public class ListComparator implements Comparator<List<Integer>> {
    @Override
    public int compare(List<Integer> list1, List<Integer> list2) {
        int result = 0;
        if (list1 == null) {
            // define null as equal to null and less than everything else
            result = (list2 == null) ? 0 : -1;
        }
        else if (list1.isEmpty()) {
            // define empty as greater than null, equal to empty, and less than non-empty
            result = (list2 == null) ? 1 : (list2.isEmpty() ? 0 : -1);
        }
        else if (list2 == null || list2.isEmpty()) {
            // define non-empty (list1) as greater than null or empty
            result = 1;
        }
        else {
            // both are non-empty so compare the first values from each
            int value1 = list1.get(0);
            int value2 = list2.get(0);
            result = Integer.compare(value1, value2);
        }
        return result;
    }
}