在 Java 中对 List 的特定列进行排序并查找元素_技术教程

的特定列进行排序并查找元素"> 的特定列进行排序并查找元素" />

本文介绍了如何在 Java 中对 List> 的指定列进行排序，并查找特定元素。通过自定义 Comparator，可以实现基于指定列的排序。同时，提供了一个查找特定元素索引的方法，并演示了如何利用该索引进行排序和元素查找。

对 List> 的特定列进行排序

在处理二维数据时，经常需要对特定列进行排序。Java 中的 List> 结构非常适合表示这种数据。以下代码展示了如何根据指定列的值对 List> 进行排序。

import java.util.*;
import java.util.stream.IntStream;

public class GridSort {
    public static void main(String[] args) {
     
   List> grid = new ArrayList<>();
        grid.add(List.of("Test0", "ABC", "123", "A1"));
        grid.add(List.of("Test3", "JKL", "901", "A4"));
        grid.add(List.of("Test1", "DEF", "345", "A2"));
        grid.add(List.of("Test4", "MNO", "234", "A5"));
        grid.add(List.of("Test2", "GHI", "678", "A3"));

        String key = "345";
        int pivotPoint = findPivotPoint(grid, key); // returns index 2

        // sorting
        Comparator> rowComparator = new Comparator>() {

            @Override
            public int compare(List o1, List o2) {
                if (pivotPoint < 0) {
                    return 0; // Skip comparison if pivotPoint is negative
                }
                String s1 = o1.get(pivotPoint);
                String s2 = o2.get(pivotPoint);
                return s1.compareTo(s2);
            }
        };

        if (pivotPoint >= 0) {
            Collections.sort(grid, rowComparator);
        }

        System.out.println("Pivot Point: " + pivotPoint);
        grid.forEach(System.out::println);
    }

    private static int findPivotPoint(List> grid, String key) {
        for (List list : grid) {
            OptionalInt indexOpt = IntStream.range(0, list.size())
                    .filter(i -> key.equals(list.get(i)))
                    .findFirst();
            if (indexOpt.isPresent()) {
                return indexOpt.getAsInt();
            }
        }
        return -1;
    }
}

代码解释:

数据初始化: 创建一个 List> 并填充数据。
findPivotPoint 方法: 该方法用于查找指定 key 在 grid 中的索引位置。如果找到匹配的 key，则返回其索引；否则返回 -1。
Comparator 实现: 创建一个自定义的 Comparator，用于比较 List 对象。compare 方法根据 pivotPoint 索引位置的值进行比较。如果 pivotPoint 小于 0，则跳过比较，直接返回0。
排序: 使用 Collections.sort 方法，传入 grid 和自定义的 Comparator，对 grid 进行排序。
输出: 打印排序后的 grid。

关键点

自定义 Comparator: Comparator 接口允许自定义排序规则。在本例中，我们根据指定列的值进行排序。
findPivotPoint 方法: 该方法简化了查找元素索引的过程。
异常处理: 如果 pivotPoint 小于 0，表示未找到指定元素，此时应该避免进行比较操作，以防止 IndexOutOfBoundsException。

注意事项

List> 必须是可变的，因为排序操作会修改列表的内容。
如果不需要修改内部的 List，可以使用 List.of() 创建不可变列表。如果尝试修改这些不可变列表，会抛出 UnsupportedOperationException。
在实际应用中，可以根据需要修改 Comparator 的实现，以满足不同的排序需求。例如，可以实现降序排序，或者根据多个列进行排序。