31. 数组划分

Posted 2020-10-19 三人木君

给出一个整数数组 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的第一个元素，这样方便很多