基于输入参数类型的动态返回类型

Question

我正在尝试编写一个小函数，该功能根据另一个元素进行两个列表，并根据另一个元素进行分类。因此，类似：

List<Integer> keys = Arrays.asList(3, 5, 2, 4, 1);
List<String> valuesToSort = Arrays.asList("A", "B", "C", "D", "E");

List<String> sortedValues = sort(keys, valuesToSort);

将导致排序列表[E，C，A，D，B]。

但是，valuestosort可能是不同事物的列表，例如整数，浮点或其他列表。理想情况下，我希望我的程序获取我投入的任何列表，根据键对其进行排序，然后返回与输入相同类型的列表。我该怎么做？如果修复了值的类型，则排序本身将很简单，例如

public List<String> sort(List<Integer> keys, List<String> values){
    Multimap<Integer, String>> multiMap = LinkedListMultimap.create();
    for (int i = 0; i < keys.size(); i++) {
        multiMap.put(keys.get(i), values.get(i));
    }
    List<String>> sortedValues = Lists.newArrayList();
    for (Integer key : Ordering.natural().sortedCopy(multiMap.keys())) {
        for (String value : multiMap.get(key)) {
            sortedValues.add(value);
        }
    }
return sortedValues;
}

，仅当value是预期类型的​​列表时才有效。

Answer 1

您可以使用泛型来实现此目的，如下所示：

public List<T> sort(List<Integer> keys, List<T> values){
    Multimap<Integer, T>> multiMap = LinkedListMultimap.create();
    for (int i = 0; i < keys.size(); i++) {
        multiMap.put(keys.get(i), values.get(i));
    }
    List<T> sortedValues = Lists.newArrayList();
    for (Integer key : Ordering.natural().sortedCopy(multiMap.keys())) {
        for (T value : multiMap.get(key)) {
            sortedValues.add(value);
        }
    }
    return sortedValues;
}

现在，当您传递字符串列表时，泛型 T 将被视为 String。当您传递整数列表时，T 将变为 Integer。这将在运行时发生。

Answer 2

这是一种方法。我假设这些键可能有重复的数字，例如 [3, 5, 2, 2, 1]。否则，更简单的算法将占上风。

• 使用引用示例的keys 列表对从0 到keys.size() 的值进行排序
• ，这将产生以下整数流。 [4 2 0 3 1]
• 然后使用它们来索引值列表。
• 这将产生所需的顺序。

List<Integer> keys = Arrays.asList(3, 5, 2, 4, 1);
List<String> valuesToSort =
        Arrays.asList("A", "B", "C", "D", "E");

List<String> result = sort(keys, valuesToSort);
System.out.println(result);

打印

[E, C, A, D, B]

通用方法

• 使其静态
• 值列表只是索引，因此可以是任何类型 T
• 键列表必须是 Integer 类型

public static <T> List<T> sort(List<Integer> keys, List<T> values){
    return IntStream.range(0, keys.size()).boxed()
            .sorted(Comparator.comparing(keys::get))
            .map(values::get).toList();
}