2022暑期牛客第九场

"蔚来杯"2022牛客暑期多校训练营9

A.Car Show

题目大意

给定一个长为 $n$ 的包含 $1,2,\dots ,m$ 的序列，求有多少区间 $[L,R]$ 包含所有 $1,2,\dots ,m$ 。

解法1：

思路

对于序列 $[L,R]$ ,若存在满足条件的区间，则 $[L,R]$ ~ $[L,N]$ 一定满足题意，记录序列数字种类数以及每个数出现的次数，使用双指针遍历即可。

代码

#include <iostream>
#include <climits>
#include <string.h>
#include <string>
#include <stdio.h>
#include <queue>
#include <vector>
#include <map>
#include <set>
#include <list>
#include <stack>
#include <algorithm>
#include <math.h>
#include <cctype>
#include <stdlib.h>
#include <unordered_map>
#define INTINF 0x3f3f3f3f
#define LLINF  0x3f3f3f3f3f3f3f3f
#define N 2000010
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
inline ll read()
{
ll s=0,f=1;
char c=getchar();
for(;c<'0'||c>'9';c=getchar()) f=-1;
for(;c>='0'&&c<='9';c=getchar())
{
s=s*10+c-'0';
}
return s*f;
}

ll n,m;
ll a[N];
ll vis[N];
ll l=1,r=0;
int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        cin>>a[i];
    }
    ll sum=0;
    ll cnt=0;
    while(r<=n)
    {
        if(cnt>=m)
        {
            sum+=n-r+1;
            vis[a[l]]--;
            if(!vis[a[l++]]) cnt--;
        }
        else
        {
            vis[a[++r]]++;
            if(vis[a[r]]==1) cnt++;
        }
    }
    cout<<sum<<endl;

    system("pause");
    return 0;
}

解法2：

思路

枚举左端点，合法的右端点集合一定是区间 $[r,n]$，且 $r$ 随着 $l$ 的递增而不减。在移动指针的同时维护区间内每种数字的个数以及出现数字的种类数即可。

代码

void solve() {
    scanf("%d%d", &n, &m);
    for (int i = 1; i <= n; i++) scanf("%d", &a[i]);
    long long ans = 0;
    int k = 0;
    for (int i = 1, j = 1; i <= n; i++) {
        while (j <= n && k < m) {
            if (!cnt[a[j]]) k++;
            cnt[a[j]]++;
            j++;
        }
        if (k == m) ans += n - j + 2;
        cnt[a[i]]--;
        if (!cnt[a[i]]) k--;
    }
    printf("%lld\n", ans);
}