CSES - Datatähti 2025 alku - Results
Submission details
Task:Kortit I
Sender:fatihmerickoc
Submission time:2024-10-28 22:20:49 +0200
Language:C++ (C++11)
Status:READY
Result:0
Feedback
groupverdictscore
#10
#20
#30
Test results
testverdicttimegroup
#1--1, 2, 3details
#2--2, 3details
#3--3details
#4--3details
#5--3details
#6--3details
#7--3details
#8--3details
#9--3details
#10--3details
#11--3details
#12--3details
#13--3details
#14--3details
#15--3details
#16--3details
#17--3details
#18--3details
#19--3details
#20--3details

Code

//
//  main.cpp
//  cpp
//
//  Created by Fatih Meric Koc on 28.10.2024.
//


#include <iostream>
#include <vector>
#include <algorithm>
#include <tuple>
using namespace std;

pair<bool, pair<vector<int>, vector<int>>> solve(int n, int a, int b) {
    // Impossible cases
    if (a + b > n)  // Total points can't exceed number of rounds
        return {false, {{}, {}}};
    if (max(a, b) > n/2 + n%2)  // Can't win more than ceil(n/2) rounds
        return {false, {{}, {}}};
    
    // For n = 2, special case handling
    if (n == 2) {
        if (a == 1 && b == 1) {
            return {true, {{1, 2}, {2, 1}}};
        }
        if (a == 0 && b == 1) {
            return {false, {{}, {}}};
        }
        if (a == 0 && b == 0) {
            return {true, {{1, 1}, {1, 1}}};
        }
    }
    
    // For all draws case
    if (a == 0 && b == 0) {
        vector<int> sequence(n);
        for (int i = 0; i < n; i++) sequence[i] = i + 1;
        return {true, {sequence, sequence}};
    }
    
    vector<int> player1(n), player2(n);
    vector<bool> used1(n + 1, false), used2(n + 1, false);
    
    // Handle player1's wins
    int pos = 0;
    for (int i = 0; i < a; i++) {
        player1[pos] = n - i;
        player2[pos] = i + 1;
        used1[n - i] = true;
        used2[i + 1] = true;
        pos++;
    }
    
    // Handle player2's wins
    for (int i = 0; i < b; i++) {
        // Find next available high card for player2
        int high = n;
        while (used2[high]) high--;
        // Find next available low card for player1
        int low = 1;
        while (used1[low]) low++;
        
        player1[pos] = low;
        player2[pos] = high;
        used1[low] = true;
        used2[high] = true;
        pos++;
    }
    
    // Fill remaining positions with draws
    int next = 1;
    while (pos < n) {
        while (next <= n && (used1[next] || used2[next])) next++;
        if (next <= n) {
            player1[pos] = next;
            player2[pos] = next;
            used1[next] = used2[next] = true;
            pos++;
        }
    }
    
    // Verify solution
    int check_a = 0, check_b = 0;
    for (int i = 0; i < n; i++) {
        if (player1[i] > player2[i]) check_a++;
        else if (player2[i] > player1[i]) check_b++;
    }
    
    if (check_a == a && check_b == b) {
        return {true, {player1, player2}};
    }
    
    return {false, {{}, {}}};
}

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);
    
    int t;
    cin >> t;
    vector<tuple<int, int, int>> test_cases(t);
    for (int i = 0; i < t; ++i) {
        int n, a, b;
        cin >> n >> a >> b;
        test_cases[i] = make_tuple(n, a, b);
    }

    for (const auto& test_case : test_cases) {
        int n, a, b;
        tie(n, a, b) = test_case;
        auto result = solve(n, a, b);
        
        if (!result.first) {
            cout << "NO\n";
        } else {
            cout << "YES\n";
            for (int x : result.second.first) {
                cout << x << " ";
            }
            cout << "\n";
            for (int x : result.second.second) {
                cout << x << " ";
            }
            cout << "\n";
        }
    }

    return 0;
}

Test details

Test 1

Group: 1, 2, 3

Verdict:

input
54
4 4 0
3 1 3
3 2 2
4 0 4
...

correct output
NO
NO
NO
NO
NO
...

user output
(empty)

Test 2

Group: 2, 3

Verdict:

