CSES - NOI 2024 - Results
Submission details
Task:Chair Game
Sender:Þórhallur Tryggvason
Submission time:2024-03-06 19:43:00 +0200
Language:C++ (C++20)
Status:READY
Result:0
Feedback
groupverdictscore
#10
#20
#30
#40
#50
#60
#70
#80
Test results
testverdicttimegroup
#1ACCEPTED0.03 s1, 7, 8details
#2ACCEPTED0.01 s1, 7, 8details
#3ACCEPTED0.03 s1, 7, 8details
#4--1, 7, 8details
#5ACCEPTED0.02 s1, 7, 8details
#6--7, 8details
#7--7, 8details
#8--2, 8details
#9--3, 4, 5, 6, 8details
#10ACCEPTED0.01 s3, 4, 5, 6, 8details
#11--3, 4, 5, 6, 8details
#12ACCEPTED0.04 s3, 4, 5, 6, 8details
#13--4, 5, 6, 7, 8details
#14--4, 5, 6, 8details
#15--4, 5, 6, 8details
#16--4, 5, 6, 8details
#17--5, 6, 7, 8details
#18--5, 6, 8details
#19--5, 6, 8details
#20--5, 6, 8details
#21ACCEPTED0.13 s1, 6, 7, 8details
#22--6, 7, 8details
#23--6, 8details
#24--6, 8details
#25--8details
#26--8details
#27ACCEPTED0.04 s3, 4, 5, 6, 8details
#28ACCEPTED0.03 s8details
#29--8details
#30--8details

Code

#include <bits/stdc++.h>

using namespace std;

struct memory {
    vector<int> values;
    unordered_map<int, memory*> mem;

    // for root
    memory() {
        this->mem = unordered_map<int, memory*>();
        this->values = vector<int>();
    }
    memory(vector<int>& chairs, vector<int>& values, uint64_t idx) {
        if (idx == values.size()) {
            this->values = values;
            this->mem = unordered_map<int, memory*>();
        } else {
            this->values = vector<int>();
            this->mem = unordered_map<int, memory*>();
            memory* n = new memory(chairs, values, idx + 1);
            this->mem.insert(make_pair(chairs[idx], n));
        }
    }

    pair<int*, uint64_t> find(vector<int>& finding, uint64_t idx) {
        if (idx == finding.size()) {
            return make_pair(values.data(), values.size());
        }
        auto f = mem.find(finding[idx]);
        if (f == mem.end()) {
            return make_pair(nullptr, 0);
        } else {
            return f->second->find(finding, idx + 1);
        }
    }
    void insert(vector<int>& chairs, vector<int>& values, uint64_t idx) {
        auto f = mem.find(values[idx]);
        if (f == mem.end()) {
            memory* n = new memory(chairs, values, idx + 1);
            this->mem.insert(make_pair(chairs[idx], n));
        } else {
            f->second->insert(chairs, values, idx + 1);
        }
    }
};

int main() {
    int t;
    cin >> t;

    memory root_mem = memory();

    while (t--) {
        int n;
        cin >> n;
        vector<int> chairs(n);
        vector<int> perm(n);
        for (int i = 0; i < n; i++) {
            cin >> chairs[i];
            perm[i] = i;
        }
        sort(chairs.begin(), chairs.end());
        auto f = root_mem.find(chairs, 0);
        if (f.first != nullptr) {
            if (f.second == 0) {
                cout << "NO" << endl;
            } else {
                cout << "YES" << endl;
                for (int* ptr = f.first; ptr < (f.first + f.second); ptr++) {
                    cout << chairs[*ptr] << " ";
                }
                cout << endl;
            }
            continue;
        }
        bool flag = false;
        do {
            vector<bool> marker(n, false);
            bool f = false;
            for (int i = 0; i < n; i++) {
                int pos = (i + chairs[perm[i]]) % n;
                if (marker[pos]) {
                    f = true;
                    break;
                } else {
                    marker[pos] = true;
                }
            }
            if (f) {
                continue;
            }
            flag = true;
            break;
        } while (next_permutation(perm.begin(), perm.end()));

        if (flag) {
            cout << "YES" << endl;
            root_mem.insert(chairs, perm, 0);
            for (int i = 0; i < n; i++) {
                cout << chairs[perm[i]] << " ";
            }
            cout << endl;
        } else {
            root_mem.insert(chairs, perm, 0);
            cout << "NO" << endl;
        }
    }
}

