数组划分

中等 数组划分 查看运行结果 

25%

通过

给出一个整数数组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)的时间复杂度来划分数组

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 len = nums.length;
	    int i = 0;
	    int j = len-1;
	    while(i<j)
	    {
	        if(nums[i]>=k)
	        {
	            int tmp = nums[i];
	            nums[i] = nums[j];
	            nums[j] = tmp;
	            j--;
	        }
	        else i++;
	    }
	    for(i=0;i<len;i++)
	    {
	        if(nums[i]>=k)
	            return i;
	        else if(i==len-1)
	            return len;
	    }
	    return 0;
    }
}