input
284
6 1 0
5 0 2
7 1 5
7 7 5
...

correct output
NO
NO
YES
1 2 3 4 5 6 7 
2 3 4 5 6 1 7 
...

user output
(empty)

Test 3

Group: 3

Verdict:

input
955
14 2 10
12 2 5
10 4 9
14 1 13
...

correct output
YES
1 2 3 4 5 6 7 8 9 10 11 12 13 ...

user output
(empty)

Test 4

Group: 3

Verdict:

input
869
17 12 9
16 8 4
15 9 9
17 11 15
...

correct output
NO
YES
1 2 3 4 5 6 7 8 9 10 11 12 13 ...

user output
(empty)

Test 5

Group: 3

Verdict:

input
761
18 3 15
19 1 15
18 8 1
19 19 17
...

correct output
YES
1 2 3 4 5 6 7 8 9 10 11 12 13 ...

user output
(empty)

Test 6

Group: 3

Verdict:

input
925
21 14 21
20 18 18
20 7 6
21 14 9
...

correct output
NO
NO
YES
1 2 3 4 5 6 7 8 9 10 11 12 13 ...

user output
(empty)

Test 7

Group: 3

Verdict:

input
529
22 3 3
22 17 5
22 6 15
22 22 20
...

correct output
YES
1 2 3 4 5 6 7 8 9 10 11 12 13 ...

user output
(empty)

Test 8

Group: 3

Verdict:

input
576
23 18 9
23 16 8
23 16 13
23 16 22
...

correct output
NO
NO
NO
NO
NO
...

user output
(empty)

Test 9

Group: 3

Verdict:

input
625
24 2 22
24 15 21
24 6 3
24 21 1
...

correct output
YES
1 2 3 4 5 6 7 8 9 10 11 12 13 ...

user output
(empty)

Test 10

Group: 3

Verdict:

input
676
25 16 25
25 15 2
25 15 7
25 15 16
...

correct output
NO
YES
1 2 3 4 5 6 7 8 9 10 11 12 13 ...

user output
(empty)

Test 11

Group: 3

Verdict:

input
729
26 2 18
26 14 18
26 5 18
26 19 13
...

correct output
YES
1 2 3 4 5 6 7 8 9 10 11 12 13 ...

user output
(empty)

Test 12

Group: 3

Verdict:

input
784
27 26 7
27 14 0
27 14 5
27 14 14
...

correct output
NO
NO
YES
1 2 3 4 5 6 7 8 9 10 11 12 13 ...

user output
(empty)

Test 13

Group: 3

Verdict:

input
841
28 26 16
28 13 19
28 5 8
28 26 4
...

correct output
NO
NO
YES
1 2 3 4 5 6 7 8 9 10 11 12 13 ...

user output
(empty)

Test 14

Group: 3

Verdict:

input
900
29 24 15
29 13 2
29 13 7
29 13 16
...

correct output
NO
YES
1 2 3 4 5 6 7 8 9 10 11 12 13 ...

user output
(empty)

Test 15

Group: 3

Verdict:

input
961
30 24 26
30 12 24
30 4 29
30 24 14
...

correct output
NO
NO
NO
NO
YES
...

user output
(empty)

Test 16

Group: 3

Verdict:

input
1000
15 12 6
33 18 30
44 4 26
6 6 5
...

correct output
NO
NO
YES
1 2 3 4 5 6 7 8 9 10 11 12 13 ...

user output
(empty)

Test 17

Group: 3

Verdict:

input
1000
45 32 30
4 0 3
46 23 10
71 19 46
...

correct output
NO
NO
YES
1 2 3 4 5 6 7 8 9 10 11 12 13 ...

user output
(empty)

Test 18

Group: 3

Verdict:

input
1000
51 29 37
75 11 72
5 2 4
31 8 26
...

correct output
NO
NO
NO
NO
YES
...

user output
(empty)

Test 19

Group: 3

Verdict:

input
1000
50 20 37
99 45 58
86 79 73
85 70 54
...

correct output
NO
NO
NO
NO
NO
...

user output
(empty)

Test 20

Group: 3

Verdict:

input
1000
26 23 5
73 53 59
64 47 41
80 75 55
...

correct output
NO
NO
NO
NO
NO
...

user output
(empty)