链接:登录—专业IT笔试面试备考平台_牛客网
来源:牛客网
题目描述
小红拿到了一个数组,她想知道有多少个长度为k的连续子数组是排列?
定义排列为:1到k每个元素都恰好出现了1次。
输入描述:
第一行输入两个正整数n,k,代表小红拿到的数组大小和连续子数组的大小。 第二行输n个正整数ai,代表数组中的元素。
输出描述:
一个整数,代表长度为k的连续子数组是排列的数量。
示例1
输入
5 2 1 2 1 1 2
输出
3
说明
有3个长度为2的连续子数组是排列:两个[1,2]和一个[2,1]
解题思路:
使用滑动窗口维护一个长度为k的区间即可
AC代码:
#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10;
int n,k,cnt;
int a[N];
int main(){
ios_base::sync_with_stdio(false);
cin>>n>>k;
for(int i=0;i<n;i++)
cin>>a[i];
unordered_map<int,int> freq; //使用哈希表来记录窗口内每个数字的出现次数
int dis = 0; //记录窗口内不同数字的个数
for(int i=0;i<n;i++){
if(freq[a[i]]==0) //将a[i]添加到哈希表中,并更新dis
dis++;
freq[a[i]]++;
if(i>=k){ //如果窗口大小大于k,移除窗口左侧的元素
freq[a[i-k]]--;
if(freq[a[i-k]]==0)
dis--;
}
if(i>=k-1&&dis==k) //如果窗口大小等于k,检查窗口是否满足条件
cnt++;
}
cout<<cnt;
return 0;
}