CSES - Datatähti 2024 alku - Results
Submission details
Task:Käännöt
Sender:perusjuntti
Submission time:2023-11-02 17:19:43 +0200
Language:C++ (C++11)
Status:READY
Result:0
Feedback
groupverdictscore
#10
#20
#30
#40
Test results
testverdicttimegroup
#10.00 s1details
#20.00 s1details
#30.00 s1details
#40.00 s1details
#50.00 s1details
#60.00 s2details
#70.00 s2details
#80.00 s2details
#90.00 s2details
#100.00 s2details
#110.00 s2details
#120.00 s2details
#130.00 s2details
#140.00 s2details
#150.00 s2details
#160.00 s2details
#170.00 s3details
#180.00 s3details
#190.00 s3details
#200.00 s3details
#210.00 s3details
#220.00 s3details
#230.00 s3details
#240.01 s3details
#250.00 s3details
#26ACCEPTED0.00 s4details
#27ACCEPTED0.00 s4details
#280.00 s4details
#290.00 s4details
#300.00 s4details
#310.00 s4details
#32ACCEPTED0.00 s4details
#330.01 s4details
#34ACCEPTED0.01 s4details

Compiler report

input/code.cpp: In function 'void k_5(int, std::vector<int>)':
input/code.cpp:151:10: warning: variable 'skip' set but not used [-Wunused-but-set-variable]
  151 |     bool skip = 1;
      |          ^~~~

Code

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

//hitto mikä tehtävä :D
//:> :> :> :> :> :> :>

bool sorted = 0;
set<vector<int>> st, st2;
vector<int> ops_5, ops_6, ops_7, ops_other;

void sort_other_k5(vector<int> &a, int n) {
    for (int i = 1; i <= n-7; ++i) {
        for (int j = 0; j < n; ++j) {
            if (a[j] == i) {
                if (j == i-1)
                    break;
                if (i % 4 == (j+1) % 4) {
                    for (int k = j-4; k >= i-1; k -= 4) {
                        swap(a[k],a[k+4]);
                        swap(a[k+1],a[k+3]);
                        ops_other.push_back(k+1);
                    } 
                }
                else {
                    if (j < n-5) {
                        swap(a[j-1],a[j+3]);
                        swap(a[j],a[j+2]);
                        ops_other.push_back(j);
                        j += 2;
                    }   
                    else {
                        swap(a[j-3],a[j+1]);
                        swap(a[j-2],a[j]);
                        ops_other.push_back(j-2);
                        j -= 2;
                    }
                    for (int k = j-4; k >= i-1; k -= 4) {
                        swap(a[k],a[k+4]);
                        swap(a[k+1],a[k+3]);
                        ops_other.push_back(k+1);
                    }
                }
            }
        }
    }
}

bool sort_6_k5(vector<int> a) {
    bool ok = 1;
    for (int i = 0; i < 6; ++i) {
        if (a[i] != i+1)
            ok = 0; 
        }
        if (ok)
            return ok;
    for (int i = 1; i <= 20; ++i) {
        if (i & 1) {
            swap(a[0],a[4]);
            swap(a[1],a[3]);
            ops_6.push_back(1);
        }
        else {
            swap(a[1],a[5]);
            swap(a[2],a[4]);
            ops_6.push_back(2);
        }
        ok = 1;
        for (int i = 0; i < 6; ++i) {
            if (a[i] != i+1)
                ok = 0; 
        }
        if (ok)
            break;
    }
    return ok;
}

void sort_7_k5(vector<int> a, vector<int> ops) {
    if (sorted)
        return;
    if (st.count(a))
        return;
    st.insert(a);
    bool ok = 1;
    for (int i = 0; i < 7; ++i) 
        if (a[i] != i+1)
            ok = 0;
    if (ok) {
        sorted = 1;
        for (int &i:ops) 
            ops_7.push_back(i);
        return;
    }
    else {
        swap(a[0],a[4]);
        swap(a[1],a[3]);
        ops.push_back(1);
        sort_7_k5(a,ops);
        ops.pop_back();
        swap(a[0],a[4]);
        swap(a[1],a[3]);
        
        swap(a[1],a[5]);
        swap(a[2],a[4]);
        ops.push_back(2);
        sort_7_k5(a,ops);
        ops.pop_back();
        swap(a[1],a[5]);
        swap(a[2],a[4]);

        swap(a[2],a[6]);
        swap(a[3],a[5]);
        ops.push_back(3);
        sort_7_k5(a,ops);
    }
}

