技術標籤：LeetCode

文章目錄

• • 1. 題目
  • 2. 解題

1. 題目

給出一個正整數陣列 nums，請你幫忙從該陣列中找出能滿足下面要求的 最長 字首，並返回其長度：

• 從字首中 刪除一個 元素後，使得所剩下的每個數字的出現次數相同。

如果刪除這個元素後沒有剩餘元素存在，仍可認為每個數字都具有相同的出現次數（也就是 0 次）。

示例 1：
輸入：nums = [2,2,1,1,5,3,3,5]
輸出：7
解釋：對於長度為 7 的子陣列 [2,2,1,1,5,3,3]，
如果我們從中刪去 nums[4]=5，
就可以得到 [2,2,1,1,3,3]，裡面每個數字都出現了兩次。

示例 2
 
：
輸入：nums = [1,1,1,2,2,2,3,3,3,4,4,4,5]
輸出：13

示例 3：
輸入：nums = [1,1,1,2,2,2]
輸出：5

示例 4：
輸入：nums = [10,2,8,9,3,8,1,5,2,3,7,6]
輸出：8
 
提示：
2 <= nums.length <= 10^5
1 <= nums[i] <= 10^5

來源：力扣（LeetCode）
連結：https://leetcode-cn.com/problems/maximum-equal-frequency
著作權歸領釦網路所有。商業轉載請聯絡官方授權，非商業轉載請註明出處。

2. 解題

class Solution {
public:
    int maxEqualFreq(vector<int>& nums) {
        unordered_map<int,int> m;//數字，數字的個數freq
        map<int,int> size_count;//數字的個數freq，freq的頻數
        int maxlen = 1, MINf, MAXf;
        for(int i = 0; i < nums.size(); ++i) 
        {
            m[
 
nums[i]]++;
            size_count[m[nums[i]]]++;
            if(--size_count[m[nums[i]]-1] <= 0)
                size_count.erase(m[nums[i]]-1);
            MINf = size_count.begin()->first;
            MAXf = size_count.rbegin()->first;
            // 1, [1,2,3,4,5] size_count[1] = 5
            // 2, [2,2,2,2] size_count[4] = 1
            // 3, [2,2,1,1,0,0,3,3,3] size_count[2] = 3, size_count[3] = 1
            // 4, [1,2,2,2,3,3,3] size_count[1] = 1, size_count[3] = 2
            // 這4種情況才可以
            if((size_count.size()==1 && (MINf==1 || size_count[MINf]==1))||(size_count.size()==2 && ((size_count[MAXf]==1 && MAXf-MINf==1) || (size_count[MINf]==1 && MINf==1))))
                maxlen = i+1;
        }
        return maxlen;
    }
};

604 ms 68.6 MB C++

我的CSDN部落格地址 https://michael.blog.csdn.net/

長按或掃碼關注我的公眾號（Michael阿明），一起加油、一起學習進步！