CSES - Datatähti 2021 loppu

Submission details
Task:	Koodarit
Sender:	xnor
Submission time:	2021-01-23 19:29:30 +0200
Language:	Rust
Status:	READY
Result:	0

Feedback
group	verdict	score
#1	RUNTIME ERROR	0
#2	RUNTIME ERROR	0
#3	RUNTIME ERROR	0

Test results
test	verdict	time	group
#1	ACCEPTED	0.01 s	1, 2, 3	details
#2	ACCEPTED	0.01 s	1, 2, 3	details
#3	RUNTIME ERROR	0.01 s	1, 2, 3	details
#4	ACCEPTED	0.01 s	1, 2, 3	details
#5	ACCEPTED	0.01 s	1, 2, 3	details
#6	RUNTIME ERROR	0.01 s	1, 2, 3	details
#7	RUNTIME ERROR	0.01 s	1, 2, 3	details
#8	ACCEPTED	0.01 s	1, 2, 3	details
#9	RUNTIME ERROR	0.01 s	1, 2, 3	details
#10	RUNTIME ERROR	0.01 s	1, 2, 3	details
#11	ACCEPTED	0.01 s	1, 2, 3	details
#12	RUNTIME ERROR	0.01 s	1, 3	details
#13	RUNTIME ERROR	0.01 s	1, 3	details
#14	RUNTIME ERROR	0.01 s	1, 3	details
#15	RUNTIME ERROR	0.01 s	1, 3	details
#16	RUNTIME ERROR	0.01 s	1, 3	details
#17	ACCEPTED	0.50 s	2, 3	details
#18	TIME LIMIT EXCEEDED	--	2, 3	details
#19	TIME LIMIT EXCEEDED	--	2, 3	details
#20	TIME LIMIT EXCEEDED	--	2, 3	details
#21	TIME LIMIT EXCEEDED	--	2, 3	details
#22	TIME LIMIT EXCEEDED	--	2, 3	details
#23	TIME LIMIT EXCEEDED	--	2, 3	details
#24	RUNTIME ERROR	0.33 s	3	details
#25	RUNTIME ERROR	0.34 s	3	details
#26	RUNTIME ERROR	0.34 s	3	details
#27	RUNTIME ERROR	0.34 s	3	details
#28	RUNTIME ERROR	0.34 s	3	details
#29	RUNTIME ERROR	0.34 s	3	details
#30	RUNTIME ERROR	0.34 s	3	details
#31	RUNTIME ERROR	0.34 s	3	details

Code

use std::io::{self, BufRead};

