java - 在java中找到第k个最小的数字

Question

下面是查找数组中第 k 个最小数字的 java 代码。此代码适用于初级版本，未使用枢轴索引。

class Func {
    int kthSmallest(int arr[], int smallestIndex, int size){
        int smallest = arr[0];
        int removeLocation=0;
        for(int i=0; i<size; i++){
            if(arr[i] < smallest){
                smallest = arr[i];
                removeLocation = i;
            }
        }
        for(int i=removeLocation; i<size; i++){
            arr[i] = arr[i+1];
        }
        if(smallestIndex == 1) {
            return smallest;
        } else return kthSmallest(arr, smallestIndex-1, size-1);
    }
}

public class Test {
    public static void main(String[] args){
        Func test = new Func();
        int number = test.kthSmallest(new int[]{3,1,2,4,5}, 2, 5);
        System.out.println(number);
    }
}

结果是

Exception in thread "main" java.lang.ArrayIndexOutOfBoundsException: 5

我找不到该错误出现的位置。

Answer 1

在第二个 for 循环中，您通过 i+1 修改访问更多超出范围的内容，如下所示

for(int i=removeLocation; i<size-1; i++){
            arr[i] = arr[i+1];
        }

输出

2