1. 程式人生 > 實用技巧 >牛客網K-Bag

牛客網K-Bag

題目描述:

連結:https://ac.nowcoder.com/acm/contest/5671/K
來源:牛客網
A sequence is called kkk-bag, if and only if it is put in order by some (maybe one) permutations of111to kkk. For example,1,2,3,2,1,3,3,2,11,2,3,2,1,3,3,2,11,2,3,2,1,3,3,2,1 is a valid333-bag sequence.
Roundgod is not satisfied with kkk-bag, so she put forward part-kkk-bag, whichis a contiguous subsequence of kkk-bag.

Wcy wants to know if the sequence of lengthnnn is a part-kkk-bag sequence.

翻譯:

當一個數列可以表示為若干個1到k的排列依次組成時,這個數列被稱為k-bag。例如1,2,3,2,1,3,3,2,1是一個3-bag。

如果一個序列是一個k-bag的連續子串,則其稱為part-k-bag。

求一個長度為n的序列是否是一個part-k-bag。

輸入描述:

翻譯:

第一行包含一個整數T(1≤T≤20),表示測試用例的數量。

然後是T個樣例。每個測試案例的第一行包含兩個整數n,k(1≤n≤5⋅10^5,1≤k≤10^9)。

每個測試案例的第二行包含n個整數表示序列。保證∑n≤2⋅10^6,序列的值在1到10 ^ 9之間。

輸出描述:

翻譯:

如果一個序列是部分k-bag序列,則列印“YES”,否則列印“NO”。

解析:

對於一個部分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。

另注:此題需要離散化,當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;
    scanf("%d",&t);
    while(t--)
    {
        f=0;
        scanf("%d%d",&n,&k);
        for(int 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 (int i=1; i<=n; i++)
            b[m++]=a[i];
            sort(b,b+m);
            m=unique(b,b+m)-b;
            for(int 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(int 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(int 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");
        }
    }
}//碼農教程請不要爬我程式碼