0419 晚 阿里笔试

做着玩

T1

题目描述

定义一个数组的权值为:众数的出现次数。例如，[2,3,3,6,6,6,1,2]的权值为3。因为众数6出现了3次。现在给定个人长度为 \(n\) 的数组，请你求出权值不小于的连续 \(k\) 的数组数量。

输入描述

第一行输入两个正整数\(n\) 和 \(k\)
第二行输入 \(n\) 个正整数 \(a_i\)
\(1 \leq k \leq n \leq 200000\)
\(1\leq a_i \leq 10^9\)

输出

一个整数，代表权值不小于\(k\)的连续子数组数量

代码

点击查看代码

#include <bits/stdc++.h>
const int N = 5e5+9;

using namespace std;
map<int,int> mp;
int a[N];
bool cmp(const pair<int,int> A, const pair<int,int> B) {
	return A.second < B.second;
}
int main() {
	int n,k;
	cin>>n>>k;
	for(int i=1; i<=n; ++i)	cin>>a[i];
	int l=1,r=0,ans=0;
	for(int i=1; i<=n; ++i){
		while(r<n){
			r++;
			mp[a[r]]++;
			auto zs = max_element(mp.begin(), mp.end(), cmp);
//			cout<<l<<" "<<r<<" "<<zs->second<<" "<<endl;
			if(zs->second>=k) {
				ans+=(n-r+1);
				break;
			}
		}
		mp[a[i]]--;
	}
	cout<<ans<<'\n';
    return 0;
}
/*
5 2
2 3 2 3 1

*/