CSES - NOI 2024 - Results
Submission details
Task:Chair Game
Sender:August Broe
Submission time:2024-03-06 19:23:15 +0200
Language:C++ (C++20)
Status:READY
Result:9
Feedback
groupverdictscore
#10
#2ACCEPTED5
#3ACCEPTED4
#40
#50
#60
#70
#80
Test results
testverdicttimegroup
#10.01 s1, 7, 8details
#20.01 s1, 7, 8details
#30.01 s1, 7, 8details
#4ACCEPTED0.32 s1, 7, 8details
#5ACCEPTED0.02 s1, 7, 8details
#6--7, 8details
#7ACCEPTED0.20 s7, 8details
#8ACCEPTED0.03 s2, 8details
#9ACCEPTED0.01 s3, 4, 5, 6, 8details
#10ACCEPTED0.01 s3, 4, 5, 6, 8details
#11ACCEPTED0.02 s3, 4, 5, 6, 8details
#12ACCEPTED0.03 s3, 4, 5, 6, 8details
#130.01 s4, 5, 6, 7, 8details
#140.04 s4, 5, 6, 8details
#150.03 s4, 5, 6, 8details
#160.03 s4, 5, 6, 8details
#170.06 s5, 6, 7, 8details
#18--5, 6, 8details
#19--5, 6, 8details
#20--5, 6, 8details
#210.02 s1, 6, 7, 8details
#220.07 s6, 7, 8details
#23--6, 8details
#24--6, 8details
#25--8details
#26--8details
#27ACCEPTED0.02 s3, 4, 5, 6, 8details
#28ACCEPTED0.02 s8details
#29--8details
#30--8details

Compiler report

input/code.cpp: In function 'std::vector<int> findSolution(std::vector<int>, std::vector<int>, std::map<int, int>, int, int)':
input/code.cpp:10:28: warning: comparison of integer expressions of different signedness: 'std::map<int, int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   10 |         if (numbers.size() == n)
      |             ~~~~~~~~~~~~~~~^~~~

Code

#include <iostream>
#include <vector>
#include <algorithm>
#include <map>

std::vector<int> findSolution(std::vector<int> chairs, std::vector<int> occupiedNext, std::map<int, int> numbers, int n, int currentSpot)
{
    if (currentSpot == 0)
    {
        if (numbers.size() == n)
        {
            if (n % 2 == 0)
            {
                return std::vector<int>();
            }
            std::vector<int> result;
            for (auto i : numbers)
            {
                result.push_back(i.first);
            }
            return result;
        }

        if (numbers.size() == 1)
        {
            std::vector<int> result;
            for (auto i : numbers)
            {
                for (int j = 0; j < i.second; j++)
                {
                    result.push_back(i.first);
                }
            }
            return result;
        }

        if (numbers.size() == 2 && numbers.find(1) != numbers.end() && numbers.find(2) != numbers.end())
        {
            return std::vector<int>();
        }

        if (numbers.size() == 3 && numbers.find(1) != numbers.end() && numbers.find(2) != numbers.end() && numbers.find(3) != numbers.end())
        {
            int ones = numbers[1];
            int twos = numbers[2];
            int threes = numbers[3];
            int min = std::min(ones, std::min(twos, threes));
            numbers[1] -= min;
            numbers[2] -= min;
            numbers[3] -= min;
            std::vector<int> result;
            for (int i = 0; i < min; i++)
            {
                chairs[3*i] = 1;
                chairs[3*i+1] = 2;
                chairs[3*i+2] = 3;
            }
            int currentSpot = 3*min;

            for (int i = 0; i < n; i++)
            {
                if (chairs[i] != -1)
                {
                    occupiedNext[(i + chairs[i]) % n] = 1;
                }
            }

            if (numbers[1] == 0)
            {
                numbers.erase(1);
            }
            if (numbers[2] == 0)
            {
                numbers.erase(2);
            }
            if (numbers[3] == 0)
            {
                numbers.erase(3);
            }
            
            return findSolution(chairs, occupiedNext, numbers, n, currentSpot);
        }


        
    }
    if (numbers.size() == 0)
    {
        return chairs;
    }

    for (auto i : numbers)
    {
        if (occupiedNext[(i.first + currentSpot) % n] == 0)
        {
            std::vector<int> nextChairs = chairs;
            nextChairs[currentSpot] = i.first;
            std::vector<int> nextOccupiedNext = occupiedNext;
            nextOccupiedNext[(i.first + currentSpot) % n] = 1;
            std::map<int, int> nextNumbers = numbers;
            nextNumbers[i.first]--;
            if (nextNumbers[i.first] == 0)
            {
                nextNumbers.erase(i.first);
            }

            std::vector<int> result = findSolution(nextChairs, nextOccupiedNext, nextNumbers, n, currentSpot + 1);
        
            if (result.size() != 0)
            {
                return result;
            }
        }
    }
    return std::vector<int>();
}

int main()
{
    int t;
    std::cin >> t;
    while (t--)
    {
        int n;
        std::cin >> n;
        std::map<int, int> m;
        for (int i = 0; i < n; i++)
        {
            int x;
            std::cin >> x;
            m[x]++;
        }

        std::vector<int> chairs(n, -1);
        std::vector<int> occupiedNext(n, 0);

        std::vector<int> result = findSolution(chairs, occupiedNext, m, n, 0);
        if (result.size() == 0)
        {
            std::cout << "NO" << std::endl;
            continue;
        }
        std::cout << "YES" << std::endl;
        for (auto i : result)
        {
            std::cout << i << " ";
        }
        std::cout << std::endl;
    }
}

Test details

Test 1

Group: 1, 7, 8

Verdict:

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:

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:

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

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
YES
2 2 2 4 5 5 8 8 9 12 6 11 16 1...
Truncated

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:

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:

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:

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:

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

Test 17

Group: 5, 6, 7, 8

Verdict:

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:

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:

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