[leetcode每日一题] 908.最小差值 I
一、题目
1.题目描述
给你一个整数数组 nums
,和一个整数 k
。
在一个操作中,您可以选择 0 <= i < nums.length
的任何索引 i
。将 nums[i]
改为 nums[i] + x
,其中 x
是一个范围为 [-k, k]
的整数。对于每个索引 i
,最多 只能 应用 一次 此操作。
nums
的 分数 是 nums
中最大和最小元素的差值。
在对 nums
中的每个索引最多应用一次上述操作后,返回 nums
的最低 分数 。
示例 1:
输入:nums = [1], k = 0
输出:0
解释:分数是 max(nums) - min(nums) = 1 - 1 = 0。
示例 2:
输入:nums = [0,10], k = 2
输出:6
解释:将 nums 改为 [2,8]。分数是 max(nums) - min(nums) = 8 - 2 = 6。
示例 3:
输入:nums = [1,3,6], k = 3
输出:0
解释:将 nums 改为 [4,4,4]。分数是 max(nums) - min(nums) = 4 - 4 = 0。
提示:
1 <= nums.length <= 104
0 <= nums[i] <= 104
0 <= k <= 104
2.基础框架
C++基础框架代码如下:
int smallestRangeI(vector<int>& nums, int k)
3.解题思路
- 题目分析
-
题目目标是返回
nums
的最低分数。 -
nums
分数是nums
数组中最大元素和最小元素的差值。 -
每个索引的位置至多进行一次
nums[i] + x
操作。 -
x
的可以从[-k, k]
范围任取。 -
既然是得到最小的差值,也就是将较高的元素和较低的元素尽可能的相等。
-
通过遍历一次数组得到数组和为
sum
,然后求得中位数mid
。 -
然后进行一次排序,得到数组最大元素值
max
和最小元素值min
。 -
然后下面进行分类讨论:
(1) 如果
min + k = mid
并且max - k = mid
,那么最大值和最小值的都可以通过加x
操作变成mid
,那么最小的差值也就是0了。(2) 如果
min + k < mid
并且max - k > mid
,这意味着min
和max
都趋近于mid
,那最小的差值也就是(max - k) - (min + k)
。(3) 如果
min + k < mid
,分两种情况进行讨论(因为顺序执行下来,执行到该判断语句则默认不满足上面的情况):- 如果
max - k <= min + k
,也就是max - k
此时能够取到min + k
得值,所以这种情况下它们之间的最小差值为0。 - 如果
max - k > min + k
,那也就是max - k
取到最接近min + k
的范围在(min + k,mid)中,那这种情况下他们之间的最小差值为(max - k) - (min + k)
(4) 如果
max - k > mid
,也同理分为两种情况进行讨论,情况和(3)类似,就不再赘述了。 - 如果
-
实现代码:
int smallestRangeI(vector<int>& nums, int k) { int n = nums.size(); sort(nums.begin(), nums.end()); int mid = (nums[0] + nums[n - 1]) >> 1; if (nums[0] + k >= mid && nums[n - 1] - k <= mid) // (1) return 0; else if (nums[0] + k < mid && nums[n - 1] - k > mid) // (2) return (nums[n - 1] - k) - (nums[0] + k); else if (nums[0] + k < mid) // (3) { if (nums[n - 1] - k <= nums[0] + k) return 0; return (nums[n - 1] - k) - (nums[0] + k); } else if (nums[n - 1] - k > mid) // (4) { if (nums[0] + k >= nums[n - 1] - k) return 0; return (nums[n - 1] - k) - (nums[0] + k); } return 0; }