CSES - NOI 2024 - Results
Submission details
Task:Chair Game
Sender:Ellinor Ã…hlander
Submission time:2024-03-06 19:55:59 +0200
Language:C++ (C++11)
Status:READY
Result:24
Feedback
groupverdictscore
#1ACCEPTED8
#2ACCEPTED5
#3ACCEPTED4
#4ACCEPTED7
#50
#60
#70
#80
Test results
testverdicttimegroup
#1ACCEPTED0.00 s1, 7, 8details
#2ACCEPTED0.00 s1, 7, 8details
#3ACCEPTED0.01 s1, 7, 8details
#4ACCEPTED0.01 s1, 7, 8details
#5ACCEPTED0.01 s1, 7, 8details
#6--7, 8details
#7--7, 8details
#8ACCEPTED0.01 s2, 8details
#9ACCEPTED0.01 s3, 4, 5, 6, 8details
#10ACCEPTED0.01 s3, 4, 5, 6, 8details
#11ACCEPTED0.01 s3, 4, 5, 6, 8details
#12ACCEPTED0.02 s3, 4, 5, 6, 8details
#13ACCEPTED0.18 s4, 5, 6, 7, 8details
#14ACCEPTED0.19 s4, 5, 6, 8details
#15ACCEPTED0.01 s4, 5, 6, 8details
#16ACCEPTED0.01 s4, 5, 6, 8details
#17ACCEPTED0.01 s5, 6, 7, 8details
#18--5, 6, 8details
#19ACCEPTED0.01 s5, 6, 8details
#20ACCEPTED0.01 s5, 6, 8details
#21ACCEPTED0.00 s1, 6, 7, 8details
#22ACCEPTED0.37 s6, 7, 8details
#230.01 s6, 8details
#240.00 s6, 8details
#250.01 s8details
#260.00 s8details
#27ACCEPTED0.01 s3, 4, 5, 6, 8details
#28ACCEPTED0.01 s8details
#290.00 s8details
#300.00 s8details

Code

#include <bits/stdc++.h>
using namespace std;

typedef long long ll;
typedef pair<ll, ll> pll;
typedef pair<int, int> pii;

#define rep(i,a,b) for (int i = a; i < int(b); i++)
#define rrep(i,a,b) for (int i = a - 1; i >= int(b); i--)

#define all(x) x.begin(), x.end()
#define pb push_back

inline void fast() { ios::sync_with_stdio(false); cin.tie(NULL); cout.tie(NULL); }

// rd

// int ll

int t, n;

vector<int> ns;

