描述:
给出一个整数数组 nums 和一个整数 k。划分数组(即移动数组 nums 中的元素),使得:
- 所有小于k的元素移到左边
- 所有大于等于k的元素移到右边
返回数组划分的位置,即数组中第一个位置 i,满足 nums[i] 大于等于 k。
样例:
Example 1:
Input:
[],9
Output:
0
Example 2:
Input:
[3,2,2,1],2
Output:1
Explanation:
the real array is[1,2,2,3].So return 1
挑战
使用 O(n) 的时间复杂度在数组上进行划分。
注意事项
你应该真正的划分数组 nums,而不仅仅只是计算比 k 小的整数数,如果数组 nums 中的所有元素都比 k 小,则返回 nums.length。
代码:
private void swap(int x,int y){
int tmp=x;
x=y;
y=tmp;
}
private int __quickSort(int[] nums,int left,int right,int k){
//if(left>right)return -1;
int j=left-1;
for(int i=left;i<=right;i++){
if(nums[i]<k){
swap(nums[j+1],nums[i]);
j++;
}
}
return j+1;
}
public int partitionArray(int[] nums, int k) {
// write your code here
if(nums==null||nums.length==0){
return 0;
}
else{
int left=0;
int right=nums.length-1;
return __quickSort(nums,0,right,k);
}
}
补充说明:
这个题目其实就是快速排序的Partition操作,对于快速排序,脑海中一定要有的一个图如下:
已知数组的左起始位置和数组的右终止位置,从做开始向后遍历,这道题目不需要选第一个为标定元素,我们只需要将大于等于v的元素都放入后半部分即可,全部遍历完成后,j+1的位置即为我们要找的第一个位置的元素。