LintCode：数组划分

给出一个整数数组nums和一个整数k。划分数组（即移动数组nums中的元素），使得：

所有小于k的元素移到左边

所有大于等于k的元素移到右边

返回数组划分的位置，即数组中第一个位置i，满足nums[i]大于等于k。
您在真实的面试中是否遇到过这个题？

Yes

样例

给出数组nums=[3,2,2,1]和 k=2，返回 1

注意

你应该真正的划分数组nums，而不仅仅只是计算比k小的整数数，如果数组nums中的所有元素都比k小，则返回nums.length。

挑战

要求在原地使用O(n)的时间复杂度来划分数组

标签 Expand

相关题目 Expand

解题思路：
双指针，将小于k的元素放在左边，大于k的元素放在右边。

public class Solution {
/**
*@param nums: The integer array you should partition
*@param k: As description
*return: The index after partition
*/
public int partitionArray(int[] nums, int k) {
//write your code here
int left = 0;int right = nums.length-1;
while(left<right){
while(left<nums.length&&nums[left]<k){
left++;
}
while(right>=0&&nums[right]>=k){
right--;
}
if(left>=right){
break;
}
swap(left, right, nums);
}
return left;
}
public void swap(int l, int r, int[] A) {
int tmp = A[l];
A[l] = A[r];
A[r] = tmp;
}
}