- 描述
给出一个整数数组 nums 和一个整数 k。划分数组(即移动数组 nums 中的元素),使得:
1、所有小于k的元素移到左边
2、所有大于等于k的元素移到右边
返回数组划分的位置,即数组中第一个位置 i,满足 nums[i] 大于等于 k。
- 样例
给出数组 nums = [3,2,2,1] 和 k = 2,返回 1.
- 思路
比k小的在左边,比k大的在右边,并不需要严格按照大小排序,那么,只需要遍历一次整个数组,将每个元素与k进行比较,比k小就移动到最左边,同时,用一个索引记录左边已经移动的数量,遍历完之后,该索引就是要求的结果。时间复杂度为n。
根据以上想法写出代码(javascript)
- 代码(javascipt)
const partitionArray = function (nums, k) {
let index=0; //用于记录左边的索引
for(let i=0;i<nums.length;i++){
if(nums[i]<k){
let temp=nums[i]; //通过交换的方式 ,将小于k的数移动到左边,并使索引自增
nums[i]=nums[index];
nums[index]=temp;
index++;
}
}
return index;
}