码迷,mamicode.com
首页 > 编程语言 > 详细

31. 数组划分

时间:2018-01-06 11:55:32      阅读:149      评论:0      收藏:0      [点我收藏+]

标签:blog   hal   好处   partition   dex   you   lintcode   就是   span   

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

31. 数组划分

标签:blog   hal   好处   partition   dex   you   lintcode   就是   span   

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

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!