Leetcode2808. 使循环数组所有元素相等的最少秒数

Every day a Leetcode

题目来源:2808. 使循环数组所有元素相等的最少秒数

解法1:哈希 + 枚举

假设答案是数组 nums 的一个元素 x,因为每个 x 每一秒可以把它旁边的两个数变成 x,所以如果两个 x 之间还有 k 个元素,那么 ⌈k/2⌉ 秒之后这些元素就可以全部变成 x。

因此把答案变成 x 需要的时间,就是两个 x 之间最多有几个元素,把元素数量除以 2 上取整。枚举所有 x,选出最优答案即可。

代码:

/*
 * @lc app=leetcode.cn id=2808 lang=cpp
 *
 * [2808] 使循环数组所有元素相等的最少秒数
 */

// @lc code=start
class Solution
{
public:
    int minimumSeconds(vector<int> &nums)
    {
        // 特判
        if (nums.size() <= 1)
            return 0;

        int n = nums.size();
        // mp[x] 表示所有等于 x 的元素下标
        unordered_map<int, vector<int>> mp;
        for (int i = 0; i < n; i++)
            mp[nums[i]].push_back(i);

        int ans = INT_MAX;
        for (auto &[_, vec] : mp)
        {
            // 计算最终元素为 x = p.first 的情况下需要的时间
            int max_gap = 0;
            for (int i = 0; i < vec.size(); i++)
            {
                // 计算两个 x 之间的元素数量
                int gap = 0;
                // 注意序列是环形的
                if (i + 1 < vec.size())
                    gap = vec[i + 1] - vec[i] - 1;
                else
                    gap = n - 1 - vec[i] + vec[0];
                max_gap = max(max_gap, gap);
            }
            // 需要的时间,就是元素的最大间隔,除以 2 上取整
            ans = min(ans, (max_gap + 1) / 2);
        }
        return ans;
    }
};
// @lc code=end

结果:

Leetcode2808. 使循环数组所有元素相等的最少秒数_第1张图片

复杂度分析:

时间复杂度:O(n),其中 n 是数组 nums 的元素个数。

空间复杂度:O(n),其中 n 是数组 nums 的元素个数。

你可能感兴趣的:(Every,day,a,LeetCode,C++,数据结构与算法,leetcode,哈希,枚举)