順豐面試提到的一個算法題
面試過程中大腦空白,睡了一覺后突然想明白了
原理非常簡單就是根據數組中元素的值對值對應的索引進行排序
哎,,,,具體看以下代碼吧[使用 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()));}}