给出一个整数数组 nums 和一个整数 k。划分数组(即移动数组 nums 中的元素),使得:
所有小于k的元素移到左边
所有大于等于k的元素移到右边
所有小于k的元素移到左边
所有大于等于k的元素移到右边
返回数组划分的位置,即数组中第一个位置 i,满足 nums[i] 大于等于 k。
public int partitionArray(int[] nums, int k) {
//write your code here
if (nums == null || nums.length == 0){
return 0;
}
int left = 0;
int right = nums.length - 1;
while (left <= right){
while(left <= right && /*left < k*/nums[left] < k){
left++;
}
while(left <= right && /*right >= k*/nums[right] >= k){
right--;
}
if(left <= right){
int temp = nums[left];
nums[left]=nums[right];
nums[right]= temp;
left++;
right--;
}
}
return left;
}
}