顺丰面试提到的一个算法题

AntsCode(码蚁编程)2025-07-24 13:45

顺丰面试提到的一个算法题

面试过程中大脑空白，睡了一觉后突然想明白了

原理非常简单就是根据数组中元素的值对值对应的索引进行排序

哎，，，，具体看以下代码吧 $使用 Java 17 中 Stream 实现$

最好别用 CSDN 提供的在线运行验证代码，我建议你还是将代码拷贝到本地目标环境下运行

java 复制代码

import java.util.AbstractMap;
import java.util.stream.IntStream;

public class Main {

    // 获取数组中第k大的元素及其索引
    public static void main(String[] args) {

        // 定义变量k，用于后续操作中指定获取的元素数量
        int k = 5;

        // 初始化一个整数数组nums，包含一组待处理的数字
        int[] nums = {12, 11, 9, 15, 10};

        // 使用IntStream.range生成一个包含数组索引的流
        // .range(0, nums.length) 生成一个从0到数组长度的索引流
        // .boxed() 将int值包装为Integer对象，以便进行后续的排序操作
        // .sorted((i1, i2) -> Integer.compare(nums[i2], nums[i1])) 根据数组nums中的值对索引进行降序排序
        // .limit(k) 限制流的大小为k，即只取数组中最大的k个元素
        // .skip(k - 1L) 跳过前k-1个元素，留下第k个最大的元素
        // .map(i -> new AbstractMap.SimpleEntry<>(i, nums[i])) 将索引i映射为一个键值对(i, nums[i])，便于后续处理
        // .findFirst() 查找流中的第一个元素，即第k个最大的元素
        // .ifPresent(...) 如果找到了元素，则执行给定的消费函数，输出数组索引及其对应的值
        IntStream.range(0, nums.length)
                .boxed()
                .sorted((i1, i2) -> Integer.compare(nums[i2], nums[i1]))
                .limit(k)
                .skip(k - 1L)
                .map(i -> new AbstractMap.SimpleEntry<>(i, nums[i]))
                .findFirst()
                .ifPresent(entry ->
                        System.out.println("数组索引: " + entry.getKey() + " --> 数组索引对应的值: " + entry.getValue())
                );

    }

}