void k_5(int n, vector<int> a) {
    vector<int> ops, rest;
    bool good = 1;
    for (int i = 0; i < n; ++i) {
        if ((a[i] & 1) != ((i+1) & 1) )
            good = 0;
    }
    if (!good) {
        cout << "NO\n";
        return;
    }
    if (n == 5) {
        vector<int> s = {1,2,3,4,5}, r = {5,4,3,2,1};
        if (a == s) 
            cout << "YES\n" << 0 << '\n';
        else if (a == r) 
            cout << "YES\n" << 1 << '\n' << 1 << '\n';
        else
            cout << "NO\n";
        return;
    }
    if (n == 6) {
        if (sort_6_k5(a)) {
            cout << "YES\n";
            cout << ops_6.size() << '\n';
            for (int &i:ops_6) cout << i << ' ';
            cout << '\n';
        }
        else 
            cout << "NO\n";
        return;
    }
    sort_other_k5(a,n);
    bool skip = 1;
    for (int i = 0; i < n; ++i) 
        if (a[i] != i+1)
            skip = 0;
    for (int i = n-7; i < n; ++i) {
        rest.push_back(a[i]-(n-7));
    }
    sort_7_k5(rest,ops);
    if (sorted) {
        cout << "YES\n";
        cout << ops_7.size() + ops_other.size() << '\n';
        for (int &i:ops_other) 
            cout << i << ' ';
        //cout << '\n';
        for (int &i:ops_7)
            cout << i+n-7 << ' ';
        cout << '\n';
    }
    else
        cout << "NO\n";
}

void swapp(vector<int> a) {
    st2.insert(a);
    
    swap(a[0],a[4]);
    swap(a[1],a[3]);
    if (!st2.count(a)) {
        st2.insert(a);
        swapp(a);    
    }
    swap(a[0],a[4]);
    swap(a[1],a[3]);
    
    swap(a[1],a[5]);
    swap(a[2],a[4]);
    if (!st2.count(a)) {
        st2.insert(a);
        swapp(a);  
    }
    swap(a[1],a[5]);
    swap(a[2],a[4]);

    swap(a[2],a[6]);
    swap(a[3],a[5]);
    if (!st2.count(a)) {
        st2.insert(a);
        swapp(a);
    }
    swap(a[2],a[6]);
    swap(a[3],a[5]);
  
    swap(a[3],a[7]);
    swap(a[4],a[6]);
    if (!st2.count(a)) {
        st2.insert(a);
        swapp(a);
    }
    swap(a[3],a[7]);
    swap(a[4],a[6]);
    
    swap(a[4],a[8]);
    swap(a[5],a[7]);
    if (!st2.count(a)) {
        st2.insert(a);
        swapp(a);
    }
    swap(a[4],a[8]);
    swap(a[5],a[7]);

}

int main() {
    int n, k;
    cin >> n >> k;
    vector<int> a(n);
    for (int &i:a) cin >> i;
    if (k == 5) 
        k_5(n,a);
    /*
    vector<int> a = {1, 2, 3, 4, 5, 6, 7, 8, 9};
    swapp(a);
    vector<vector<int>> v;
    do {
        if (st2.count(a)) {
            v.push_back(a);
        }
    } while (next_permutation(a.begin(),a.end()));
    for (auto u : v) {
        k_5(9,u);
        for (int &i:u) cout << i << ' ';
        cout << '\n';
        sorted = 0;
        st.clear();
        ops_5.clear();
        ops_6.clear();
        ops_7.clear();
        ops_other.clear();
    }
    */
}

Test details

Test 1

Group: 1

Verdict:

input
5 2
1 2 3 4 5

correct output
YES
0

user output
(empty)

Test 2

Group: 1

Verdict:

input
5 2
2 1 3 4 5

correct output
YES
1
1

user output
(empty)

Test 3

Group: 1

Verdict:

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

Test 4

Group: 1

Verdict:

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

Test 5

Group: 1

Verdict:

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

Test 6

Group: 2

Verdict:

input
5 3
1 2 3 4 5

correct output
YES
0

user output
(empty)

Test 7

Group: 2

Verdict:

input
5 3
3 5 4 1 2

correct output
NO

user output
(empty)

Test 8

Group: 2

Verdict:

input
5 3
5 2 1 4 3

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

user output
(empty)

Test 9

Group: 2

Verdict:

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

Test 10

Group: 2

Verdict:

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

Test 11

Group: 2

Verdict:

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

Test 12

Group: 2

Verdict:

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

Test 13

Group: 2

Verdict:

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

Test 14

Group: 2

Verdict:

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

Test 15

Group: 2

Verdict:

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

correct output
NO

user output
(empty)

Test 16

Group: 2

Verdict:

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

Test 17

Group: 3

Verdict:

input
5 4
1 2 3 4 5

correct output
YES
0

user output
(empty)

Test 18

Group: 3

Verdict:

input
5 4
5 1 2 3 4

correct output
YES
4
1 2 1 2

user output
(empty)

Test 19

Group: 3

Verdict:

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

Test 20

Group: 3

Verdict:

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

Test 21

Group: 3

Verdict:

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

Test 22

Group: 3

Verdict:

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

Test 23

Group: 3

Verdict:

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

Test 24

Group: 3

Verdict:

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

correct output
NO

user output
(empty)

Test 25

Group: 3

Verdict:

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

Test 26

Group: 4

Verdict: ACCEPTED

input
5 5
1 2 3 4 5

correct output
YES
0

user output
YES
0

Test 27

Group: 4

Verdict: ACCEPTED

input
5 5
5 4 3 2 1

correct output
YES
1
1

user output
YES
1
1

Test 28

Group: 4

Verdict:

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
NO

Test 29

Group: 4

Verdict:

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
NO

Test 30

Group: 4

Verdict:

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
NO

Test 31

Group: 4

Verdict:

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
NO

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:

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
NO

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
30927
496 493 489 485 481 477 473 46...
Truncated