LintCode-数组划分

时间:2023-03-10 01:22:18
LintCode-数组划分

题目描述:

  给出一个整数数组nums和一个整数k。划分数组(即移动数组nums中的元素),使得:

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

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

注意事项

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

样例

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

 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) {
int length = nums.length;
for(int i=0;i<length;i++){
if(nums[i]>=k){
for(int j=length-1;j>=i;j--){
if(nums[j]<k){
int temp;
temp = nums[i];
nums[i] = nums[j];
nums[j] = temp;
length = j+1;
break;
}
if(j==i){
return j;
}
}
}
}
return nums.length;
}
}