题目:
给出一个整数数组nums和一个整数k。划分数组(即移动数组nums中的元素),使得:
- 所有小于k的元素移到左边
- 所有大于等于k的元素移到右边
返回数组划分的位置,即数组中第一个位置i,满足nums[i]大于等于k。
解答:
两个指针交换
注意边界和只在l < r的时候才能进行交换
class Solution {
public:
int partitionArray(vector<int> &nums, int k) {
// write your code here
int l = 0;
int r = nums.size() - 1;
while (l < r) {
while (l < nums.size() && nums[l] < k)
l++;
while (r >= 0 && nums[r] >= k)
r--;
if (l == nums.size())
return nums.size();
if (r == -1)
return 0;
if (l < r) {
int temp = nums[l];
nums[l] = nums[r];
nums[r] = temp;
}
}
return l;
}
};