31.Partition Array-数组划分（中等题）

数组划分

1. 题目

   给出一个整数数组 nums 和一个整数 k。划分数组（即移动数组 nums 中的元素），使得：
   所有小于k的元素移到左边
   所有大于等于k的元素移到右边
   返回数组划分的位置，即数组中第一个位置 i，满足 nums[i] 大于等于 k。

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

2. 样例

   给出数组 nums = [3,2,2,1] 和 k = 2，返回 1.

3. 挑战

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

4. 题解

首尾双指针，分别查找大于等于k和小于k的数，再进行交换。遍历完成后的首指针就是答案。如果数组全部元素都大于或小于k，则首指针不会移动，此时比较下nums[0]和k的大小关系就知道答案是0还是nums.length。

public class Solution {
    /** 
     *@param nums: The integer array you should partition
     *@param k: As description
     *return: The index after partition
     */
    public int partitionArray(int[] nums, int k) {
        if (nums.length == 0)
        {
            return 0;
        }
        int start = 0;
        int end = nums.length - 1;
        while (start < end)
        {
            while (start < nums.length && nums[start] < k)
            {
                start++;
            }
            while (end > 0 && nums[end] >= k)
            {
                end--;
            }
            if (start < end)
            {
                swap(nums,start,end);
            }
        }
       if (start == 0)
       {
           return nums[0] > k ? 0 : nums.length;
       }
       return start;
    }

    private void swap(int[] nums, int i, int j)
    {
        int temp = nums[i];
        nums[i] = nums[j];
        nums[j] = temp;
    }
}

Last Update 2016.9.21