lintcode(11)——数组划分

最新推荐文章于 2025-06-26 16:04:08 发布

原创最新推荐文章于 2025-06-26 16:04:08 发布 · 182 阅读

0 ·

CC 4.0 BY-SA版权

C++ 专栏收录该内容

43 篇文章

订阅专栏

该博客探讨了如何将整数数组根据特定值k进行划分，将所有小于k的元素移到左侧，大于等于k的移到右侧。文章提供两个样例解释划分过程，并提出了在O(n)时间复杂度内完成数组划分的挑战。

31. 数组划分

给出一个整数数组 nums 和一个整数 k。划分数组（即移动数组 nums 中的元素），使得：

所有小于k的元素移到左边
所有大于等于k的元素移到右边
返回数组划分的位置，即数组中第一个位置 i，满足 nums[i] 大于等于 k。

样例

Example 1:
Input:
[],9
Output:
0
Example 2:
Input:
[3,2,2,1],2
Output:1
Explanation:
the real array is[1,2,2,3].So return 1
挑战

使用 O(n) 的时间复杂度在数组上进行划分。

注意事项

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

class Solution {
public:
    int partitionArray(vector<int> &nums, int k) {
        // write your code here
        vector<int> result1;
        vector<int> result2;
        int i;
        if(nums.empty())
        {
            return 0;
        }
        for(i=0;i<nums.size();i++)
        {
            if(nums[i]<k)
            {
                result1.push_back(nums[i]);
            }
            else
            {
                result2.push_back(nums[i]);
            }
        }
        for(i=0;i<result1.size();i++)
        {
            nums[i]=result1[i];
        }
        for(i=0;i<result2.size();i++)
        {
            nums[result1.size()+i]=result2[i];
        }
        return result1.size();
    }
};

class Solution {
public:
    int partitionArray(vector<int> &nums, int k) {
        // write your code here
       int i = 0, j = nums.size() - 1;
        while (i <= j) {
            while (i <= j && nums[i] < k) i++;
            while (i <= j && nums[j] >= k) j--;
            if (i <= j) {
                int temp = nums[i];
                nums[i] = nums[j];
                nums[j] = temp;
                i++;
                j--;
            }
        }
        return i;
    }
};