【两根指针——快速选择系列】Lintcode 31. 数组划分

Lintcode 31. 数组划分

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

• 所有小于k的元素移到左边
• 所有大于等于k的元素移到右边
  返回数组划分的位置，即数组中第一个位置 i，满足 nums[i] 大于等于 k。
• 你应该真正的划分数组 nums，而不仅仅只是计算比 k 小的整数数，如果数组 nums 中的所有元素都比 k 小，则返回 nums.length。

解题思路：通过头尾指针跳过小于k的前缀和大于等于k的后缀，可以找到与第一个大于等于k的值和最后一个小于k的值。进行交换后可达到划分数组的目的，直到找到两个指针相遇为止。

class Solution {
    
    
public:
    /**
     * @param nums: The integer array you should partition
     * @param k: An integer
     * @return: The index after partition
     */
    int partitionArray(vector<int> &nums, int k) {
    
    
        if (0 == nums.size()) {
    
    
            return 0;
        }
        
        int left = 0, right = nums.size() - 1;
        while (left <= right) {
    
    
            while (left <= right && nums[left] < k) {
    
    
                left++;
            }
            while (left <= right && nums[right] >= k) {
    
    
                right--;
            }
            if (left <= right) {
    
    
                swap(nums[left], nums[right]);
                left++;
                right--;
            }
        }
        return left;
    } 
};