数组划分

来源:互联网 发布:淘宝有货到付款的产品 编辑:程序博客网 时间:2024/06/10 07:55

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

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

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

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

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

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


class Solution {public:    int partitionArray(vector<int> &nums, int k) {        if(nums.empty()||nums.size()==0)             return 0;        int i=0, n=nums.size()-1;        while(true)         {            while(i<n&&nums[n]>=k)            {                n--;            }            while(i<n&&nums[i]<k)             {                i++;            }            if(i==n)                 break;            swap(i,n,nums);        }        if(i==0 && nums[i]>=k)               return n;        if(n==nums.size()-1 && nums[i]<k)               return n+1;        return n+1;    }    void swap(int i,int n,vector<int> &nums)    {         int temp=nums[i];         nums[i]=nums[n];         nums[n]=temp;    }};


0 0
原创粉丝点击