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的第一个元素,这样方便很多

原文地址:https://www.cnblogs.com/TheLaughingMan/p/8213239.html