顺丰面试提到的一个算法题
顺丰面试提到的一个算法题
面试过程中大脑空白,睡了一觉后突然想明白了
原理非常简单就是根据数组中元素的值对值对应的索引进行排序
哎,,,,具体看以下代码吧[使用 Java 17 中 Stream 实现]
最好别用 CSDN 提供的在线运行验证代码,我建议你还是将代码拷贝到本地目标环境下运行
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()));}}