【Leetcode】P5616 数组的最小偏移量

Leetcode P5616 数组的最小偏移量

给你一个由 n 个正整数组成的数组 nums 。

你可以对数组的任意元素执行任意次数的两类操作：

• 如果元素是 偶数 ，除以 2。例如，如果数组是 [1,2,3,4] ，那么你可以对最后一 个元素执行此操作，使其变成 [1,2,3,2]
• 如果元素是 奇数 ，乘上 2。例如，如果数组是 [1,2,3,4] ，那么你可以对第一个元素执行此操作，使其变成 [2,2,3,4]

数组的 偏移量 是数组中任意两个元素之间的 最大差值 。返回数组在执行某些操作之后可以拥有的 最小偏移量 。

示例 1:

输入：nums = [1,2,3,4]
输出：1
解释：你可以将数组转换为 [1,2,3,2]，然后转换成 [2,2,3,2]，偏移量是 3 - 2 = 1

示例2:

输入：nums = [4,1,5,20,3]
输出：3
解释：两次操作后，你可以将数组转换为 [4,2,5,5,3]，偏移量是 5 - 2 = 3

示例3:

输入：nums = [2,10,8]
输出：3

提示：

• n == nums.length
• $2\le n\le 10^5$
• $1\le nums[i]\le 10^9$

vivo周赛第四题。vivo考了四个数组操作，还是有点东西的。这题当时并没有时间想，但估计想了也不会做。。

首先，看数据范围大概能知道这是贪心的，关键就是贪心策略了。发现每个数都可以乘2或者除以2，也就是说x和2x是等价的。所以建一个最大堆，对于偶数直接放进去，对于奇数，我们乘以2然后放进去，让所有数都从大往小变化。

检查堆的最大元素，如果是偶数，说明这个数还可以缩小，也就是缩小这个数组的偏移量。所以就变成x/2，再放进堆里，并更新此时的最小元素和结果res。如果是奇数，说明偏移量不能再小了。

为什么？首先，这个最大的奇数不能再小了，这是显然的。关键是最小的那个元素为什么不能乘以2呢？因为，能乘2的只能是奇数，而我们开始放的都是偶数，所以这个数至少已经在堆顶呆过一次了。我们的res只会越来越小，所以当这个数在堆顶时，当时的res只可能更大。

代码如下：

class Solution {
public:
    int minimumDeviation(vector<int>& nums) {
        priority_queue<int> pq;
        int minv = INT_MAX;
        for(int i = 0; i < nums.size(); i++) {
            int x = nums[i];
            if(x % 2) x *= 2;
            pq.push(x);
            minv = min(minv, x);
        }
        int res = pq.top() - minv;
        while(pq.top() % 2 == 0) {
            int x = pq.top();
            pq.pop();
            x /= 2;
            minv = min(minv, x);
            pq.push(x);
            res = min(res, pq.top() - minv);
        }
        return res;
    }
};