思路
双指针。左右两个指针,首先分别从左边寻找大于k的第一个数,从右边寻找小于k的第一个数,然后交换。重复此过程,直到left>right。注意,在这里的循环条件是left<=right,目的是让循环结束后数组的划分更加清晰,可以直接返回left,而不需要在循环外面再写另外的判断条件。
复杂度
时间复杂度O(n)
空间复杂度O(1)
代码
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) {
// write your code here
if (nums.length == 0) {
return 0;
}
int left = 0, right = nums.length - 1;
while (left <= right) {
while (left <= right && nums[left] < k) {
left++;
}
while (left <= right && nums[right] >= k) {
right--;
}
if (left <= right) {
int t = nums[left];
nums[left] = nums[right];
nums[right] = t;
left++;
right--;
}
}
return left;
}
}