题意:给你一段长度为n的序列判断是否为K-bag的子序列,定义K-bag为k个数的全排列
思路:这题的思路有很多,有hash+map,有dp,也有差分。说一下差分把,首先对与这段序列,有开头一定在1~min(k,n)之间,用一个数组维护这段区间,如果这个数组t[i]的前缀和等于0,那么就表示这个位置可以作为开头,反之则不能,基于这个结论,对两个相同的数字之间的的距离进行分类,如果两个数字之间的距离>=k,即表示1~k都是可以作为开头的,如果小于k,那么就表示这个区间必有一个位置为开头,如果 这个序列的 第一个区间开头是 x1,那么后面的区间开头一定是 x1+y*k,y是一个整数,那么 我只要把当前开头 % k 就可以得到第一区间的开头,这个值可以加到第一个区间开头。这个区间值 +1 的操作可以用差分数组维护
具体代码如下:
#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
#include <vector>
#include <cmath>
#include <queue>
#include <set>
#include <map>
#include <stack>
#include <unordered_map>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int, int> pii;
#define rep(i, a, n) for(int i = a; i <= n; i++)
#define per(i, a, n) for(int i = n; i >= a; i--)
#define IOS std::ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
#define fopen freopen("file.in","r",stdin);freopen("file.out","w",stdout);
#define fclose fclose(stdin);fclose(stdout);
#define PI 3.14159265358979323846
const int inf = 1e9;
const ll onf = 1e18;
const int maxn = 5e5+10;
inline int read(){
int x=0,f=1;char ch=getchar();
while (!isdigit(ch)){if (ch=='-') f=-1;ch=getchar();}
while (isdigit(ch)){x=(x<<3)+(x<<1)+ch-48;ch=getchar();}
return x*f;
}
int t[maxn], a[maxn], ord[maxn];
bool cmp(int x, int y){
return a[x] < a[y];
}
inline bool cf(){
int n=read(), k=read();
int flag = 1;
for(int i = 1; i <= n; i++){
a[i]=read();
if(a[i]>k||a[i]<1) flag = 0;
}
if(!flag) return 0;
for(int i = 1; i <= n; i++) ord[i]=i;
stable_sort(ord+1, ord+n+1, cmp);
int sz = min(n, k);
for(int i = 1; i <= sz; i++) t[i] = 0;
for(int i = 2; i <= n; i++){
if(a[ord[i]]==a[ord[i-1]]){
int l = ord[i-1], r=ord[i]-1;
if(r-l+1>=k) continue;
t[1]--;
l = l%sz, r = r%sz;
if(l==0) l=sz; if(r==0) r=sz;
if(l<=r) t[l]++, t[r+1]--;
else t[1]++, t[r+1]--, t[l]++, t[sz+1]--;
}
}
for(int i = 1; i <= sz; i++){
t[i]+=t[i-1];
if(t[i]==0) return 1;
}
return 0;
}
signed main(){
int t = read();
while(t--){
if(cf()){
puts("YES");
}else puts("NO");
}
return 0;
}