[leetcode每日一题] 908.最小差值 I

[leetcode每日一题] 908.最小差值 I

一、题目

原题链接：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.解题思路

• 题目分析

1. 题目目标是返回nums的最低分数。

2. nums分数是nums数组中最大元素和最小元素的差值。

3. 每个索引的位置至多进行一次nums[i] + x操作。

4. x的可以从[-k, k]范围任取。

5. 既然是得到最小的差值，也就是将较高的元素和较低的元素尽可能的相等。

6. 通过遍历一次数组得到数组和为sum，然后求得中位数mid。

7. 然后进行一次排序，得到数组最大元素值max和最小元素值min。

8. 然后下面进行分类讨论：

   (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;
  }