CSES - Datatähti 2024 alku - Results
Submission details
Task:Käännöt
Sender:anotn
Submission time:2023-11-07 21:45:21 +0200
Language:C++17
Status:READY
Result:100
Feedback
groupverdictscore
#1ACCEPTED13
#2ACCEPTED19
#3ACCEPTED33
#4ACCEPTED35
Test results
testverdicttimegroup
#1ACCEPTED0.00 s1details
#2ACCEPTED0.00 s1details
#3ACCEPTED0.00 s1details
#4ACCEPTED0.00 s1details
#5ACCEPTED0.00 s1details
#6ACCEPTED0.00 s2details
#7ACCEPTED0.01 s2details
#8ACCEPTED0.00 s2details
#9ACCEPTED0.00 s2details
#10ACCEPTED0.00 s2details
#11ACCEPTED0.01 s2details
#12ACCEPTED0.01 s2details
#13ACCEPTED0.01 s2details
#14ACCEPTED0.01 s2details
#15ACCEPTED0.01 s2details
#16ACCEPTED0.01 s2details
#17ACCEPTED0.00 s3details
#18ACCEPTED0.00 s3details
#19ACCEPTED0.01 s3details
#20ACCEPTED0.01 s3details
#21ACCEPTED0.01 s3details
#22ACCEPTED0.01 s3details
#23ACCEPTED0.01 s3details
#24ACCEPTED0.01 s3details
#25ACCEPTED0.01 s3details
#26ACCEPTED0.00 s4details
#27ACCEPTED0.00 s4details
#28ACCEPTED0.01 s4details
#29ACCEPTED0.01 s4details
#30ACCEPTED0.01 s4details
#31ACCEPTED0.01 s4details
#32ACCEPTED0.01 s4details
#33ACCEPTED0.01 s4details
#34ACCEPTED0.01 s4details

Compiler report

input/code.cpp: In function 'std::vector<int> bubble(std::vector<int>&)':
input/code.cpp:12:26: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   12 |         for(int i = 0; i < unsorted.size()-1; i++)
      |                        ~~^~~~~~~~~~~~~~~~~~~
