31 - 数组划分

2017.9.18

数组划分，采用的就是快排的思路。

细节需要处理一下，比如数组都小于k的时候，要返回数组的长度。

以及当nums【height】< k时，位置需要向后加1.

public class Solution {
/*
* @param nums: The integer array you should partition
* @param k: An integer
* @return: The index after partition
*/
public static int partitionArray(int[] nums, int k) {
// write your code here
if(nums == null ||nums.length == 0){
return 0;
}
int low = 0;
int height = nums.length-1;
int tmp = nums[height];
while(low < height){
while(low < height && nums[low] < k){
low ++;
}
nums[height] = nums[low];
while(low < height && nums[height] >= k){
height --;
}
nums[low] = nums[height];
}
if(tmp < k){
return low+1;
}
if(nums[nums.length-1] < k){
return nums.length;
}
return low;
}
}