题目来源:2808. 使循环数组所有元素相等的最少秒数
假设答案是数组 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
结果:
复杂度分析:
时间复杂度:O(n),其中 n 是数组 nums 的元素个数。
空间复杂度:O(n),其中 n 是数组 nums 的元素个数。