int32_t main()
{
    fast();

    cin >> t;

    rep(qq,0,t)
    {
        cin >> n;

        ns.assign(n, -1);

        vector<bool> taken(n, false);
        bool s2 = true;

        bool s3 = true, s4 = true;

        ll sum = 0;

        rep(i,0,n)
        {
            cin >> ns[i];

            sum += ns[i];

            if (taken[ns[i]]) s2 = false;
            else taken[ns[i]] = true;

            if (ns[i] > 2) s3 = false;

            if (ns[i] > 3) s4 = false;
        }

        if (sum % n != 0)
        {
            cout << "NO\n";
            continue;
        }

        if (n <= 16)
        {
            sort(all(ns));

            bool ff = false;

            do
            {
                vector<bool> visited(n, false);

                rep(i,0,n)
                {
                    int go = (i + ns[i]) % n;
                    visited[go] = true;
                }

                bool f = true;
                rep(i,0,n) if(!visited[i]) f = false;

                if (f)
                {
                    ff = true;

                    cout << "YES\n";
                    rep(i,0,n) cout << ns[i] << " ";
                    cout << "\n";

                    break;
                }
            } 
            while (next_permutation(all(ns)));

            if (!ff)
            {
                assert(1 == 0);
            }

            continue;
        }

        if (s2)
        {
            cout << "YES\n";
            rep(i,1,n+1) cout << i << " ";
            cout << "\n";
            continue;
        }

        if (s3)
        {
            bool same = true;
            rep(i,1,n) if (ns[i] != ns[0]) same = false;

            if (same)
            {
                cout << "YES\n";
                rep(i,0,n) cout << ns[i] << " ";
                cout << "\n";
                continue;
            }
            else assert(1 == 0);
        }

        if (s4)
        {
            bool same = true;
            rep(i,1,n) if (ns[i] != ns[0]) same = false;

            if (same)
            {
                cout << "YES\n";
                rep(i,0,n) cout << ns[i] << " ";
                cout << "\n";
                continue;
            }

            int twos = 0;
            rep(i,0,n) if (ns[i] == 2) twos++;

            cout << "YES\n";
            rep(i,0,(n - twos) / 2)
            {
                cout << "3 1 ";
            }
            rep(i,0,twos) cout << "2 ";

            cout << "\n";

            continue;
        }

        else // s5
        {
            bool same = true;
            rep(i,1,n) if (ns[i] != ns[0]) same = false;

            if (same)
            {
                cout << "YES\n";
                rep(i,0,n) cout << ns[i] << " ";
                cout << "\n";
                continue;
            }

            cout << "YES\n";

            if (sum / n == 2)
            {
                int twos = 0;
                rep(i,0,n) if (ns[i] == 2) twos++;

                int threes = 0;
                rep(i,0,n) if (ns[i] == 3) threes++;

                rep(i,0,threes)
                {
                    cout << "3 1 ";
                }
                rep(i,0,twos) cout << "2 ";

                int tmp = n - (threes * 2) - twos;

                rep(i,0,tmp / 3)
                {
                    cout << "4 1 1 ";
                }
            }

            else if (sum / n == 3)
            {
                int twos = 0;
                rep(i,0,n) if (ns[i] == 2) twos++;

                int threes = 0;
                rep(i,0,n) if (ns[i] == 3) threes++;

                rep(i,0,threes)
                {
                    cout << "3 ";
                }

                int tmp = n - threes - (twos * 2);

                rep(i,0,tmp / 3)
                {
                    cout << "4 4 1 ";
                }

                rep(i,0,twos) cout << "4 2 ";

                //
            }

            else assert(1 == 0);

            cout << "\n";
            continue;
        }

        //

        cout << "YES\n";

        //
    }
}

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: 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
1 2 2 5 6 2 8 6 
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
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: ACCEPTED

input
1000
1
1
2
1 2
...

correct output
YES

NO
YES
3 1 2 
...

user output
YES

NO
YES
1 2 3 
...
Truncated

Test 9

Group: 3, 4, 5, 6, 8

Verdict: ACCEPTED

input
988
1
1
2
1 1
...

correct output
YES

YES
1 1 
YES
...

user output
YES

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

YES
1 1 
YES
...

user output
YES

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: ACCEPTED

input
963
1
1
2
1 1
...

correct output
YES

YES
1 1 
YES
...

user output
YES

YES
1 1 
YES
...
Truncated

Test 14

Group: 4, 5, 6, 8

Verdict: ACCEPTED

input
979
1
1
2
1 1
...

correct output
YES

YES
1 1 
YES
...

user output
YES

YES
1 1 
YES
...
Truncated

Test 15

Group: 4, 5, 6, 8

Verdict: ACCEPTED

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: ACCEPTED

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

Test 17

Group: 5, 6, 7, 8

Verdict: ACCEPTED

input
980
1
1
2
1 1
...

correct output
YES

YES
1 1 
YES
...

user output
YES

YES
1 1 
YES
...
Truncated

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: 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: ACCEPTED

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
YES
3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 ...
Truncated

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: ACCEPTED

input
843
1
1
2
1 1
...

correct output
YES

YES
1 1 
YES
...

user output
YES

YES
1 1 
YES
...
Truncated

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
NO
NO
NO
NO
NO
...
Truncated

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)

Error:
code: input/code.cpp:216: int32_t main(): Assertion `1 == 0' failed.

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)

Error:
code: input/code.cpp:216: int32_t main(): Assertion `1 == 0' failed.

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)

Error:
code: input/code.cpp:216: int32_t main(): Assertion `1 == 0' failed.

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)

Error:
code: input/code.cpp:216: int32_t main(): Assertion `1 == 0' failed.

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)

Error:
code: input/code.cpp:216: int32_t main(): Assertion `1 == 0' failed.