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

官方题解

思路
对于一个部分k-bag序列，一定存在一个α满足:

对于任意整数y都有[gk+ a, gk+公＋k-1]这一段区间里所有的数都不相等(这个区间可能不被[1,n]完全包含，此时指它和[1,n]的交）；
并且可以限制α取0到k一1之间的某个整数。容易发现只要上面的条件也是充要的。定义pre;表示最大的满足ap= ai;p<i的p，如果不存在则未定义。
对于一个有定义的pre：对α取值的限制为:存在一个值 j 满足 j mod k=a且 pre≤j<i；
由此合法的α将是某一段或某两段区间。
具体来说，当 prei ≤ i- k 时对 α没有限制;否则当pre mod k <i mod k时：
要求pre mod k≤ u <i mod k;
当pre mod k > i mod k时要求pre;
mod k≤ t<k或0≤t<i mod k。对于这些限制求交或者求补集的并即可判断是否存在一个合法的c。

我们寻找两个相邻的相同元素
则它们必属于两个不同排列
记它们相距长度为len[i] len[i]len[i]
枚举起点（无重复）之后往后面跳
如果有一处长度不是k kk就不行
另注：此题需要离散化，当k > n 时。

代码

#include<bits/stdc++.h>
#define ll long long
using namespace std;
int cnt,f;
int n,k,m;
int a[500010];
int b[500010];
int num[500010];
bool dp[500010];
int main()
{
    int t;
    int i,j;
    scanf("%d",&t);
    while(t--)
    {
        f=0;
        scanf("%d%d",&n,&k);
        for(i=1;i<=n;i++)
        {
            scanf("%d",&a[i]);
            if(a[i]>k) f=1;
        }
        if(f==1) printf("NO\n");
        else
        {
            m=0;
            for(i=1;i<=n;i++)
            b[m++]=a[i];
            sort(b,b+m);
            m=unique(b,b+m)-b;
            for(i=1;i<=n;i++)
                a[i]=lower_bound(b,b+m,a[i])-b+1;
            memset(num,0,sizeof(num));
            memset(dp,0,sizeof(dp));
            cnt=0;
            int x=n;
            for(i=n;i>=1;i--)
            {
                num[a[i]]++;
                if(num[a[i]]==2) cnt++;
                if(i+k<=n)
                {
                    num[a[i+k]]--;
                    if(num[a[i+k]]==1) cnt--;
                }
                if(i+k>n && cnt==0) dp[i]=1;
                else if(dp[i+k]&&cnt==0) dp[i]=1;
                if(dp[i]==1) x=i;
            }
            if(x>k)
            {
                printf("NO\n");
                continue;
            }
            memset(num,0,sizeof(num));
            for(i=1;i<x;i++)
            {
                if(num[a[i]]==1)
                {
                    printf("NO\n");
                    f=1;
                    break;
                }
                num[a[i]]=1;
            }
            if (f==0) printf("YES\n");
        }
    }
    return 0;
}