【C++竞赛 A】xxx的项链

时间限制：2s 内存限制：64MB
问题描述
xxx有一个长度为n的宝石链，宝石有m种不同的颜色。xxx想截取其中连续的一段做一个项链。为了让项链更漂亮，xxx希望项链中的宝石包含所有颜色。
输入描述
第一行一个整数T(1≤T≤10)表示数据组数。
对于每组数据第一行两个整数n(1≤n≤〖10〗^5),m(1≤m≤30)
第二行n个整数C_1,C_2,…,C_n (0≤C_im)表示宝石链上每个宝石的颜色
输出描述
对于每组数据输出一行：一个整数，如果存在满足条件的方案，输出满足条件的最短长度，否则输出-1。
输入样例
2
4 3
0 1 2 0
4 3
0 1 1 0
输出样例
3
-1
样例解释
样例一: 截取[0 1 2],[1 2 0],[0 1 2 0]均满足条件，因此输出3
样例二: 没有满足条件的，输出-1

【题目链接】:
【题解】

枚举区间的左端点i,然后二分右端点；
用前缀和来记录到某个位置所有的颜色的贝壳的拥有情况;
右端点减去左端点的拥有情况就是中间的拥有情况;

【完整代码】

#include <bits/stdc++.h>#define rep1(i,a,b) for (int i = a;i <= b;i++)using namespace std;#define pb push_back;const int MAXN = 1e5+10;int n,q,m;int pre[MAXN][34],temp[34];bool ok(int l,int r){    int cnt = m;    rep1(i,0,m-1)    {        temp[i] = pre[r][i]-pre[l-1][i];        if (temp[i]==0)        {            cnt--;            return false;        }    }    if (cnt==m)        return true;    else        return false;}int main(){    //freopen("D:\\rush.txt","r",stdin);    int T;    scanf("%d",&T);    while (T--)    {        int ans = 21e8;        memset(pre,0,sizeof(pre));        scanf("%d%d",&n,&m);        rep1(i,1,n)        {            int x;            scanf("%d",&x);            pre[i][x]++;            for (int j = 0;j<=m-1;j++)                pre[i][j] += pre[i-1][j];        }        rep1(i,1,n)        {            int l = i,r = n,ans1=-1;            while (l<=r)            {                int m = (l+r)>>1;                if (ok(i,m))                    ans1=m,r = m-1;                else                    l = m+1;            }            if (ans1!=-1)                ans = min(ans,ans1-i+1);        }        if (ans==21e8)            puts("-1");        else            printf("%d\n",ans);    }    return 0;}