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

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;
}

但这只适用于值是预期类型的列表的情况。




Answer 1

匿名用户

这里有一个方法。我推测这些键可能有重复的数字，比如[3,5,2,2,1]。否则，更简单的算法将占上风。

• 将值从0排序到键。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
• 键列表必须是整数
类型

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();
}

Answer 2

您可以按如下方式使用泛型：

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。这将在运行时发生。