在遍历哈希表的时候,我一般采取的是iterator的方法来遍历,这样遍历很不方便,在leetcode 697题中,了解了一种新的哈希表遍历方法,这种基于auto关键字的遍历应该是在C++ 11中支持的,非常方便。
题干如下:
题目本身不难,我们可以采用哈希表来记录数组中每个元素出现的频率,当出现频率最高的元素是唯一的时候,我们只要取该元素第一次出现和最后一次出现的区间长度,返回该区间的长度。当出现频率最高的元素不唯一时,我们需要比较他们的区间长度,取最小的区间长度返回。
同时,每个元素的区间长度等于该元素最后一次出现的位置下标减去第一次出现的位置下标再+1。
哈希表的结构为[key, value],其中,key为元素之,value为一个有三个元素的数组,数组0存储频率,1存储第一次出现的位置下标,2存储最后一次出现的下标。
代码如下:
class Solution {
public: //力扣官方解法,可以参考
int findShortestSubArray(vector<int>& nums) {
int n = nums.size();
unordered_map<int,vector<int>> mp;
for(int i = 0;i<n;i++){
if(mp.count(nums[i])){ //已经在哈希表中存在,则更新频率和最后一次出现的位置下标
mp[nums[i]][0]++;
mp[nums[i]][2] = i;
}else{ //还没有在哈希表中出现,设定初值
mp[nums[i]] = {1,i,i};
}
}
int ans = 0; //储存度
int maxFreq = 0;
for(auto& [_ , vec] : mp){
if(vec[0] > maxFreq){ //找到了更大的频率元素
maxFreq = vec[0];
ans = vec[2] - vec[1] + 1;
}else if(vec[0] == maxFreq){ //频率相等去更短者
ans = min(ans,vec[2] - vec[1] + 1);
}
}
return ans;
}
};