CSES - Datatähti 2024 alku - Results
Submission details
Task:Käännöt
Sender:ToukoP
Submission time:2023-11-05 23:03:11 +0200
Language:Rust
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.69 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.97 s3details
#190.00 s3details
#200.00 s3details
#210.00 s3details
#220.00 s3details
#230.00 s3details
#240.00 s3details
#250.00 s3details
#260.00 s4details
#270.00 s4details
#280.00 s4details
#290.00 s4details
#300.00 s4details
#310.00 s4details
#320.00 s4details
#330.00 s4details
#340.00 s4details

Code

use std::io::*;

const BUFFER_SIZE: usize = 5;

fn solve(list: &mut [i32], start: i32, n: i32, k: i32) -> Option<Vec<i32>> {
    let mut res = Vec::new();
    let mut reverse = |list: &mut [i32], p: i32| {
        #[cfg(feature = "print")]
        {
            println!("Reverse: {p} k: {k}");
            println!("{:?}", list);
        }

        res.push(p);
        list[(p as usize)..((p + k) as usize)].reverse();

        #[cfg(feature = "print")]
        {
            println!("{:?}", list);
        }
    };

    let get = |list: &[i32], i: i32| list[i as usize];

    let mut index = start;
    for i in start..(start + n) {
        if get(list, i) == start + 1 {
            index = i;
            break;
        }
    }

    #[cfg(feature = "print")]
    println!(
        "index: {}  target number: {}  start: {}",
        index,
        start + 1,
        start
    );

    if index == start {
        #[cfg(feature = "print")]
        println!("index not found\n");
        return Some(Vec::new());
    }

    let dist = (index - start) as i32;
    let pos = start + dist;
    let target = start + k - 1;
    loop {
        let tmp_target = target + k - 1;

        #[cfg(feature = "print")]
        println!("dist: {dist}  target pos: {target}  pos: {pos} tmp target pos: {tmp_target}");

        if pos == target {
            break;
        } else if pos > tmp_target {
            reverse(list, pos - k);
        } else if pos < tmp_target {
            reverse(list, (pos + target) / 2 - k / 2);
        } else {
            reverse(list, tmp_target - k + 1);
            break;
        }
    }
    reverse(list, start);

    #[cfg(feature = "print")]
    {
        println!("Done");
        println!("{:?}\n", list);
    }

    if let Some(r) = solve(list, start + 1, n - 1, k) {
        res.extend(r);
        Some(res)
    } else {
        None
    }
}

fn main() -> Result<()> {
    let mut input = String::new();
    stdin().read_line(&mut input)?;

    let mut split = input.split(" ");
    let n: i32 = split.next().unwrap().trim().parse().unwrap();
    let k: i32 = split.next().unwrap().trim().parse().unwrap();

    let mut list_input = String::new();
    stdin().read_line(&mut list_input)?;
    let vector = list_input
        .split(" ")
        .map(|s| s.trim().parse().unwrap())
        .collect::<Vec<i32>>();

    let mut list = [0; BUFFER_SIZE];
    list[(0 as usize)..(n as usize)].clone_from_slice(&vector);

    #[cfg(feature = "print")]
    println!("{:?}\n", list);

    if let Some(result) = solve(&mut list, 0, n, k) {
        println!(
            "YES\n{}",
            result
                .iter()
                .map(|n| n.to_string())
                .collect::<Vec<_>>()
                .join(" ")
        );
    } else {
        println!("NO")
    }

    Ok(())
}

Test details

Test 1

Group: 1

Verdict:

input
5 2
1 2 3 4 5

correct output
YES
0

user output
YES

Test 2

Group: 1

Verdict:

input
5 2
2 1 3 4 5

correct output
YES
1
1

user output
YES
0

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)

Error:
thread 'main' panicked at 'range end index 20 out of range for slice of length 5', input/c...

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)

Error:
thread 'main' panicked at 'range end index 100 out of range for slice of length 5', input/...

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)

Error:
thread 'main' panicked at 'range end index 100 out of range for slice of length 5', input/...

Test 6

Group: 2

Verdict:

input
5 3
1 2 3 4 5

correct output
YES
0

user output
YES

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
YES
0

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)

Error:
thread 'main' panicked at 'range end index 20 out of range for slice of length 5', input/c...

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)

Error:
thread 'main' panicked at 'range end index 20 out of range for slice of length 5', input/c...

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)

Error:
thread 'main' panicked at 'range end index 500 out of range for slice of length 5', input/...

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)

Error:
thread 'main' panicked at 'range end index 500 out of range for slice of length 5', input/...

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)

Error:
thread 'main' panicked at 'range end index 500 out of range for slice of length 5', input/...

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)

Error:
thread 'main' panicked at 'range end index 500 out of range for slice of length 5', input/...

Test 15

Group: 2

Verdict:

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

correct output
NO

user output
(empty)

Error:
thread 'main' panicked at 'range end index 500 out of range for slice of length 5', input/...

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)

Error:
thread 'main' panicked at 'range end index 500 out of range for slice of length 5', input/...

Test 17

Group: 3

Verdict:

input
5 4
1 2 3 4 5

correct output
YES
0

user output
YES

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)

Error:
thread 'main' panicked at 'range end index 500 out of range for slice of length 5', input/...

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)

Error:
thread 'main' panicked at 'range end index 500 out of range for slice of length 5', input/...

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)

Error:
thread 'main' panicked at 'range end index 500 out of range for slice of length 5', input/...

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)

Error:
thread 'main' panicked at 'range end index 500 out of range for slice of length 5', input/...

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)

Error:
thread 'main' panicked at 'range end index 500 out of range for slice of length 5', input/...

Test 24

Group: 3

Verdict:

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

correct output
NO

user output
(empty)

Error:
thread 'main' panicked at 'range end index 500 out of range for slice of length 5', input/...

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)

Error:
thread 'main' panicked at 'range end index 500 out of range for slice of length 5', input/...

Test 26

Group: 4

Verdict:

input
5 5
1 2 3 4 5

correct output
YES
0

user output
YES

Test 27

Group: 4

Verdict:

input
5 5
5 4 3 2 1

correct output
YES
1
1

user output
YES
0

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

Error:
thread 'main' panicked at 'range end index 500 out of range for slice of length 5', input/...

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

Error:
thread 'main' panicked at 'range end index 500 out of range for slice of length 5', input/...

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

Error:
thread 'main' panicked at 'range end index 500 out of range for slice of length 5', input/...

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

Error:
thread 'main' panicked at 'range end index 500 out of range for slice of length 5', input/...

Test 32

Group: 4

Verdict:

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

correct output
NO

user output
(empty)

Error:
thread 'main' panicked at 'range end index 500 out of range for slice of length 5', input/...

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

Error:
thread 'main' panicked at 'range end index 499 out of range for slice of length 5', input/...

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

Error:
thread 'main' panicked at 'range end index 500 out of range for slice of length 5', input/...