Test details

Test 1

Group: 1, 7, 8

Verdict: ACCEPTED

input
637
1
1
2
1 1
...

correct output
YES

YES
1 1 
YES
...

user output
YES

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 1 1 1 2 7 
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 1 2 8 
YES
...
Truncated

Test 4

Group: 1, 7, 8

Verdict:

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
(empty)

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
1 2 2 7 2 3 7 8 
YES
1 3 3 4 4 5 5 7 
YES
...
Truncated

Test 6

Group: 7, 8

Verdict:

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
(empty)

Test 7

Group: 7, 8

Verdict:

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
(empty)

Test 8

Group: 2, 8

Verdict:

input
1000
1
1
2
1 2
...

correct output
YES

NO
YES
3 1 2 
...

user output
(empty)

Test 9

Group: 3, 4, 5, 6, 8

Verdict:

input
988
1
1
2
1 1
...

correct output
YES

YES
1 1 
YES
...

user output
(empty)

Test 10

Group: 3, 4, 5, 6, 8

Verdict: ACCEPTED

input
199
1
1
2
1 1
...

correct output
YES

YES
1 1 
YES
...

user output
YES

YES
1 1 
YES
...
Truncated

Test 11

Group: 3, 4, 5, 6, 8

Verdict:

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
(empty)

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:

input
963
1
1
2
1 1
...

correct output
YES

YES
1 1 
YES
...

user output
(empty)

Test 14

Group: 4, 5, 6, 8

Verdict:

input
979
1
1
2
1 1
...

correct output
YES

YES
1 1 
YES
...

user output
(empty)

Test 15

Group: 4, 5, 6, 8

Verdict:

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
(empty)

Test 16

Group: 4, 5, 6, 8

Verdict:

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
(empty)

Test 17

Group: 5, 6, 7, 8

Verdict:

input
980
1
1
2
1 1
...

correct output
YES

YES
1 1 
YES
...

user output
(empty)

Test 18

Group: 5, 6, 8

Verdict:

input
947
1
1
2
1 1
...

correct output
YES

YES
1 1 
YES
...

user output
(empty)

Test 19

Group: 5, 6, 8

Verdict:

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
(empty)

Test 20

Group: 5, 6, 8

Verdict:

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
(empty)

Test 21

Group: 1, 6, 7, 8

Verdict: ACCEPTED

input
715
1
1
2
1 1
...

correct output
YES

YES
1 1 
YES
...

user output
YES

YES
1 1 
YES
...
Truncated

Test 22

Group: 6, 7, 8

Verdict:

input
843
1
1
2
1 1
...

correct output
YES

YES
1 1 
YES
...

user output
(empty)

Test 23

Group: 6, 8

Verdict:

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
(empty)

Test 24

Group: 6, 8

Verdict:

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
(empty)

Test 25

Group: 8

Verdict:

input
1000
100
88 70 59 44 28 10 19 19 42 16 ...

correct output
NO
NO
NO
NO
NO
...

user output
(empty)

Test 26

Group: 8

Verdict:

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
(empty)

Test 27

Group: 3, 4, 5, 6, 8

Verdict: ACCEPTED

input
1000
1
1
2
1 1
...

correct output
YES

YES
1 1 
YES
...

user output
YES

YES
1 1 
YES
...
Truncated

Test 28

Group: 8

Verdict: ACCEPTED

input
1000
1
1
2
2 2
...

correct output
YES

YES
2 2 
YES
...

user output
YES

YES
2 2 
YES
...
Truncated

Test 29

Group: 8

Verdict:

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
(empty)

Test 30

Group: 8

Verdict:

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
(empty)