时间限制: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;
}