LeetCode Everyday - 分组的最大数量

121 阅读1分钟

携手创作,共同成长!这是我参与「掘金日新计划 · 8 月更文挑战」的第23天,点击查看活动详情 >>

分组的最大数量

给你一个正整数数组 grades ,表示大学中一些学生的成绩。你打算将 所有 学生分为一些 有序 的非空分组,其中分组间的顺序满足以下全部条件:

第 i 个分组中的学生总成绩 小于 第 (i + 1) 个分组中的学生总成绩,对所有组均成立(除了最后一组)。 第 i 个分组中的学生总数 小于 第 (i + 1) 个分组中的学生总数,对所有组均成立(除了最后一组)。 返回可以形成的 最大 组数。

示例1:

输入:grades = [10,6,12,7,3,5]
输出:3
解释:下面是形成 3 个分组的一种可行方法:
- 第 1 个分组的学生成绩为 grades = [12] ,总成绩:12 ,学生数:1
- 第 2 个分组的学生成绩为 grades = [6,7] ,总成绩:6 + 7 = 13 ,学生数:2
- 第 3 个分组的学生成绩为 grades = [10,3,5] ,总成绩:10 + 3 + 5 = 18 ,学生数:3 
可以证明无法形成超过 3 个分组。

示例2:

输入: grades = [8,8]
输出: 1
解释: 只能形成 1 个分组,因为如果要形成 2 个分组的话,会导致每个分组中的学生数目相等。

提示:

  • 1 <= grades.length <= 105
  • 1 <= grades[i] <= 105

解题思路:

题目要求每个分组都比前一个分组的长度与总和都大,可以想到贪心的方法,使得每次加入分组的元素尽可能小。

首先对数组排序
设置变量记录上一分组的长度与总和
遍历数组,每一次循环分一个组
如果当前分组的长度或者总和不都大于上一组,则继续向分组内添加元素
记录循环次数并返回答案

我的答案:

/**
 * @param {number[]} grades
 * @return {number}
 */
var maximumGroups = function(grades) {
    grades.sort(function(a, b) { 
        return a - b
    })
    var lastLen = 0             //lastLen记录上一次分组长度
    var lastSum = 0             //lastSum记录上一次分组和
    var cnt = 0                 //cnt记录分组个数
    for (var i = 0; i < grades.length; i++) {
        var len = 0, sum = 0    //分别记录当前分组长度与总和
        //如果当前len或者sum不都大于上一组,则继续向分组内添加
        for (; len < lastLen + 1 || sum < lastSum + 1; len++, i++) {    
            if (i >= grades.length)
                break
            sum += grades[i]
        }
        lastSum = sum
        lastLen = len
        cnt++
    }
    return cnt
};

最后

如果有更好的解法或者思路, 欢迎在评论区和我交流~ ღ( ´・ᴗ・` )