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');
}
}