CSES - Datatähti 2024 alku - Results
Submission details
Task:Käännöt
Sender:stpn129
Submission time:2023-11-09 10:34:05 +0200
Language:C++ (C++11)
Status:READY
Result:0
Feedback
groupverdictscore
#10
#20
#30
#40
Test results
testverdicttimegroup
#10.04 s1details
#20.04 s1details
#30.04 s1details
#40.04 s1details
#50.04 s1details
#60.04 s2details
#70.04 s2details
#80.04 s2details
#90.04 s2details
#100.04 s2details
#110.04 s2details
#120.04 s2details
#130.04 s2details
#140.04 s2details
#150.04 s2details
#160.05 s2details
#170.04 s3details
#180.04 s3details
#190.04 s3details
#200.04 s3details
#210.04 s3details
#220.04 s3details
#230.04 s3details
#240.04 s3details
#250.04 s3details
#260.04 s4details
#270.04 s4details
#280.04 s4details
#290.04 s4details
#300.04 s4details
#310.04 s4details
#320.04 s4details
#330.04 s4details
#340.04 s4details

Compiler report

input/code.cpp: In function 'void check()':
input/code.cpp:187:31: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  187 |             for (int i = 0; i < a.size(); ++i) {
      |                             ~~^~~~~~~~~~
input/code.cpp:191:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  191 |                 if (i == a.size() - 1) cout << "}";
      |                     ~~^~~~~~~~~~~~~~~
input/code.cpp:199:31: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  199 |             for (int i = 0; i < b.size(); ++i) {
      |                             ~~^~~~~~~~~~
input/code.cpp:203:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<in...

Code

#include <bits/stdc++.h>

using namespace std;
 
void init_code() {
    ios_base::sync_with_stdio(0);
    cin.tie(0);
#ifndef ONLINE_JUDGE
    freopen("input.txt", "r", stdin);
    freopen("output.txt", "w", stdout);
#endif
}
 
bool is_ok(int j, int i, int n, int k) {
    if (j < 1 || (j + (k - 1)) > n || j < i) {
        return false;
    }
 
    return true;
}
 
void rev(int i, int k, vector<int>& a, vector<int>& pos, vector<int>& res) {
    reverse(a.begin() + i, a.begin() + i + k);
 
    for (int ik = 0; ik <= k - 1; ++ik) {
        pos[a[i + ik]] = i + ik;
    }
 
    res.push_back(i);
}
 
vector<int> compress(vector<int>& a) {
    vector<int> b = a;
 
    sort(b.begin(), b.end());
    b.erase(unique(b.begin(), b.end()), b.end());
 
    for (int &x : a)
        x = int(lower_bound(b.begin(), b.end(), x) - b.begin()) + 1;
 
    return a;
}

vector<int> solve2(int n, int k, vector<int> a) {
    queue<vector<int>>  q;
    q.push(a);

    map<vector<int>, int> d;
    d[a] = 0;

    map<vector<int>, vector<int>> prev;
    
    map<vector<int>, int> prev_op;
    prev_op[a] = -1;
    while (!q.empty()) {
        vector<int> v = q.front();
        q.pop();

        /*if (d[v] > 10) {
            break;
        }*/

        if (v[0] == 1) {
            //cout << d[v] << '\n';
            
            vector<int> ops;
            while (prev_op[v] != -1) {
                ops.push_back(prev_op[v] + 1);
                v = prev[v];
            }

            reverse(ops.begin(), ops.end());

            return ops;
        }

        for (int i = 0; i < n - (k - 1); ++i) {
            vector<int> next = v;
            reverse(next.begin() + i, next.begin() + i + k);
            if (!d.count(next) || d[next] > d[v] + 1) {
                q.push(next);
                prev[next] = v;
                prev_op[next] = i;
                d[next] = d[v] + 1;
            }
        }
    }

    return vector<int> {-1};
}

 
vector<int> solve(int n, int k, vector<int> a) {
    a.insert(a.begin(), -1);

    vector<int> pos(n + 1);
    for (int i = 1; i <= n; ++i) {
        pos[a[i]] = i;
    }

    vector<int> res;
 
    if (a[1] == 1) {
        return vector<int> {-1};
    }
     
    /*if (n <= 8) {
        a.erase(a.begin());
 
        res = solve2(n, k, a);

        return res;
    }*/
 
    for (int i = 1; i <= 1; ++i) {
        int j = pos[i];
        
        vector<int> fix;
        if ((j - i) % (k - 1) != 0) {
            queue<pair<vector<int>, int>> q;    
            q.push({vector<int> (), j});
 
            int x = 0;
            while (!q.empty() && x <= 10) {
                vector<int> op = q.front().first;
                int cur_j = q.front().second;
 
                q.pop();
 
                if (cur_j < i) {
                    continue;
                }
 
                if ((cur_j - i) % (k - 1) == 0) {
                    fix = op;
                    break;
                }
 
                for (int it = cur_j - (k - 1); it <= cur_j; ++it) {
                    if (is_ok(it, i, n, k)) {
                        op.push_back(it);
                        int new_j = it + k - (abs(cur_j - it) + 1);
                        q.push({op, new_j});
                        op.pop_back();
                        x++;
                    }
                }
            }
        }
        
        for (auto it : fix) {
            rev(it, k, a, pos, res);
        }
        
        j = pos[i];
 
        if ((j - i) % (k - 1) == 0) {
            while (j != i) {
                rev(j - (k - 1), k, a, pos, res);
                j -= (k - 1);
            }
        } else {
            break;
        }
    }

    if (a[1] == 1) return res;

    return vector<int> {-1};
}


void check() {
    int n = 8, k = 5;

    vector<int> a(n);

    iota(a.begin(), a.end(), 1);

    int t = 0;
    do {
        //random_shuffle(a.begin(), a.end());

        vector<int> b = solve(n, k, a);
        if (b.size() > 0 && b[0] != -1) {
            cout << "mp[";
            for (int i = 0; i < a.size(); ++i) {
                if (i == 0) cout << "{";
                cout << a[i];
                //cout << ' ';
                if (i == a.size() - 1) cout << "}";
                else cout << ", ";
            }
            //cout << '\n';
            //cout << b.size() << '\n';
            
            cout << "] = ";
            
            for (int i = 0; i < b.size(); ++i) {
                if (i == 0) cout << "{";
                cout << b[i];
                //cout << ' ';
                if (i == b.size() - 1) cout << "}";
                else cout << ", ";
            }
            //cout << '\n';
            cout << ";\n";
        }
        t++;
    } while (next_permutation(a.begin(), a.end()));  
}

signed main() {
    init_code();
    int t = 1;
    // cin >> t;
    while (t--) {
        check();
        //solve();
    }
    return 0;
}

Test details

Test 1

Group: 1

Verdict:

input
5 2
1 2 3 4 5

correct output
YES
0

user output
mp[{2, 3, 1, 4, 5, 6, 7, 8}] =...
Truncated

Test 2

Group: 1

Verdict:

input
5 2
2 1 3 4 5

correct output
YES
1
1

user output
mp[{2, 3, 1, 4, 5, 6, 7, 8}] =...
Truncated

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
mp[{2, 3, 1, 4, 5, 6, 7, 8}] =...
Truncated

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
mp[{2, 3, 1, 4, 5, 6, 7, 8}] =...
Truncated

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
mp[{2, 3, 1, 4, 5, 6, 7, 8}] =...
Truncated

Test 6

Group: 2

Verdict:

input
5 3
1 2 3 4 5

correct output
YES
0

user output
mp[{2, 3, 1, 4, 5, 6, 7, 8}] =...
Truncated

Test 7

Group: 2

Verdict:

input
5 3
3 5 4 1 2

correct output
NO

user output
mp[{2, 3, 1, 4, 5, 6, 7, 8}] =...
Truncated

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
mp[{2, 3, 1, 4, 5, 6, 7, 8}] =...
Truncated

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
mp[{2, 3, 1, 4, 5, 6, 7, 8}] =...
Truncated

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
mp[{2, 3, 1, 4, 5, 6, 7, 8}] =...
Truncated

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
mp[{2, 3, 1, 4, 5, 6, 7, 8}] =...
Truncated

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
mp[{2, 3, 1, 4, 5, 6, 7, 8}] =...
Truncated

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
mp[{2, 3, 1, 4, 5, 6, 7, 8}] =...
Truncated

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
mp[{2, 3, 1, 4, 5, 6, 7, 8}] =...
Truncated

Test 15

Group: 2

Verdict:

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

correct output
NO

user output
mp[{2, 3, 1, 4, 5, 6, 7, 8}] =...
Truncated

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
mp[{2, 3, 1, 4, 5, 6, 7, 8}] =...
Truncated

Test 17

Group: 3

Verdict:

input
5 4
1 2 3 4 5

correct output
YES
0

user output
mp[{2, 3, 1, 4, 5, 6, 7, 8}] =...
Truncated

Test 18

Group: 3

Verdict:

input
5 4
5 1 2 3 4

correct output
YES
4
1 2 1 2

user output
mp[{2, 3, 1, 4, 5, 6, 7, 8}] =...
Truncated

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
mp[{2, 3, 1, 4, 5, 6, 7, 8}] =...
Truncated

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
mp[{2, 3, 1, 4, 5, 6, 7, 8}] =...
Truncated

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
mp[{2, 3, 1, 4, 5, 6, 7, 8}] =...
Truncated

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
mp[{2, 3, 1, 4, 5, 6, 7, 8}] =...
Truncated

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
mp[{2, 3, 1, 4, 5, 6, 7, 8}] =...
Truncated

Test 24

Group: 3

Verdict:

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

correct output
NO

user output
mp[{2, 3, 1, 4, 5, 6, 7, 8}] =...
Truncated

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
mp[{2, 3, 1, 4, 5, 6, 7, 8}] =...
Truncated

Test 26

Group: 4

Verdict:

input
5 5
1 2 3 4 5

correct output
YES
0

user output
mp[{2, 3, 1, 4, 5, 6, 7, 8}] =...
Truncated

Test 27

Group: 4

Verdict:

input
5 5
5 4 3 2 1

correct output
YES
1
1

user output
mp[{2, 3, 1, 4, 5, 6, 7, 8}] =...
Truncated

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
mp[{2, 3, 1, 4, 5, 6, 7, 8}] =...
Truncated

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
mp[{2, 3, 1, 4, 5, 6, 7, 8}] =...
Truncated

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
mp[{2, 3, 1, 4, 5, 6, 7, 8}] =...
Truncated

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
mp[{2, 3, 1, 4, 5, 6, 7, 8}] =...
Truncated

Test 32

Group: 4

Verdict:

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

correct output
NO

user output
mp[{2, 3, 1, 4, 5, 6, 7, 8}] =...
Truncated

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
mp[{2, 3, 1, 4, 5, 6, 7, 8}] =...
Truncated

Test 34

Group: 4

Verdict:

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
mp[{2, 3, 1, 4, 5, 6, 7, 8}] =...
Truncated