QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#589706 | #7942. $K$ Subsequences | eroinevs | WA | 16ms | 3660kb | C++14 | 1.5kb | 2024-09-25 19:48:23 | 2024-09-25 19:48:24 |
Judging History
answer
#include <bits/stdc++.h>
#define ll long long
#define int long long
#define ff first
#define ss second
#define PI pair<int, int>
#define rep(i, n) for(int i = 0; i < (n); ++i)
#define bust ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
using namespace std;
const ll INF = (ll)1e9 + 7, mod = 998244353;
void solve(){
int n, k;
cin >> n >> k;
int a[n], cnt1 = 0, cnt2 = 0, q = 0;
for(auto&e: a) cin >> e;
int ind = 0;
cout << ind+1 << ' ';
queue<int> stp, stn; // stack ends with 1, ends with -1
for(int i = 1; i < k; i++){
stp.push(i);
}
for(int i = 1; i < n; ++i){
if(a[i] == a[i-1]){
if(a[i] == 1){
stp.push(ind);
if(!stn.empty()){
ind = stn.front();
stn.pop();
}
else{
ind = stp.front();
stp.pop();
}
}
else{
stn.push(ind);
if(!stp.empty()){
ind = stp.front();
stp.pop();
}
else{
ind = stn.front();
stn.pop();
}
}
ind %= k;
}
cout << ind+1 << ' ';
}
cout << '\n';
}
signed main() {
bust
int t;
cin >> t;
while (t--) {
solve();
}
return 0;
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 0ms
memory: 3660kb
input:
5 3 2 1 -1 1 4 2 -1 1 1 -1 7 3 1 1 1 1 1 1 1 10 3 1 1 1 1 -1 -1 1 1 1 1 12 4 1 1 1 1 -1 -1 -1 -1 1 1 1 1
output:
1 1 1 1 1 2 2 1 2 3 1 2 3 1 1 2 3 1 1 2 2 1 3 2 1 2 3 4 4 1 2 3 3 4 1 2
result:
ok Correct (5 test cases)
Test #2:
score: -100
Wrong Answer
time: 16ms
memory: 3560kb
input:
18434 10 1 -1 1 1 -1 -1 1 -1 -1 1 1 10 2 -1 -1 -1 1 1 -1 1 1 1 1 10 2 1 -1 -1 -1 -1 1 1 -1 1 1 10 7 1 1 -1 1 -1 1 1 -1 -1 1 9 1 -1 1 -1 1 1 -1 1 -1 1 8 1 -1 -1 -1 -1 1 1 -1 -1 10 3 -1 -1 -1 1 1 1 1 -1 -1 -1 9 1 1 -1 -1 1 -1 -1 -1 -1 -1 10 10 -1 1 1 1 1 1 1 1 1 1 10 4 -1 1 -1 1 -1 1 1 -1 1 1 9 3 1 1 ...
output:
1 1 1 1 1 1 1 1 1 1 1 2 1 1 2 2 2 1 2 1 1 1 2 1 2 2 1 1 1 2 1 2 2 2 2 2 3 3 4 4 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 2 3 3 1 2 3 3 1 2 1 1 1 1 1 1 1 1 1 1 1 2 3 4 5 6 7 8 9 1 1 1 1 1 1 2 2 2 3 1 2 2 3 3 3 3 3 1 1 2 2 2 3 4 4 4 1 1 2 3 4 5 6 7 7 7 1 1 2 2 2 3 3 1 1 1 1 1 1 1 1 1 1 1 1 2...
result:
wrong answer Jury found better answer than participant (test case 339)