lintcode之数组划分

题目：

给出一个整数数组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;
    }
};