数组划分

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

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

返回数组划分的位置，即数组中第一个位置i，满足nums[i]大于等于k。

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

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

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

class Solution {
public:
    int partitionArray(vector<int> &nums, int k) {
        if(nums.empty()||nums.size()==0)
             return 0;
        int i=0, n=nums.size()-1;
        while(true) 
        {
            while(i<n&&nums[n]>=k)
            {
                n--;
            }
            while(i<n&&nums[i]<k) 
            {
                i++;
            }
            if(i==n) 
                break;
            swap(i,n,nums);
        }
        if(i==0 && nums[i]>=k) 
              return n;
        if(n==nums.size()-1 && nums[i]<k) 
              return n+1;
        return n+1;
    }
    void swap(int i,int n,vector<int> &nums)
    {
         int temp=nums[i];
         nums[i]=nums[n];
         nums[n]=temp;
    }
};