LintCode-数组划分

给出一个整数数组nums和一个整数k。划分数组(即移动数组nums中的元素),使得:

  • 所有小于k的元素移到左边
  • 所有大于等于k的元素移到右边

返回数组划分的位置,即数组中第一个位置i,满足nums[i]大于等于k。

您在真实的面试中是否遇到过这个题?  Yes
样例

给出数组nums=[3,2,2,1]和 k=2,返回 1

注意

你应该真正的划分数组nums,而不仅仅只是计算比k小的整数数,如果数组nums中的所有元素都比k小,则返回nums.length。

挑战

要求在原地使用O(n)的时间复杂度来划分数组

标签 
Expand  

相关题目 
Expand 

分析:原理就是快排的partition

代码:

class Solution {
public:
    int partitionArray(vector<int> &nums, int k) {
        // write your code here
        int p = 0;
        for(int i=0;i<nums.size();i++)
        {
            if(nums[i]<k)
            {
                swap(nums[i],nums[p]);
                p++;
            }
        }
        return p;
    }
};
    原文作者:LintCode题目解答
    原文地址: https://blog.csdn.net/wangyuquanliuli/article/details/47777981
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