interviewalgo
, , ,
Source
Edit
History

697. 数组的度

Introduction

Question: 697. 数组的度

给定一个非空且只包含非负数的整数数组 nums，数组的度的定义是指数组里任一元素出现频数的最大值。

你的任务是在 nums 中找到与 nums 拥有相同大小的度的最短连续子数组，返回其长度。

 

示例 1：

输入：[1, 2, 2, 3, 1]
输出：2
解释：
输入数组的度是2，因为元素1和2的出现频数最大，均为2.
连续子数组里面拥有相同度的有如下所示:
[1, 2, 2, 3, 1], [1, 2, 2, 3], [2, 2, 3, 1], [1, 2, 2], [2, 2, 3], [2, 2]
最短连续子数组[2, 2]的长度为2，所以返回2.

示例 2：

输入：[1,2,2,3,1,4,2]
输出：6

 

提示：

  • nums.length 在1到 50,000 区间范围内。
  • nums[i] 是一个在 0 到 49,999 范围内的整数。

解法一

Analysis

首先算出整个数组的度，然后使用双指针来确定当前区间[i:j]，并动态的更新度：

  1. 先移动j，直到当前的度等于整个数组的度
  2. 移动i，直到当前的度小于整个数组的度，并计算长度
  3. 返回第一步

Implement

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
int findShortestSubArray(vector<int>& nums) {
    unordered_map<int,int> imap;
    for(auto i: nums) imap[i]++;
    int d = 0;
    for(auto &p: imap) d = max(d, p.second);

    imap.clear();
    int td = 0;
    int res = nums.size();
    for(int i = 0, j = 0;j < nums.size(); j++) {
        imap[nums[j]]++;
        td = max(imap[nums[j]], td);
        if (td == d) {
            while (i < j && nums[i] != nums[j])
            {
                imap[nums[i]]--;
                i++;
            }
            // cout << i << " " << j << endl;
            res = min(res, j - i + 1);
            imap[nums[i]]--;
            td--;
            i++;
        }
    }
    return res;

}