| Task: | Chair Game |
| Sender: | Sofie Fu |
| Submission time: | 2024-03-19 17:45:31 +0200 |
| Language: | C++ (C++20) |
| Status: | READY |
| Result: | 17 |
| group | verdict | score |
|---|---|---|
| #1 | ACCEPTED | 8 |
| #2 | ACCEPTED | 5 |
| #3 | ACCEPTED | 4 |
| #4 | WRONG ANSWER | 0 |
| #5 | WRONG ANSWER | 0 |
| #6 | WRONG ANSWER | 0 |
| #7 | WRONG ANSWER | 0 |
| #8 | WRONG ANSWER | 0 |
| test | verdict | time | group | |
|---|---|---|---|---|
| #1 | ACCEPTED | 0.01 s | 1, 7, 8 | details |
| #2 | ACCEPTED | 0.01 s | 1, 7, 8 | details |
| #3 | ACCEPTED | 0.03 s | 1, 7, 8 | details |
| #4 | ACCEPTED | 0.01 s | 1, 7, 8 | details |
| #5 | ACCEPTED | 0.04 s | 1, 7, 8 | details |
| #6 | WRONG ANSWER | 0.01 s | 7, 8 | details |
| #7 | WRONG ANSWER | 0.01 s | 7, 8 | details |
| #8 | ACCEPTED | 0.01 s | 2, 8 | details |
| #9 | ACCEPTED | 0.01 s | 3, 4, 5, 6, 8 | details |
| #10 | ACCEPTED | 0.01 s | 3, 4, 5, 6, 8 | details |
| #11 | ACCEPTED | 0.01 s | 3, 4, 5, 6, 8 | details |
| #12 | ACCEPTED | 0.02 s | 3, 4, 5, 6, 8 | details |
| #13 | WRONG ANSWER | 0.01 s | 4, 5, 6, 7, 8 | details |
| #14 | WRONG ANSWER | 0.01 s | 4, 5, 6, 8 | details |
| #15 | WRONG ANSWER | 0.01 s | 4, 5, 6, 8 | details |
| #16 | WRONG ANSWER | 0.01 s | 4, 5, 6, 8 | details |
| #17 | WRONG ANSWER | 0.01 s | 5, 6, 7, 8 | details |
| #18 | WRONG ANSWER | 0.01 s | 5, 6, 8 | details |
| #19 | ACCEPTED | 0.01 s | 5, 6, 8 | details |
| #20 | WRONG ANSWER | 0.01 s | 5, 6, 8 | details |
| #21 | ACCEPTED | 0.01 s | 1, 6, 7, 8 | details |
| #22 | WRONG ANSWER | 0.01 s | 6, 7, 8 | details |
| #23 | WRONG ANSWER | 0.01 s | 6, 8 | details |
| #24 | WRONG ANSWER | 0.01 s | 6, 8 | details |
| #25 | WRONG ANSWER | 0.02 s | 8 | details |
| #26 | WRONG ANSWER | 0.02 s | 8 | details |
| #27 | ACCEPTED | 0.01 s | 3, 4, 5, 6, 8 | details |
| #28 | WRONG ANSWER | 0.01 s | 8 | details |
| #29 | WRONG ANSWER | 0.02 s | 8 | details |
| #30 | WRONG ANSWER | 0.02 s | 8 | details |
Code
#include <bits/stdc++.h>
using namespace std;
#define int long long
#define uset unordered_set
#define vo vector
#define pb push_back
#define sz(x) x.size()
#define fi first
#define se second
typedef long long ll;
typedef pair<int ,int> pii;
typedef vo<int> vi;
#define all(x) x.begin(), x.end()
#define rep(i, a, b) for(ll i=(a); i<(b); i++)
#define pr(x) cout << #x << " " << x << endl;
ll const inf = LLONG_MAX, mxn = 104;
int n, t, arr[mxn], taken[mxn], ans[mxn];
vo<vi> seq;
// gör en dp
bool solve(vi &v){
rep(i, 0, n){
cout << v[i] << " ";
} cout << endl;
rep(i, 0, n){
cout << taken[i] << " ";
} cout << endl;
rep(i, 0, n){
int pos = (i+arr[v[i]])%n;
pr(i)
pr(v[i])
pr(arr[v[i]])
pr(pos)
if(taken[pos]) return 0;
taken[pos] = 1;
}
return 1;
}
void generate(int pos, vi &used){
if(pos==n){
return;
}
rep(i, 0, n){
if(!used[i]){
used[i] = 1;
if(pos==0){
seq.pb({});
}
seq.back().pb(i);
generate(pos+1, used);
used[i] = 0;
}
}
}
void succ(){
cout << "YES\n";
rep(i, 0, n){
cout << ans[i] << " ";
}
cout << "\n";
}
void fail(){
cout << "NO\n";
}
void solve(){
cin>>n; seq.clear();
int subtask3 = 1, subtask2 = 1, ct3=0, ct1=0, sum=0;
uset<int> perm;
rep(i, 0, n){
cin>>arr[i]; sum+=arr[i];
if(arr[i] > 2) subtask3 = 0;
if(perm.count(arr[i])) {
subtask2 = 0;
}
perm.insert(arr[i]);
if(arr[i]==3) ct3++;
if(arr[i]==1) ct1++;
}
if(sum%n!=0){
cout << "NO\n"; return;
}
if(n<=8){
vi cmpr(n); rep(i, 0, n) cmpr[i] = i;
do{
vi taken(n, 0);
rep(i, 0, n) {
taken[(i+arr[cmpr[i]])%n]++;
}
if(find(all(taken), 0) != taken.end()) continue;
cout << "YES\n";
rep(i, 0, n) cout << arr[cmpr[i]] << " \n"[i==n-1];
return;
} while(next_permutation(all(cmpr)));
cout << "NO\n";
}
else if(subtask3){
int p = arr[0], allsame = 1;
rep(i, 0, n){
if(arr[i] != p) allsame = 0;
}
if(allsame){
rep(i, 0, n) ans[i] = arr[i];
succ();
}
else fail();
}
else if(subtask2){
if(n%2==0){
fail();
}
else{
int prev = n;
rep(i, 0, n){
ans[i] = prev;
prev -= 2;
if(prev <= 0){
prev = n-1;
}
}
succ();
}
}
else{
if(ct3==1){
if(n-1 - ct1 % 2){
rep(i, 0, n){
if(i < n-1-ct1){
ans[i] = 2;
}
else if(i==n-1-ct1) ans[i] = 3;
else{
ans[i] = 1;
}
}
succ();
}
else{
fail();
}
}
else{
if(ct3==n){
rep(i, 0, n) ans[i] = arr[i];
succ();
}
else fail();
}
}
}
signed main(){
cin.tie(0)->sync_with_stdio(0);
cin>>t;
rep(_, 0, t){
solve();
}
}
/*
3 1 3 1 2 2 2
*/Test details
Test 1
Group: 1, 7, 8
Verdict: ACCEPTED
| input |
|---|
| 637 1 1 2 1 1 ... |
| correct output |
|---|
| YES 1 YES 1 1 YES ... |
| user output |
|---|
| YES 1 YES 1 1 YES ... Truncated |
Test 2
Group: 1, 7, 8
Verdict: ACCEPTED
| input |
|---|
| 246 7 1 1 1 1 1 1 1 7 1 1 2 1 1 7 1 ... |
| correct output |
|---|
| YES 1 1 1 1 1 1 1 YES 1 1 1 1 2 7 1 YES ... |
| user output |
|---|
| YES 1 1 1 1 1 1 1 YES 1 1 2 7 1 1 1 YES ... Truncated |
Test 3
Group: 1, 7, 8
Verdict: ACCEPTED
| input |
|---|
| 810 8 1 1 1 1 1 1 1 1 8 1 1 1 8 1 1 2 1 ... |
| correct output |
|---|
| YES 1 1 1 1 1 1 1 1 YES 1 1 2 8 1 1 1 1 YES ... |
| user output |
|---|
| YES 1 1 1 1 1 1 1 1 YES 1 1 1 1 1 2 8 1 YES ... Truncated |
Test 4
Group: 1, 7, 8
Verdict: ACCEPTED
| input |
|---|
| 1000 8 8 8 5 2 8 7 6 5 8 6 5 2 2 8 2 1 6 ... |
| correct output |
|---|
| NO YES 8 2 2 6 2 5 1 6 NO NO ... |
| user output |
|---|
| NO YES 6 2 5 1 6 8 2 2 NO NO ... Truncated |
Test 5
Group: 1, 7, 8
Verdict: ACCEPTED
| input |
|---|
| 1000 8 2 1 7 7 2 3 8 2 8 4 1 5 4 7 3 5 3 ... |
| correct output |
|---|
| YES 7 2 2 7 1 3 8 2 YES 4 4 7 3 3 5 5 1 YES ... |
| user output |
|---|
| YES 2 7 1 2 3 7 8 2 YES 4 1 5 3 5 3 7 4 YES ... Truncated |
Test 6
Group: 7, 8
Verdict: WRONG ANSWER
| input |
|---|
| 1000 16 15 16 6 4 14 2 1 6 2 16 10 2 9... |
| correct output |
|---|
| NO NO NO NO NO ... |
| user output |
|---|
| NO NO NO NO NO ... Truncated |
Test 7
Group: 7, 8
Verdict: WRONG ANSWER
| input |
|---|
| 1000 16 2 4 13 6 8 16 12 8 16 5 9 5 9 ... |
| correct output |
|---|
| YES 13 5 2 8 12 2 8 5 16 16 9 6 9 ... |
| user output |
|---|
| NO YES 2 2 2 2 2 2 2 2 2 2 2 2 2 2 3 ... Truncated |
Test 8
Group: 2, 8
Verdict: ACCEPTED
| input |
|---|
| 1000 1 1 2 1 2 ... |
| correct output |
|---|
| YES 1 NO YES 3 1 2 ... |
| user output |
|---|
| YES 1 NO YES 3 1 2 ... Truncated |
Test 9
Group: 3, 4, 5, 6, 8
Verdict: ACCEPTED
| input |
|---|
| 988 1 1 2 1 1 ... |
| correct output |
|---|
| YES 1 YES 1 1 YES ... |
| user output |
|---|
| YES 1 YES 1 1 YES ... Truncated |
Test 10
Group: 3, 4, 5, 6, 8
Verdict: ACCEPTED
| input |
|---|
| 199 1 1 2 1 1 ... |
| correct output |
|---|
| YES 1 YES 1 1 YES ... |
| user output |
|---|
| YES 1 YES 1 1 YES ... Truncated |
Test 11
Group: 3, 4, 5, 6, 8
Verdict: ACCEPTED
| input |
|---|
| 1000 100 1 1 1 2 1 1 2 2 1 1 1 1 1 2 1 ... |
| correct output |
|---|
| NO NO NO NO NO ... |
| user output |
|---|
| NO NO NO NO NO ... Truncated |
Test 12
Group: 3, 4, 5, 6, 8
Verdict: ACCEPTED
| input |
|---|
| 1000 100 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ... |
| correct output |
|---|
| YES 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ... |
| user output |
|---|
| YES 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ... Truncated |
Test 13
Group: 4, 5, 6, 7, 8
Verdict: WRONG ANSWER
| input |
|---|
| 963 1 1 2 1 1 ... |
| correct output |
|---|
| YES 1 YES 1 1 YES ... |
| user output |
|---|
| YES 1 YES 1 1 YES ... Truncated |
Test 14
Group: 4, 5, 6, 8
Verdict: WRONG ANSWER
| input |
|---|
| 979 1 1 2 1 1 ... |
| correct output |
|---|
| YES 1 YES 1 1 YES ... |
| user output |
|---|
| YES 1 YES 1 1 YES ... Truncated |
Test 15
Group: 4, 5, 6, 8
Verdict: WRONG ANSWER
| input |
|---|
| 1000 100 3 3 1 2 1 1 2 3 1 3 2 1 1 3 1 ... |
| correct output |
|---|
| NO NO NO NO NO ... |
| user output |
|---|
| NO NO NO NO NO ... Truncated |
Test 16
Group: 4, 5, 6, 8
Verdict: WRONG ANSWER
| input |
|---|
| 1000 100 1 2 2 2 2 1 1 1 2 3 1 1 3 2 1 ... |
| correct output |
|---|
| YES 2 2 2 3 1 2 3 1 2 3 1 3 1 3 1 ... |
| user output |
|---|
| NO NO NO NO NO ... Truncated |
Test 17
Group: 5, 6, 7, 8
Verdict: WRONG ANSWER
| input |
|---|
| 980 1 1 2 1 1 ... |
| correct output |
|---|
| YES 1 YES 1 1 YES ... |
| user output |
|---|
| YES 1 YES 1 1 YES ... Truncated |
Test 18
Group: 5, 6, 8
Verdict: WRONG ANSWER
| input |
|---|
| 947 1 1 2 1 1 ... |
| correct output |
|---|
| YES 1 YES 1 1 YES ... |
| user output |
|---|
| YES 1 YES 1 1 YES ... Truncated |
Test 19
Group: 5, 6, 8
Verdict: ACCEPTED
| input |
|---|
| 1000 100 1 2 4 2 1 3 1 2 2 3 1 1 3 1 4 ... |
| correct output |
|---|
| NO NO NO NO NO ... |
| user output |
|---|
| NO NO NO NO NO ... Truncated |
Test 20
Group: 5, 6, 8
Verdict: WRONG ANSWER
| input |
|---|
| 1000 100 3 4 4 4 4 4 4 3 3 3 4 4 2 3 3 ... |
| correct output |
|---|
| YES 4 2 4 4 1 3 4 2 4 2 3 4 2 4 4 ... |
| user output |
|---|
| NO NO NO NO NO ... Truncated |
Test 21
Group: 1, 6, 7, 8
Verdict: ACCEPTED
| input |
|---|
| 715 1 1 2 1 1 ... |
| correct output |
|---|
| YES 1 YES 1 1 YES ... |
| user output |
|---|
| YES 1 YES 1 1 YES ... Truncated |
Test 22
Group: 6, 7, 8
Verdict: WRONG ANSWER
| input |
|---|
| 843 1 1 2 1 1 ... |
| correct output |
|---|
| YES 1 YES 1 1 YES ... |
| user output |
|---|
| YES 1 YES 1 1 YES ... Truncated |
Test 23
Group: 6, 8
Verdict: WRONG ANSWER
| input |
|---|
| 1000 100 3 4 5 1 4 4 2 3 2 3 4 1 1 1 2 ... |
| correct output |
|---|
| NO NO NO NO NO ... |
| user output |
|---|
| NO NO NO NO NO ... Truncated |
Test 24
Group: 6, 8
Verdict: WRONG ANSWER
| input |
|---|
| 1000 100 5 3 4 3 5 3 3 5 5 4 5 5 5 5 2 ... |
| correct output |
|---|
| YES 4 4 5 5 2 4 4 5 3 5 5 2 5 5 2 ... |
| user output |
|---|
| NO NO NO NO NO ... Truncated |
Test 25
Group: 8
Verdict: WRONG ANSWER
| input |
|---|
| 1000 100 88 70 59 44 28 10 19 19 42 16 ... |
| correct output |
|---|
| NO NO NO NO NO ... |
| user output |
|---|
| NO NO NO NO NO ... Truncated |
Test 26
Group: 8
Verdict: WRONG ANSWER
| input |
|---|
| 1000 100 31 72 52 30 77 56 79 10 88 11 ... |
| correct output |
|---|
| YES 31 62 14 10 66 63 1 82 37 92 3... |
| user output |
|---|
| NO NO NO NO NO ... Truncated |
Test 27
Group: 3, 4, 5, 6, 8
Verdict: ACCEPTED
| input |
|---|
| 1000 1 1 2 1 1 ... |
| correct output |
|---|
| YES 1 YES 1 1 YES ... |
| user output |
|---|
| YES 1 YES 1 1 YES ... Truncated |
Test 28
Group: 8
Verdict: WRONG ANSWER
| input |
|---|
| 1000 1 1 2 2 2 ... |
| correct output |
|---|
| YES 1 YES 2 2 YES ... |
| user output |
|---|
| YES 1 YES 2 2 YES ... Truncated |
Test 29
Group: 8
Verdict: WRONG ANSWER
| input |
|---|
| 1000 100 87 81 29 35 8 98 77 50 46 34 5... |
| correct output |
|---|
| YES 34 74 25 91 80 18 95 26 88 12 ... |
| user output |
|---|
| YES 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 ... Truncated |
Test 30
Group: 8
Verdict: WRONG ANSWER
| input |
|---|
| 1000 100 65 92 39 22 67 41 17 65 97 71 ... |
| correct output |
|---|
| YES 9 38 24 59 69 24 63 3 22 35 24... |
| user output |
|---|
| NO NO YES 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 ... Truncated |
