31. 数组划分

31. 数组划分

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

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

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

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

样例

例1:

输入:

[],9

输出:

0

例2:

输入:

[3,2,2,1],2

输出:1

解释:

真实的数组为[1,2,2,3].所以返回 1

挑战

使用 O(n) 的时间复杂度在数组上进行划分。

注意事项

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

public class Solution {

    /**

     * @param nums: The integer array you should partition

     * @param k: An integer

     * @return: The index after partition

     */

    public int partitionArray(int[] nums, int k) {

            int i=0,j=nums.length-1;

            while (i<=j){

                if (nums[i]<k){

                    i++;

                    continue;

                }

                if (nums[j]>=k){

                    j--;

                    continue;

                }

//                if (nums[i]>k&&nums[j]<k){

                    // int temp=nums[i];

                    // nums[i]=nums[j];

                    // nums[j]=temp;

//                }

                i++;

                j--;

            }

            return i>nums.length?nums.length:i;

    }

}