给出一个整数数组 nums 和一个整数 k。划分数组(即移动数组 nums 中的元素),使得:
所有小于k的元素移到左边
所有大于等于k的元素移到右边
返回数组划分的位置,即数组中第一个位置 i,满足 nums[i] 大于等于 k。
类似与快速排序 相向双指针
在左边找到一个大于等于k的元素 右边找到一个小于k的元素 交换位置
public class Solution {
/**
* @param nums: The integer array you should partition
* @param k: An integer
* @return: The index after partition
*/
public int partitionArray(int[] nums, int k) {
int n = nums.length;
if(nums == null || n == 0)
return 0;
int left = 0, right = n - 1;
while(left < right){
while(left < right && nums[left] < k){
left++;
}
while(left < right && nums[right] >= k){
right--;
}
if(left < right){
int temp = nums[left];
nums[left] = nums[right];
nums[right] = temp;
}
}
if(nums[left] < k)
return left + 1;
return left;
}
}