leetcode 2294. 划分数组使最大差为 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

分析：现将数组排序，接着贪心分组。取划分的子序列最小值为当前遍历值，之后向右找到划分子序列的最大值，即与最小值差值小于等于 k 的值。

cpp 复制代码

/**
 * Return an array of arrays of size *returnSize.
 * The sizes of the arrays are returned as *returnColumnSizes array.
 * Note: Both returned array and *columnSizes array must be malloced, assume caller calls free().
 */
int cmp(const void *a,const void *b)
{
    int *aa=(int*)a;
    int *bb=(int*)b;
    return *aa-*bb;
}

int** divideArray(int* nums, int numsSize, int k, int* returnSize, int** returnColumnSizes) {
    qsort(nums,numsSize,sizeof(int),cmp);

    int len=numsSize/3;*returnSize=len;
    int **ans=(int**)malloc(sizeof(int*)*len);
    *returnColumnSizes=(int*)malloc(sizeof(int)*len);
    for(int i=0;i<len;++i)
    {
        ans[i]=(int*)malloc(sizeof(int)*3);
        (*returnColumnSizes)[i]=3;
    }
    for(int i=0,t=0;i<numsSize;i+=3,t++)
    {
        if(nums[i+2]-nums[i]>k)
        {
            *returnSize=0;
            free(*returnColumnSizes);
            for(int j=0;j<len;++j)
                free(ans[j]);
            free(ans);
            return NULL;
        }
        ans[t][0]=nums[i],ans[t][1]=nums[i+1],ans[t][2]=nums[i+2];
    }

    return ans;
}