题目描述
分析
题目大意是K-bag序列定义为由多个1-k的排列顺序连接起来的序列,想问你给定序列是不是k-bag的连续子序列。
那么我们很容易知道,对于序列中每个完整的K序列,有且只有一个某个元素
那么我们可以记录每一个元素,到后面第一个开始重复的元素的距离
然后循环第一对重复元素之间的距离,每次循环以此为基准,一个个地挖掉K序列,直到最后,如果可行,标记。
最后输出即可
~~~~~~~~~~~~~~~~~~~~~~~~ 分 ~~ 割 ~~ 线 ~~~~~~~~~~~~~~~~~~~~~~~~
交了之后WA了,然后发现K太大,考虑离散化。
又发现
可能大于
。于是特判。
然后就AC了。
代码
#include<bits/stdc++.h>
#define ll long long
using namespace std;
ll dp[500010],a[500010],f[500010],n,m,i,j,k,l,tot,cnt,top,o,p,T;
int main()
{
for (scanf("%lld",&T);T--;)
{
scanf("%lld%lld",&n,&k);
memset(dp,0,sizeof(dp));
memset(a,0,sizeof(a));
memset(f,0,sizeof(f));
ll fl=0;
for (i=1;i<=n;i++)
{
scanf("%lld",&a[i]);
f[i]=a[i];
if (a[i]>k) fl=1;
}
if (fl)
{
puts("NO");
continue;
}
fl=0;
sort(f+1,f+n+1);
cnt=unique(f+1,f+n+1)-1-f;
for (i=1;i<=n;i++)
a[i]=lower_bound(f+1,f+cnt+1,a[i])-f;
cnt=0;
memset(f,0,sizeof(f));
p=1;
for (i=1;i<=n;i++)
{
while(!f[a[p]] && p<=n)
f[a[p]]++,p++;
f[a[i]]--;dp[i]=p-i;
}
for (i=1;i<=min(k,dp[1]+1);i++)
{
ll dd=1;
for (j=i;j<=n;j+=k)
{
if (j+dp[j]>=n+1) continue;
else if (dp[j]^k)
{
dd=0;
break;
}
}
if (dd)
{
fl=1;break;
}
}
if (fl) puts("YES");
else puts("NO");
}
}