在 Java 中对 List<List> 的指定列进行排序和查找-小浪学习网

在 Java 中对 List<List> 的指定列进行排序和查找的指定列进行排序和查找” />

本文将详细介绍如何在 Java 中处理 List<List<String>> 类型的数据，并实现以下功能：对指定列进行排序，在排序后的列表中使用二分查找（或类似方法）查找特定元素，并输出包含该元素的完整行。

问题背景

在实际开发中，我们经常会遇到需要处理二维数据的情况，例如从 CSV 文件读取的数据或者从数据库查询的结果。List<List<String>> 是一种常见的表示二维字符串数据的方式。对这种数据进行排序和查找是常见的需求，例如对表格数据按某一列排序，然后查找特定记录。

解决方案

以下代码提供了一种解决方案，该方案使用 Comparator 接口自定义排序规则，并使用 IntStream 高效查找目标元素。

import java.util.ArrayList; import java.util.Collections; import java.util.Comparator; import java.util.List; import java.util.OptionalInt; import java.util.stream.IntStream;  public class Gridsort {     public static void main(String[] args) {         List<List<String>> 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); // 返回索引 2          // 排序         Comparator<List<String>> rowComparator = new Comparator<List<String>>() {             @Override             public int compare(List<String> o1, List<String> o2) {                 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<List<String>> grid, String key) {         for (List<String> 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;     } }

代码解释：

立即学习“Java免费学习笔记（深入）”；

数据初始化： 创建一个 List<List<String>> 对象 grid，并添加一些示例数据。注意：grid 必须是可变的，因为需要对其进行排序。可以使用 new ArrayList<>() 创建。
查找基准点（findPivotPoint 方法）： 该方法用于在 grid 中查找包含目标字符串 key 的列的索引。它遍历 grid 中的每一行，并使用 IntStream 查找包含 key 的元素的索引。如果找到，则返回该索引；否则，返回 -1。
自定义排序规则（Comparator）： 创建一个 Comparator<List<String>> 对象 rowComparator，用于定义排序规则。该 Comparator 比较两行数据在 pivotPoint 索引处的字符串值。
排序： 使用 Collections.sort() 方法对 grid 进行排序，使用自定义的 rowComparator 作为排序规则。只有当 pivotPoint 大于等于0 时才进行排序。
输出结果： 打印排序后的 grid。

运行结果：

如果 key 为 “345”，则输出：

Pivot Point: 2 [Test0, ABC, 123, A1] [Test4, MNO, 234, A5] [Test1, DEF, 345, A2] [Test2, GHI, 678, A3] [Test3, JKL, 901, A4]

如果 key 为 “foo”，则输出原始列表：

Pivot Point: -1 [Test0, ABC, 123, A1] [Test3, JKL, 901, A4] [Test1, DEF, 345, A2] [Test4, MNO, 234, A5] [Test2, GHI, 678, A3]

注意事项

确保 List<List<String>> 是可变的，以便可以对其进行排序。
List.of() 创建的 List<String> 是不可变的。如果尝试修改这些列表，将会抛出异常。如果需要修改行数据，请使用 new ArrayList<>(List.of(…)) 创建行。
如果目标字符串 key 不存在于任何列中，则 findPivotPoint 方法返回 -1，并且不会进行排序。
可以根据实际需求修改 Comparator 的实现，例如忽略大小写、使用不同的排序算法等。