leetcode-2966.划分数组并满足最大差限制

HNU混子

于 2025-06-20 16:09:32 发布

阅读量435

点赞数 6

CC 4.0 BY-SA版权

分类专栏： leetcode 文章标签：算法 leetcode 动态规划

本文链接：https://blog.csdn.net/xhhgffd/article/details/148794754

leetcode 专栏收录该内容

7 篇文章

订阅专栏

题目描述

解题思路

这里只需要注意每个子数组的大小都是3，我一开始看成了3个子数组.....

然后每个子数组的最大差一定是排好序之后的最大值减去最小值，所以我们只需要先对整体进行排序，然后每三个一组，用最大值减最小值判断一下就好了。

代码

class Solution {
public:
    vector<vector<int>> divideArray(vector<int>& nums, int k) {
        ranges::sort(nums.begin(),nums.end());
        vector<vector<int>> ret;
        int step = nums.size() / 3;
        int n = nums.size();
        for(int i = 0; i < n; i += 3)
        {
            if(nums[i + 2] - nums[i] > k)
            return ret;
        } 
        ret.resize(step);
        for(int i = 0;i < step; i++)
        {
            ret[i].insert(ret[i].begin(),nums.begin() + (3 * i), nums.begin() + (3 * (i + 1)));
        }
        return ret;
    }
};