fn main() {
    let mut lines = io::BufReader::new(io::stdin()).lines();

    let line1 = lines.next().unwrap().unwrap();
    let line2 = lines.next().unwrap().unwrap();

    let l1: Vec<&str> = line1.split_ascii_whitespace().collect();
    let l2: Vec<&str> = line2.split_ascii_whitespace().collect();

    let n = l1[0].parse::<usize>().unwrap();
    let mut k = l1[1].parse::<usize>().unwrap();

    let mut xs = Vec::new();

    for i in 0..n {
        xs.push(l2[i].parse::<usize>().unwrap());
    }

    xs.sort_unstable();

    let mut i = 0;
    while i < xs.len() - 1 {
        if xs[i] == xs[i + 1] {
            k -= 1;
            if k == 0 {
                println!("0");
                return;
            }
            xs.remove(i);
            xs.remove(i);
            continue;
        }

        i += 1;
    }

    let mut chunks = Vec::new();
    for i in 0..xs.len() / 2 {
        let f = xs[2 * i];
        let l = xs[2 * i + 1];
        chunks.push(Chunk {
            len: 2,
            first: f,
            last: l,
            all: vec![0, l - f],
            no_first: vec![0],
            no_last: vec![0],
            no_firstlast: vec![0],
        });
    }
    if xs.len() % 2 == 1 {
        chunks.push(Chunk {
            len: 1,
            first: xs[xs.len() - 1],
            last: xs[xs.len() - 1],
            all: vec![0],
            no_first: vec![0],
            no_last: vec![0],
            no_firstlast: vec![0],
        });
    }

    loop {
        if chunks.len() == 1 {
            break;
        }

        let mut new_chunks = Vec::new();

        for k in 0..chunks.len() / 2 {
            let c1 = &chunks[2 * k];
            let c2 = &chunks[2 * k + 1];

            let mut c3 = Chunk {
                len: c1.len + c2.len,
                first: c1.first,
                last: c2.last,
                all: vec![0],
                no_first: vec![0],
                no_last: vec![0],
                no_firstlast: vec![0],
            };

            for m in 1..=(c3.len / 2).min(k+1) {
                let mut cost = usize::MAX;
                for j in m + 1 - c1.all.len().min(m+1)..c2.all.len().min(m + 1) {
                    let nc = c1.all[m - j] + c2.all[j];
                    if nc < cost {
                        cost = nc;
                    }
                }
                for j in m - c1.no_last.len().min(m)..c2.no_first.len().min(m) {
                    let nc = c1.no_last[m - 1 - j] + c2.no_first[j] + c2.first - c1.last;
                    if nc < cost {
                        cost = nc;
                    }
                }
                c3.all.push(cost);
            }

            for m in 1..=((c3.len - 1) / 2).min(k+1) {
                let mut cost = usize::MAX;
                for j in m + 1 - c1.no_first.len().min(m+1)..c2.all.len().min(m + 1) {
                    let nc = c1.no_first[m - j] + c2.all[j];
                    if nc < cost {
                        cost = nc;
                    }
                }
                for j in m - c1.no_firstlast.len().min(m)..c2.no_first.len().min(m) {
                    let nc = c1.no_firstlast[m - 1 - j] + c2.no_first[j] + c2.first - c1.last;
                    if nc < cost {
                        cost = nc;
                    }
                }
                c3.no_first.push(cost);
            }

            for m in 1..=((c3.len - 1) / 2).min(k+1) {
                let mut cost = usize::MAX;
                for j in m + 1 - c1.all.len().min(m+1)..c2.no_last.len().min(m + 1) {
                    let nc = c1.all[m - j] + c2.no_last[j];
                    if nc < cost {
                        cost = nc;
                    }
                }
                for j in m - c1.no_last.len().min(m)..c2.no_firstlast.len().min(m) {
                    let nc = c1.no_last[m - 1 - j] + c2.no_firstlast[j] + c2.first - c1.last;
                    if nc < cost {
                        cost = nc;
                    }
                }
                c3.no_last.push(cost);
            }

            for m in 1..=((c3.len - 2) / 2).min(k+1) {
                let mut cost = usize::MAX;
                for j in m + 1 - c1.no_first.len().min(m+1)..c2.no_last.len().min(m + 1) {
                    let nc = c1.no_first[m - j] + c2.no_last[j];
                    if nc < cost {
                        cost = nc;
                    }
                }
                for j in m - c1.no_firstlast.len().min(m)..c2.no_firstlast.len().min(m) {
                    let nc = c1.no_firstlast[m - 1 - j] + c2.no_firstlast[j] + c2.first - c1.last;
                    if nc < cost {
                        cost = nc;
                    }
                }
                c3.no_firstlast.push(cost);
            }

            new_chunks.push(c3);
        }

        if chunks.len() % 2 == 1 {
            new_chunks.push(chunks.last().unwrap().clone());
        }

        chunks = new_chunks;
    }

    println!("{}", chunks[0].all[k]);
}

#[derive(Clone, Debug)]
struct Chunk {
    len: usize,
    first: usize,
    last: usize,
    all: Vec<usize>,
    no_first: Vec<usize>,
    no_last: Vec<usize>,
    no_firstlast: Vec<usize>,
}

Test details

Test 1

Group: 1, 2, 3

Verdict: ACCEPTED

input
`8 3 3 1 2 7 9 3 4 7` view save

correct output
`1` view save

user output
`1` view save

Test 2

Group: 1, 2, 3

Verdict: ACCEPTED

input
`2 1 2 13` view save

correct output
`11` view save

user output
`11` view save

