2020牛客暑期多校训练营（第六场）K-Bag

题目描述

分析

题目大意是K-bag序列定义为由多个1-k的排列顺序连接起来的序列，想问你给定序列是不是k-bag的连续子序列。
那么我们很容易知道，对于序列中每个完整的K序列，有且只有一个某个元素
那么我们可以记录每一个元素，到后面第一个开始重复的元素的距离
然后循环第一对重复元素之间的距离，每次循环以此为基准，一个个地挖掉K序列，直到最后，如果可行，标记。
最后输出即可
~~~~~~~~~~~~~~~~~~~~~~~~ 分 ~~ 割 ~~ 线 ~~~~~~~~~~~~~~~~~~~~~~~~
交了之后WA了，然后发现K太大，考虑离散化。
又发现 $\large{a_i}$可能大于 $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");
	}
}