31. 数组划分
给出一个整数数组 nums 和一个整数 k。划分数组(即移动数组 nums 中的元素),使得:
- 所有小于k的元素移到左边
- 所有大于等于k的元素移到右边
返回数组划分的位置,即数组中第一个位置 i,满足 nums[i] 大于等于 k。
注意事项
你应该真正的划分数组 nums,而不仅仅只是计算比 k 小的整数数,如果数组 nums 中的所有元素都比 k 小,则返回 nums.length。
样例
给出数组 nums = [3,2,2,1]
和 k = 2
,返回 1
.
挑战
使用 O(n) 的时间复杂度在数组上进行划分。
试了很久的把>=k的往后扔,老是不对,结果看了下其他人的思路是把<k的往前扔,吐血
1 int partitionArray(vector<int> &nums, int k) { 2 // write your code here 3 if(nums.empty()){ 4 return 0; 5 } 6 int index=0; 7 for(int i=0;i<nums.size();i++){ 8 if(nums[i]<k){ 9 swap(nums[i],nums[index]); 10 index++; 11 } 12 } 13 return index; 14 }
index是从0增加还是有好处的,本来就是要求>=k的第一个元素,这样方便很多