Test 3

Group: 1, 2, 3

Verdict: RUNTIME ERROR

input
`20 10 16 20 6 15 19 12 11 17 20 6 15...` view save

correct output
`6` view save

user output
(empty)

Error:

thread 'main' panicked at 'index out of bounds: the len is 2 but the index is 4', /usr/src...

Test 4

Group: 1, 2, 3

Verdict: ACCEPTED

input
`14 5 11 3 8 3 14 8 10 13 11 10 17 1...` view save

correct output
`0` view save

user output
`0` view save

Test 5

Group: 1, 2, 3

Verdict: ACCEPTED

input
`15 1 8 5 1 8 18 15 6 20 14 9 10 9 1...` view save

correct output
`0` view save

user output
`0` view save

Test 6

Group: 1, 2, 3

Verdict: RUNTIME ERROR

input
`10 3 10 19 16 15 12 5 14 8 3 15` view save

correct output
`4` view save

user output
(empty)

Error:

thread 'main' panicked at 'index out of bounds: the len is 2 but the index is 2', /usr/src...

Test 7

Group: 1, 2, 3

Verdict: RUNTIME ERROR

input
`202 90 177 187 183 647 616 580 499 78...` view save

correct output
`213` view save

user output
(empty)

Error:

thread 'main' panicked at 'index out of bounds: the len is 2 but the index is 75', /usr/sr...

Test 8

Group: 1, 2, 3

Verdict: ACCEPTED

input
`2000 512 141 583 135 833 900 308 248 58...` view save

correct output
`0` view save

user output
`0` view save

Test 9

Group: 1, 2, 3

Verdict: RUNTIME ERROR

input
`2000 972 685 4 289 865 93 159 48 866 56...` view save

correct output
`276` view save

user output
(empty)

Error:

thread 'main' panicked at 'index out of bounds: the len is 2 but the index is 209', /usr/s...

Test 10

Group: 1, 2, 3

Verdict: RUNTIME ERROR

input
`2000 1000 698 153 298 118 631 341 238 7 ...` view save

correct output
`517` view save

user output
(empty)

Error:

thread 'main' panicked at 'index out of bounds: the len is 2 but the index is 253', /usr/s...

Test 11

Group: 1, 2, 3

Verdict: ACCEPTED

input
`2000 1 983 144 449 584 839 166 77 885...` view save

correct output
`0` view save

user output
`0` view save

Test 12

Group: 1, 3

Verdict: RUNTIME ERROR

input
`1464 320 846762124 954854396 12767390 7...` view save

correct output
`35809369` view save

user output
(empty)

Error:

thread 'main' panicked at 'index out of bounds: the len is 2 but the index is 320', /usr/s...

Test 13

Group: 1, 3

Verdict: RUNTIME ERROR

input
`2000 231 801945178 924940258 369188694 ...` view save

correct output
`7831421` view save

user output
(empty)

Error:

thread 'main' panicked at 'index out of bounds: the len is 2 but the index is 231', /usr/s...

Test 14

Group: 1, 3

Verdict: RUNTIME ERROR

input
`2000 461 464790475 932031556 838378103 ...` view save

correct output
`37272564` view save

user output
(empty)

Error:

thread 'main' panicked at 'index out of bounds: the len is 2 but the index is 461', /usr/s...

Test 15

Group: 1, 3

Verdict: RUNTIME ERROR

input
`2000 100 484046702 267135814 995006323 ...` view save

correct output
`1268400` view save

user output
(empty)

Error:

thread 'main' panicked at 'index out of bounds: the len is 2 but the index is 100', /usr/s...

Test 16

Group: 1, 3

Verdict: RUNTIME ERROR

input
`2000 996 98352148 438929491 242618159 1...` view save

correct output
`445965905` view save

user output
(empty)

Error:

thread 'main' panicked at 'index out of bounds: the len is 2 but the index is 996', /usr/s...

Test 17

Group: 2, 3

Verdict: ACCEPTED

input
`65879 19675 896 316 972 476 636 227 716 78...` view save

correct output
`0` view save

user output
`0` view save

