牛客網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.
翻譯:
當一個數列可以表示為若干個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"); } } }//碼農教程請不要爬我程式碼