CF2157C Meximum Array 2
题意
构造长度为 $n$ 的序列 $a$,满足 $q$ 个限制条件。
思路
对于 $a_i$:
- 若 $a_i$ 被 min 限制且被 MEX 限制,则易得填 $k+1$ 即可;
- 若 $a_i$ 仅被 min 限制,则填 $k$ 即可。
- 若 $a_i$ 不属于以上两种,则循环填 $1$ 到 $k-1$。
代码
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61
| #include<bits/stdc++.h> #define fir first #define sec second #define int long long #define pii pair<int,int> #define fep(i,s,e) for(int i=s;i<e;i++) #define pef(i,s,e) for(int i=s;i>e;i--) #define rep(i,s,e) for(int i=s;i<=e;i++) #define per(i,s,e) for(int i=s;i>=e;i--) namespace FastIO{ template<typename T>inline void read(T &x){ x=0;int f=1;char c=getchar(); for(;!isdigit(c);c=getchar())if(c=='-')f=-1; for(;isdigit(c);c=getchar())x=(x<<1)+(x<<3)+(c^48);x*=f; } template<typename T,typename...Args> inline void read(T &x,Args&...args){ read(x); read(args...); } template<typename T>void print(T x){ if(x<0)x=-x,putchar('-'); if(x>9)print(x/10); putchar((x%10)^48); } } using namespace std; using namespace FastIO; const int INF=1e18; const int maxn=2e5+10; signed main(){ int t;read(t); while(t--){ int n,k,q;read(n,k,q); vector<int>a(n+1,0); vector<int>mn(n+1,0),mx(n+1,0); while(q--){ int c,l,r;read(c,l,r); rep(i,l,r){ if(c==1)mn[i]=1; else mx[i]=1; } } rep(i,1,n){ if(mn[i]&&mx[i]){ a[i]=k+1; } else if(mx[i]){ a[i]=i%k; } else{ a[i]=k; } } rep(i,1,n){ print(a[i]); putchar(' '); } putchar('\n'); } }
|