数组划分--lintcode

问题描述：

详见：http://www.lintcode.com/zh-cn/problem/partition-array/

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

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

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

注意事项：

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

#ifndef C31_H
#define C31_H
#include<iostream>
#include<vector>
using namespace std;
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) {
		// write your code here
		if (nums.empty())
			return 0;
		int left = 0, right = nums.size() - 1;
		//两根指针
		while (left < right)
		{
			if (nums[left] >= k)
			{
				if (nums[right] < k)
				{
					swap(nums[left], nums[right]);
					left++;
					right--;
				}
				else
				{
					right--;
				}
			}
			else
			{
				if (nums[right] >= k)
				{
					left++;
					right--;
				}
				else
				{
					left++;
				}
			}
		}
		for (int i = 0; i < nums.size(); ++i)
		{
			if (nums[i] >= k)
				return i;
		}
		return nums.size(); 
	}
};
#endif