给定一个长度为n的数组，数组中的每个元素满足：1 <= a <= n；
求这个数组有多少个子区间，使得在这个子区间存在数字v，使得v在这个子区间出现的次数 >= m;

示例

输入
5 2
1 2 1 4 2

输出
4
解释：[1, 3], [1, 4], [1, 5], [2, 5]都存在一个数字使得这个数字出现次数 >= 2

示例2

输入
7 3
1 2 1 1 2 4 2
输出
5
解释：[1, 4], [1, 5], [1, 6], [1, 7], [2, 7]

解题思路：

• 用map来保存同一个数字出现的位置；
• 当把当前数字num加入map中的对应位置后，如果当前数字在这之前出现超过了m次，则从当前开始向前数m个当前数字num，得到这倒数第m个出现的当前数字num的位置index；
• 那么区间[1, i], [2, i], …, [index, i]都是满足条件的区间，因为都至少有一个数字出现次数 >= m;
• 在遍历的过程中，在当前数字之前可能已经有数字能满足条件了，因此要使用一个变量来存储当前数字之前出现超过m次的位置；

代码

Java

import java.util.*;

public class Main{

    private static List<List<Integer>> all = new ArrayList<>();

    public static void main(String[] args) {
        Scanner input = new Scanner(System.in);
        int n = input.nextInt();
        int m = input.nextInt();
        Map<Integer, List<Integer>> map = new HashMap<>();
        int res = 0;
        int pre = 0; // pre记录当前数字之前出现了m次的数字的第一次出现的位置
        for (int i = 1; i <= n; i++) {
            int num = input.nextInt();
            if (map.containsKey(num)) {
                List<Integer> list = map.get(num);
                list.add(i);
                if (list.size() >= m) {
                    int index = list.get(list.size() - m);
                    pre = Math.max(pre, index); //取当前出现超过m次的数字和之前出现超过m次的数字的位置的最大值，不然有课可能漏
                }
            } else {
                List<Integer> list = new ArrayList<>();
                list.add(i);
                map.put(num, list);
            }
            res += pre;
            for (int k = 1; k <= pre; k++) {
                List<Integer> temp = new ArrayList<>();
                temp.add(k);
                temp.add(i);
                all.add(temp);
            }
        }
        System.out.println(res);
        System.out.println(all);
    }
}

输出

5 2
1 2 1 4 2
4
[[1, 3], [1, 4], [1, 5], [2, 5]]


7 3
1 2 1 1 2 4 2
5
[[1, 4], [1, 5], [1, 6], [1, 7], [2, 7]]

C++

#include <iostream>
#include <vector>
#include <unordered_map>
#include <algorithm>
using namespace std;

int main() {
	int n, m;
	cin >> n >> m;

	unordered_map<int, vector<int>> maps;
	int res = 0;
	int pre = 0;
	vector<vector<int>> all;
	for (int i = 1; i <= n; i++) {
		int num;
		cin >> num;
		if (maps.find(num) != maps.end()) {
			auto list = maps.find(num);
			list->second.push_back(i);
			if (list->second.size() >= m) {
				int index = list->second[list->second.size() - m];
				pre = max(pre, index);
			}
		}
		else {
			vector<int> indexs;
			indexs.push_back(i);
			maps[num] = indexs;
		}
		res += pre;
		for (int k = 1; k <= pre; k++) {
			all.push_back({ k, i });
		}
	}
	cout << res << endl;
	for (int i = 0; i < all.size(); i++) {
		cout << all[i][0] << " " << all[i][1] << endl;
	}
	return 0;
}

输出

5 2
1 2 1 4 2
4
1 3
1 4
1 5
2 5


7 3
1 2 1 1 2 4 2
5
1 4
1 5
1 6
1 7
2 7