Test 18

Group: 2, 3

Verdict: TIME LIMIT EXCEEDED

input
`200000 53820 995 720 135 767 943 742 191 26...` view save

correct output
`0` view save

user output
(empty)

Test 19

Group: 2, 3

Verdict: TIME LIMIT EXCEEDED

input
`200000 32297 527 947 84 851 908 833 339 112...` view save

correct output
`0` view save

user output
(empty)

Test 20

Group: 2, 3

Verdict: TIME LIMIT EXCEEDED

input
`200000 99982 561 174 242 275 460 109 664 68...` view save

correct output
`322` view save

user output
(empty)

Test 21

Group: 2, 3

Verdict: TIME LIMIT EXCEEDED

input
`200000 99955 911 33 314 861 298 117 972 982...` view save

correct output
`245` view save

user output
(empty)

Test 22

Group: 2, 3

Verdict: TIME LIMIT EXCEEDED

input
`200000 99972 783 1000 673 611 87 452 702 92...` view save

correct output
`290` view save

user output
(empty)

Test 23

Group: 2, 3

Verdict: TIME LIMIT EXCEEDED

input
`200000 99961 795 136 128 643 60 422 371 839...` view save

correct output
`252` view save

user output
(empty)

Test 24

Group: 3

Verdict: RUNTIME ERROR

input
`195426 76599 442872072 619088799 118541378 ...` view save

correct output
`143376538` view save

user output
(empty)

Error:

thread 'main' panicked at 'index out of bounds: the len is 2 but the index is 76577', /usr...

Test 25

Group: 3

Verdict: RUNTIME ERROR

input
`200000 1661 894106972 620084612 931442312 ...` view save

correct output
`33089` view save

user output
(empty)

Error:

thread 'main' panicked at 'index out of bounds: the len is 2 but the index is 1638', /usr/...

Test 26

Group: 3

Verdict: RUNTIME ERROR

input
`200000 86032 211444153 846442677 297198384 ...` view save

correct output
`196001810` view save

user output
(empty)

Error:

thread 'main' panicked at 'index out of bounds: the len is 2 but the index is 86005', /usr...

Test 27

Group: 3

Verdict: RUNTIME ERROR

input
`200000 28275 28280312 349705372 96535649 84...` view save

correct output
`11627219` view save

user output
(empty)

Error:

thread 'main' panicked at 'index out of bounds: the len is 2 but the index is 28254', /usr...

Test 28

Group: 3

Verdict: RUNTIME ERROR

input
`200000 81473 178022892 112774306 250584651 ...` view save

correct output
`162430841` view save

user output
(empty)

Error:

thread 'main' panicked at 'index out of bounds: the len is 2 but the index is 81448', /usr...

Test 29

Group: 3

Verdict: RUNTIME ERROR

input
`200000 99987 297598052 494409138 182268523 ...` view save

correct output
`489497036` view save

user output
(empty)

Error:

thread 'main' panicked at 'index out of bounds: the len is 2 but the index is 99969', /usr...

Test 30

Group: 3

Verdict: RUNTIME ERROR

input
`200000 99971 316462272 843156468 434342923 ...` view save

correct output
`483167476` view save

user output
(empty)

Error:

thread 'main' panicked at 'index out of bounds: the len is 2 but the index is 99947', /usr...

Test 31

Group: 3

Verdict: RUNTIME ERROR

input
`200000 99964 811543559 465033274 620180191 ...` view save

correct output
`481497328` view save

user output
(empty)

Error:

thread 'main' panicked at 'index out of bounds: the len is 2 but the index is 99937', /usr...

Datatähti 2021 loppu

Koodarit

Code

Test details

Test 1

Test 2

Test 3

Test 4

Test 5

Test 6

Test 7

Test 8

Test 9

Test 10

Test 11

Test 12

Test 13

Test 14

Test 15

Test 16

Test 17

Test 18

Test 19

Test 20

Test 21

Test 22

Test 23

Test 24

Test 25

Test 26

Test 27

Test 28

Test 29

Test 30

Test 31

Tasks

Submissions (xnor)