描述
在农场里,农民们有一群牛,每头牛的体重不同。农民们将所有牛的体重记录在一个数组中。现在农民们想要知道,如果将这些牛的体重从小到大排序,那么第k小的体重是多少。请你编写一个程序,找出数组中第k小的元素。
你必须设计并实现时间复杂度为 O(n) 的算法解决此问题。
示例1
输入:
[600,500,800,700,550,650],3
返回值:
600
思路
没啥技术含量,学个java里面的排序函数吧
java
import java.util.*;
public class Solution {
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param weights int整型一维数组
* @param k int整型
* @return int整型
*/
public int findKthSmallest (int[] weights, int k) {
Arrays.sort(weights);
int weight = weights[k - 1];
return weight;
}
}