数组划分--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:
int partitionArray(vector<int> &nums, int k) {
int i = 0, j = nums.size() - 1;
while (i != j&&j>=0&&i<=nums.size())
{
if (nums[i] < k)
++i;
if (nums[j] >= k)
--j;
if (nums[i] >= k&&nums[j] < k)
swap(nums[i], nums[j]);
}
if (i == nums.size() - 1)
return nums.size();
else
return i;

}

};

#endif