明确:什么时候使用Partition方法来优化算法?
首先减治减治,减而治之,如果我们要找到的是某个特定量,如第几大的数、第几小的数,第多少个加入的人,诸如此类,常规来说我们需要先将整个集合(数组)排序,然后输出第几个数,但是减治算法要求我们,用最少的处理次数得到需要的元素,即减少了没有必要的排序,比如找数组中第三大的数,我们只用找一个数,数组中仅存在两个数比它小即可,而无需处理比它更大的数。
方法实现例:数组中第K个最大元素:
定义一个partition方法来找到指定集合中比左边界对应的数值小的元素个数,而元素个数的值就代表初始左边界
**对应的数值比该集合中多少个数大,记该数为n,即左边界是第n+1大的数,如果要找的第几大的数比n+1还要大,那就以得到的n+1为新的
**左边界(否则设定右边界为n-1),再在新的左右边界中重复partition,但是要注意初始的索引n要设定为左边界的值
class Solution {
public int findKthLargest(int[] nums, int k) {
int len=nums.length;
int left=0;
int right=len-1;
while(true){
int target=partition(nums,left,right);
if(target==len-k){
return nums[target];
}else if(target>len-k){
right=target-1;
}else {
left=target+1;
}
}
}
public int partition(int[] nums,int left,int right){
int pivot=nums[left];
int j=left;
for(int i=left+1;i<=right;i++){
if(nums[i]<pivot){
j++;
swap(nums,i,j);
}
}
swap(nums,left,j);
return j;
}
public void swap(int[] nums,int a,int b){
int temp=nums[a];
nums[a]=nums[b];
nums[b]=temp;
}
}