1846.减小和重新排列数组后的最大元素贪心求极值的三种写法！-CSDN博客

本文链接：https://blog.csdn.net/BreezePython/article/details/118755568

1846.减小和重新排列数组后的最大元素

https://leetcode-cn.com/problems/maximum-element-after-decreasing-and-rearranging/solution/5732jian-xiao-he-zhong-xin-pai-lie-shu-z-gsl8/

难度：中等

题目：

给你一个正整数数组arr。请你对 arr执行一些操作(也可以不进行任何操作)，使得数组满足以下条件：

arr中第一个元素必须为1。
任意相邻两个元素的差的绝对值小于等于1，也就是说，对于任意的 1 <= i < arr.length(数组下标从 0 开始)，都满足abs(arr[i] - arr[i - 1]) <= 1。abs(x)为x的绝对值。
你可以执行以下 2 种操作任意次：

减小 arr中任意元素的值，使其变为一个更小的正整数。
重新排列arr中的元素，你可以以任意顺序重新排列。
请你返回执行以上操作后，在满足前文所述的条件下，arr中可能的最大值。

提示：

1 <= arr.length <= 105
1 <= arr[i] <= 109

示例：

示例 1：

输入：arr = [2,2,1,2,1]
输出：2
解释：
我们可以重新排列 arr 得到 [1,2,2,2,1] ，该数组满足所有条件。
arr 中最大元素为 2 。
示例 2：

输入：arr = [100,1,1000]
输出：3
解释：
一个可行的方案如下：
1. 重新排列 arr 得到 [1,100,1000] 。
2. 将第二个元素减小为 2 。
3. 将第三个元素减小为 3 。
现在 arr = [1,2,3] ，满足所有条件。
arr 中最大元素为 3 。
示例 3：

输入：arr = [1,2,3,4,5]
输出：5
解释：数组已经满足所有条件，最大元素为 5 。

分析

看题看了半天，结果发现是一道如此简单的问题...思路就是循环贪心取最小。

首先我们需要对数组进行排序
初始设置 ret = 1
之后由于是排序的，所以arr[i] >= arr[i-1]
由于不能取比当前值大的值，所以每次的取值都是ret = min(arr[i],ret + 1)即可

解题：

按部就班

class Solution:
    def maximumElementAfterDecrementingAndRearranging(self, arr):
        arr.sort()
        arr[0] = 1
        for i in range(1, len(arr)):
            tmp = arr[i] - arr[i - 1]
            if tmp != 0:
                arr[i] = arr[i - 1] + 1
        return arr[-1]

简写

class Solution:
    def maximumElementAfterDecrementingAndRearranging(self, arr):
        ret = 1
        for i in sorted(arr)[1:]:
            ret = min(i,ret + 1)
        return ret

来个偏门的reduce操作

class Solution:
    def maximumElementAfterDecrementingAndRearranging(self, arr):
        arr.sort()
        arr[0] = 1
        return reduce(lambda x, y: min(x + 1, y), arr)

欢迎关注我的公众号: 清风Python，带你每日学习Python算法刷题的同时，了解更多python小知识。

我的个人博客：https://qingfengpython.cn

力扣解题合集：https://github.com/BreezePython/AlgorithmMarkdown