LeetCode - 1402 权值排序最大和【排序】

1402. 做菜顺序

示例 1：

输入：satisfaction = [-1,-8,0,5,-9]
输出：14
解释：去掉第二道和最后一道菜，最大的喜爱时间系数和为 (-1*1 + 0*2 + 5*3 = 14) 。每道菜都需要花费 1 单位时间完成。
示例 2：

输入：satisfaction = [4,3,2]
输出：20
解释：按照原来顺序相反的时间做菜 (2*1 + 3*2 + 4*3 = 20)
示例 3：

输入：satisfaction = [-1,-4,-5]
输出：0
解释：大家都不喜欢这些菜，所以不做任何菜可以获得最大的喜爱时间系数。
示例 4：

输入：satisfaction = [-2,5,-1,0,3,-3]
输出：35

提示：

• n == satisfaction.length
• 1 <= n <= 500
• -10^3 <= satisfaction[i] <= 10^3

思路

  1. 题意：计算数组最大累加和：a[0]*1+a[1]*2+a[2]*3 ……，可删除部分数字。

  2. 方法：若数组全负返回0。否则，对数组排序。从大到小计算

假设数组仅有一个元素

a[0]

数组有两个元素，a[0]最大，a[1]第二大

a[1] + a[0]*2 > a[0]                                                        即a[0] + a[1] > 0

数组有三个元素，a[0]最大，a[1]第二大，a[2]第三大

a[2] + a[1]*2 + a[0]*3 > a[1] + a[0]*2                             即a[0] + a[1] + a[2] > 0

对数组从大到小遍历，设置一个maxSum，若累加和大于0，则将累加和叠加到maxSum中；若累加和小于0，则停止遍历。

代码

class Solution {
    public int maxSatisfaction(int[] satisfaction) {
        boolean flag = true;
        for (int num: satisfaction) {
            if (num > 0) {
                flag = false;
                break;
            }
        }
        if (flag)
            return 0;
        
        int maxSum = 0;
        int sum = 0;
        Arrays.sort(satisfaction);
        for (int i = satisfaction.length - 1; i >= 0; i--) {
            if (sum + satisfaction[i] > 0) {
                sum += satisfaction[i];
                maxSum += sum;
            }
            else
                break;
        }
        return maxSum;
    }
}

运行结果