划分数组使最大差为 K

171 阅读3分钟

持续创作,加速成长!这是我参与「掘金日新计划 · 6 月更文挑战」的第16天,点击查看活动详情

说在前面

🎈每天进行一道算法题目练习,今天的题目是“划分数组使最大差为 K”。

问题描述

给你一个整数数组 nums 和一个整数 k 。你可以将 nums 划分成一个或多个 子序列 ,使 nums 中的每个元素都 恰好 出现在一个子序列中。

在满足每个子序列中最大值和最小值之间的差值最多为 k 的前提下,返回需要划分的 最少 子序列数目。

子序列 本质是一个序列,可以通过删除另一个序列中的某些元素(或者不删除)但不改变剩下元素的顺序得到。

示例 1:

输入:nums = [3,6,1,2,5], k = 2
输出:2
解释:
可以将 nums 划分为两个子序列 [3,1,2][6,5] 。
第一个子序列中最大值和最小值的差值是 3 - 1 = 2 。
第二个子序列中最大值和最小值的差值是 6 - 5 = 1 。
由于创建了两个子序列,返回 2 。可以证明需要划分的最少子序列数目就是 2 。

示例 2:

输入:nums = [1,2,3], k = 1
输出:2
解释:
可以将 nums 划分为两个子序列 [1,2][3] 。
第一个子序列中最大值和最小值的差值是 2 - 1 = 1 。
第二个子序列中最大值和最小值的差值是 3 - 3 = 0 。
由于创建了两个子序列,返回 2 。注意,另一种最优解法是将 nums 划分成子序列 [1][2,3]

示例 3:

输入:nums = [2,2,4,5], k = 0
输出:3
解释:
可以将 nums 划分为三个子序列 [2,2][4][5] 。
第一个子序列中最大值和最小值的差值是 2 - 2 = 0 。
第二个子序列中最大值和最小值的差值是 4 - 4 = 0 。
第三个子序列中最大值和最小值的差值是 5 - 5 = 0 。
由于创建了三个子序列,返回 3 。可以证明需要划分的最少子序列数目就是 3 。

提示:

1 <= nums.length <= 10^5
0 <= nums[i] <= 10^5
0 <= k <= 10^5

思路分析

首先我们应该要先理解一下题意,题目会给我们两个参数,一个数组nums和一个整数k,我们需要对这个数组nums进行分割,将其分割成n组,分割后的每一组数据中的最大值和最小值之差不能超过k,我们需要计算出组数n的最小值,也就是要我们计算最少分割成多少组可以得到满足题意的数组。
知道了题意之后其实就很简单了,我们只需要先将原数组进行排序,然后依次遍历将差值小于等于k的放到一个数组,超过k的则需要另开一个数组来存放,遍历完一遍之后我们就可以得出答案了。

  • 1、数组排序 修改内置sort函数的排序规则即可。
nums = nums.sort((a,b)=>{
    return a - b;
});
  • 2、遍历分组 因为是排过序的数组,所以新开数组时第一个元素即为最小值,我们只需要将遍历的元素和最小值进行比较,判断是否需要另开一数组即可。
let res = 1,p = nums[0];
for(let i = 1; i < nums.length; i++){
    if(nums[i] > p + k) {
        res++;
        p = nums[i];
    }
}

AC代码

/**
 * @param {number[]} nums
 * @param {number} k
 * @return {number}
 */
 var partitionArray = function(nums, k) {
    nums = nums.sort((a,b)=>{
        return a - b;
    });
    let res = 1,p = nums[0];
    for(let i = 1; i < nums.length; i++){
        if(nums[i] > p + k) {
            res++;
            p = nums[i];
        }
    }
    return res;
};

说在后面

🎉这里是JYeontu,现在是一名前端工程师,有空会刷刷算法题,平时喜欢打打羽毛球🏸 ,平时也喜欢写些东西,既为自己记录📋,也希望可以对大家有那么一丢丢的帮助,写的不好望多多谅解🙇,写错的地方望指出,定会认真改进😊,在此谢谢大家的支持,我们下文再见🙌。