牛客周赛Round 43 D-小红的子数列排列判断

链接：登录—专业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;
}