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

原题链接: link.

题目大意：连续数个1到k的排列拼起来的序列被称为k-bag,例如1 2 3 3 2 1 2 3 1就是3-bag，一个k-bag的连续子序列称为part-k-bag，现在给出一个序列和k，要判断这个序列是否是part-k-bag

思路：
(1)如果一个序列是part-k-bag，那么它去掉一个前缀(长度可为0)和一个后缀(长度可为0)以后就是k-bag序列，去掉的前缀长度小于k，且不含重复数字
(2)反过来，一个k-bag序列左端和右端各加上一个不含重复数字(且都小于等于k)的序列后就是一个part-k-bag
(3)如果一个序列有大于k的数字，直接pass
(4)判断每个位置能否作为一个1到k的排列的起点，用一个数组保存true和false，只要连续k个数字都不相同(且都小于等于k)即满足要求，如果到了序列末尾都没有重复数字也算满足要求
(5)忽略一段前缀(长度由0到k-1且不含重复数字)，如果从下一个位置开始每隔k个位置都符合(4)的要求，那这个序列就是part-k-bag
(6)枚举前缀的所有情况，如果没有任何一种情况符合要求，这个序列就不是part-k-bag

代码：

#define _USE_MATH_DEFINES
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <string>
#include <math.h>
#include <ctype.h>
#include <vector>
#include <stack>
#include <map>
#include <set>
#include <algorithm>
#include <iostream>
#include <limits.h>
#include <iomanip>
#include <queue>
#include <functional>
#define M 500005
const long long N = 1e9+7;
#define ll long long
using namespace std;

int a[M];
int fullk[M];
bool visit[M];
int n,k;
set<int> s;
bool check()
{
	
	memset(fullk,0,sizeof(fullk));
	if(k>500000)
		s.clear();
	else
		memset(visit,false,sizeof(visit));
	int in=0;
    int first=0;
	for(int i=0;i<n;i++){
		if(k>500000){
			while(in<n&&!s.count(a[in])){	
				s.insert(a[in]);
				in++;
			}
		}
		else{
			while(in<n&&!visit[a[in]]){	
				visit[a[in]]=true;
				in++;
			}
		}
		if(in-i==k || in>=n)fullk[i]=true;
        if(i==0)first=in;
		if(k>500000)s.erase(a[i]);
		else visit[a[i]]=false;
	}
	bool flag=true;
	for(int i=0;i<k&&i<=first;i++){
		flag=true;
		for(int j=0;i+j*k<n;j++){
			if(!fullk[i+j*k]){
				flag=false;
				break;
			}
		}
		if(flag)break;
	}
	return flag;
}


int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
 
    //freopen("test14.in","r",stdin);
    //freopen("test14.out","w",stdout);
	int t;
	cin>>t;
	while(t--){
		cin>>n>>k;
		bool flag=true;
		for(int i=0;i<n;i++){
			cin>>a[i];
			if(a[i]>k)flag=false;
		}
		if(!flag)cout<<"NO\n";
		else{
			flag=check();
			if(!flag)cout<<"NO\n";
			else cout<<"YES\n";
		}
	}
    return 0;
}