input/code.cpp: In function 'std::vector<int> quadruple(std::vector<int>&)':
input/code.cpp:53:21: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   53 |         while(index != unsorted.size()-1){
      |               ~~~~~~^~~~~~~~~~~~~~~~~~~~
input/code.cpp:94:26: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   94 |         for(int i = 0; i <= unsorted.size()-4; i++){
      |                        ~~^~~~~~~~~~~~~~~~~~~~
inp...

Code

#include <iostream>
#include <vector>
#include <set>
#include <algorithm>
using namespace std;
 
vector<int> bubble(vector<int>& unsorted){
    vector<int> ans;
    bool loop = true;
    while(loop){
        loop = false;
        for(int i = 0; i < unsorted.size()-1; i++)
            if(unsorted[i] > unsorted[i+1]){
                swap(unsorted[i], unsorted[i+1]);
                ans.push_back(i+1);
                loop = true;
            }
    }
    return ans;
}

vector<int> triple(vector<int>& unsorted){
    vector<int> ans;
    int highest_not_sorted = unsorted.size();

    while(highest_not_sorted > 0){
        auto index = find(unsorted.begin(), unsorted.end(), highest_not_sorted) - unsorted.begin();
        
        if(abs(index - highest_not_sorted)%2 == 0)
            return vector<int>(1000001);
        
        for(int i = index; i < highest_not_sorted-1; i += 2){
            swap(unsorted[i], unsorted[i+2]);
            ans.push_back(i+1);
        }
        highest_not_sorted--;
    }

    return ans;
}

void swap_quad(vector<int>& input, int index){
    swap(input[index], input[index+3]);
    swap(input[index+1], input[index+2]);
}

vector<int> quadruple(vector<int>& unsorted){
    vector<int> ans;

    while(unsorted.size() > 6){
        int index = find(unsorted.begin(), unsorted.end(), unsorted.size()) - unsorted.begin();
        
        while(index != unsorted.size()-1){
            while((unsorted.size()-1 - index)%3 != 0){
                if(index == 0){
                    swap_quad(unsorted, index);
                    ans.push_back(index+1);
                    index += 3;
                }

                else if((unsorted.size()-1 - index) < 3){
                    swap_quad(unsorted, index-2);
                    ans.push_back(index-1);
                    index--;
                } else {
                    swap_quad(unsorted, index-1);
                    ans.push_back(index);
                    index++;
                }
            }
            swap_quad(unsorted, index);
            ans.push_back(index+1);
            index += 3;
        }
        unsorted.pop_back();
    }

    set<vector<int>> brute;
    vector<pair<vector<int>, vector<int>>> queue;
    queue.push_back(make_pair(unsorted, vector<int>()));

    auto sorted = unsorted;
    sort(sorted.begin(), sorted.end());

    while(!queue.empty()){
        auto vec = queue[0].first;
        auto steps = queue[0].second;
        queue.erase(queue.begin());

        if(vec == sorted){
            ans.insert(ans.end(), steps.begin(), steps.end());
            return ans;
        }
        for(int i = 0; i <= unsorted.size()-4; i++){
            auto vec_copy = vec;
            auto steps_copy = steps;
            swap_quad(vec_copy, i);
            steps_copy.push_back(i+1);
            if(brute.find(vec_copy) == brute.end()){
                brute.insert(vec_copy);
                queue.push_back(make_pair(vec_copy, steps_copy));
            }
        }
    }

    return vector<int>(1000001);
}

void swap_quint(vector<int>& input, int index){
    swap(input[index], input[index+4]);
    swap(input[index+1], input[index+3]);
}

vector<int> quintuple(vector<int>& unsorted){
    vector<int> ans;

    while(unsorted.size() > 7){
        int index = find(unsorted.begin(), unsorted.end(), unsorted.size()) - unsorted.begin();

        if((unsorted.size()-1-index)%2 != 0)
            return vector<int>(1000001);
        
        while(index != unsorted.size()-1){
            while((unsorted.size()-1 - index)%4 != 0){
                if(index == 0){
                    swap_quint(unsorted, index);
                    ans.push_back(index+1);
                    index += 4;
                }

                else if((unsorted.size()-1 - index) < 4){
                    swap_quint(unsorted, index-3);
                    ans.push_back(index-2);
                    index -= 2;
                } else {
                    swap_quint(unsorted, index-1);
                    ans.push_back(index);
                    index += 2;
                }
            }
            swap_quint(unsorted, index);
            ans.push_back(index+1);
            index += 4;
        }
        unsorted.pop_back();
    }

    set<vector<int>> brute;
    vector<pair<vector<int>, vector<int>>> queue;
    queue.push_back(make_pair(unsorted, vector<int>()));

    auto sorted = unsorted;
    sort(sorted.begin(), sorted.end());

    while(!queue.empty()){
        auto vec = queue[0].first;
        auto steps = queue[0].second;
        queue.erase(queue.begin());

        if(vec == sorted){
            ans.insert(ans.end(), steps.begin(), steps.end());
            return ans;
        }
        for(int i = 0; i <= unsorted.size()-5; i++){
            auto vec_copy = vec;
            auto steps_copy = steps;
            swap_quint(vec_copy, i);
            steps_copy.push_back(i+1);
            if(brute.find(vec_copy) == brute.end()){
                brute.insert(vec_copy);
                queue.push_back(make_pair(vec_copy, steps_copy));
            }
        }
    }

    return vector<int>(1000001);
}
 
int main(){
    int n, k;
    cin >> n >> k;
 
    vector<int> unsorted(n);
    for(int i = 0; i < n; i++){
        int a;
        cin >> a;
        unsorted[i] = a;
    }
    
    vector<int> ans;
    if(k == 2)
        ans = bubble(unsorted);
    
    else if(k == 3)
        ans = triple(unsorted);

    else if(k == 4)
        ans = quadruple(unsorted);

    else
        ans = quintuple(unsorted);
    
    if(ans.size() <= 1000000){
        cout << "YES\n" << ans.size()+2 << "\n1 1 ";
        for(auto c : ans)
            cout << c << " ";
    }
    else
        cout << "NO";
}

Test details

Test 1

Group: 1

Verdict: ACCEPTED

input
5 2
1 2 3 4 5

correct output
YES
0

user output
YES
2
1 1 

Test 2

Group: 1

Verdict: ACCEPTED

input
5 2
2 1 3 4 5

correct output
YES
1
1

user output
YES
3
1 1 1 

Test 3

Group: 1

Verdict: ACCEPTED

input
20 2
6 20 18 2 16 13 19 17 8 14 11 ...

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

user output
YES
134
1 1 2 3 4 5 6 7 8 9 10 11 12 1...
Truncated

Test 4

Group: 1

Verdict: ACCEPTED

input
100 2
100 92 62 88 12 7 43 31 19 72 ...

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

user output
YES
2440
1 1 1 2 3 4 5 6 7 8 9 10 11 12...
Truncated

Test 5

Group: 1

Verdict: ACCEPTED

input
100 2
100 99 98 97 96 95 94 93 92 91...

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

user output
YES
4952
1 1 1 2 3 4 5 6 7 8 9 10 11 12...
Truncated

Test 6

Group: 2

Verdict: ACCEPTED

input
5 3
1 2 3 4 5

correct output
YES
0

user output
YES
2
1 1 

Test 7

Group: 2

Verdict: ACCEPTED

input
5 3
3 5 4 1 2

correct output
NO

user output
NO

Test 8

Group: 2

Verdict: ACCEPTED

input
5 3
5 2 1 4 3

correct output
YES
8
1 2 1 3 1 2 3 1

user output
YES
4
1 1 1 3 

Test 9

Group: 2

Verdict: ACCEPTED

input
20 3
19 14 1 18 3 4 11 20 13 6 17 8...

correct output
YES
52
8 10 12 14 16 18 1 3 5 7 9 11 ...

user output
YES
44
1 1 8 10 12 14 16 18 1 3 5 7 9...
Truncated

Test 10

Group: 2

Verdict: ACCEPTED

input
20 3
9 6 13 18 5 10 3 2 7 20 1 4 19...

correct output
YES
50
10 12 14 16 18 13 15 17 4 6 8 ...

user output
YES
36
1 1 10 12 14 16 18 13 15 17 4 ...
Truncated

Test 11

Group: 2

Verdict: ACCEPTED

input
500 3
53 52 21 76 25 142 5 4 83 176 ...

correct output
YES
15194
334 336 338 340 342 344 346 34...

user output
YES
15176
1 1 334 336 338 340 342 344 34...
Truncated

Test 12

Group: 2

Verdict: ACCEPTED

input
500 3
51 44 147 172 1 28 27 82 233 1...

correct output
YES
15565
366 368 370 372 374 376 378 38...

user output
YES
15545
1 1 366 368 370 372 374 376 37...
Truncated

Test 13

Group: 2

Verdict: ACCEPTED

input
500 3
75 46 179 62 221 14 67 154 89 ...

correct output
YES
15920
454 456 458 460 462 464 466 46...

user output
YES
15904
1 1 454 456 458 460 462 464 46...
Truncated

Test 14

Group: 2

Verdict: ACCEPTED

input
500 3
161 54 285 12 71 142 111 94 97...

correct output
YES
15931
408 410 412 414 416 418 420 42...

user output
YES
15909
1 1 408 410 412 414 416 418 42...
Truncated

Test 15

Group: 2

Verdict: ACCEPTED

input
500 3
122 260 455 113 315 276 433 43...

correct output
NO

user output
NO

Test 16

Group: 2

Verdict: ACCEPTED

input
500 3
499 500 497 498 495 496 493 49...

correct output
YES
62264
2 4 6 8 10 12 14 16 18 20 22 2...

user output
YES
62252
1 1 2 4 6 8 10 12 14 16 18 20 ...
Truncated

Test 17

Group: 3

Verdict: ACCEPTED

input
5 4
1 2 3 4 5

correct output
YES
0

user output
YES
2
1 1 

Test 18

Group: 3

Verdict: ACCEPTED

input
5 4
5 1 2 3 4

correct output
YES
4
1 2 1 2

user output
YES
6
1 1 1 2 1 2 

Test 19

Group: 3

Verdict: ACCEPTED

input
500 4
58 14 107 124 4 113 24 290 56 ...

correct output
YES
15698
389 392 395 398 401 404 407 41...

user output
YES
14588
1 1 389 392 395 398 401 404 40...
Truncated

Test 20

Group: 3

Verdict: ACCEPTED

input
500 4
113 187 278 242 23 67 48 298 3...

correct output
YES
15004
480 481 480 482 485 488 491 49...

user output
YES
13709
1 1 483 485 488 491 494 497 48...
Truncated

Test 21

Group: 3

Verdict: ACCEPTED

input
500 4
5 233 199 35 213 354 11 134 30...

correct output
YES
16770
458 461 464 467 470 473 476 47...

user output
YES
15398
1 1 458 461 464 467 470 473 47...
Truncated

Test 22

Group: 3

Verdict: ACCEPTED

input
500 4
169 47 21 137 57 138 360 147 4...

correct output
YES
15889
497 371 372 371 373 376 379 38...

user output
YES
14691
1 1 497 374 376 379 382 385 38...
Truncated

Test 23

Group: 3

Verdict: ACCEPTED

input
500 4
493 409 291 313 156 443 496 40...

correct output
YES
22886
480 481 480 482 485 488 491 49...

user output
YES
21619
1 1 483 485 488 491 494 497 13...
Truncated

Test 24

Group: 3

Verdict: ACCEPTED

input
500 4
137 99 100 226 326 298 140 340...

correct output
NO

user output
NO

Test 25

Group: 3

Verdict: ACCEPTED

input
500 4
500 499 498 497 496 495 494 49...

correct output
YES
41458
1 2 1 2 5 8 11 14 17 20 23 26 ...

user output
YES
41582
1 1 1 3 5 8 11 14 17 20 23 26 ...
Truncated

Test 26

Group: 4

Verdict: ACCEPTED

input
5 5
1 2 3 4 5

correct output
YES
0

user output
YES
2
1 1 

Test 27

Group: 4

Verdict: ACCEPTED

input
5 5
5 4 3 2 1

correct output
YES
1
1

user output
YES
3
1 1 1 

Test 28

Group: 4

Verdict: ACCEPTED

input
500 5
145 26 285 154 147 314 141 40 ...

correct output
YES
13786
216 220 224 228 232 236 240 24...

user output
YES
13008
1 1 216 220 224 228 232 236 24...
Truncated

Test 29

Group: 4

Verdict: ACCEPTED

input
500 5
137 22 399 292 249 6 51 224 42...

correct output
YES
13465
456 460 464 468 472 476 480 48...

user output
YES
12913
1 1 456 460 464 468 472 476 48...
Truncated

Test 30

Group: 4

Verdict: ACCEPTED

input
500 5
153 52 85 100 329 60 433 468 4...

correct output
YES
13642
377 378 377 380 384 388 392 39...

user output
YES
12956
1 1 381 384 388 392 396 400 40...
Truncated

Test 31

Group: 4

Verdict: ACCEPTED

input
500 5
267 326 95 108 189 32 291 366 ...

correct output
YES
14639
213 214 213 216 220 224 228 23...

user output
YES
13749
1 1 217 220 224 228 232 236 24...
Truncated

Test 32

Group: 4

Verdict: ACCEPTED

input
500 5
15 450 272 80 321 101 247 438 ...

correct output
NO

user output
NO

Test 33

Group: 4

Verdict: ACCEPTED

input
499 5
497 498 499 496 495 494 493 49...

correct output
YES
30886
3 7 11 15 19 23 27 31 35 39 43...

user output
YES
30878
1 1 3 7 11 15 19 23 27 31 35 3...
Truncated

Test 34

Group: 4

Verdict: ACCEPTED

input
500 5
499 500 497 498 495 496 493 49...

correct output
YES
30919
1 4 8 12 16 20 24 28 32 36 40 ...

user output
YES
30883
1 1 1 4 8 12 16 20 24 28 32 36...
Truncated