1846.减小和重新排列数组后的最大元素
难度:中等
题目:
给你一个正整数数组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)