牛客暑假第六场 K-bag 差分+思维

题意：给你一段长度为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;
}