CSES - Datatähti 2025 alku - Results
Submission details
Task:Tikut
Sender:Saksunoob
Submission time:2024-10-30 21:21:58 +0200
Language:Rust (2021)
Status:READY
Result:27
Feedback
groupverdictscore
#1ACCEPTED7
#2ACCEPTED8
#3ACCEPTED12
#40
#50
#60
Test results
testverdicttimegroup
#1ACCEPTED0.00 s1, 3, 4, 5, 6details
#2ACCEPTED0.00 s1, 4, 5, 6details
#3ACCEPTED0.00 s1, 4, 5, 6details
#4ACCEPTED0.00 s1, 4, 5, 6details
#5ACCEPTED0.00 s2, 5, 6details
#6ACCEPTED0.00 s2, 5, 6details
#7ACCEPTED0.00 s3, 5, 6details
#8ACCEPTED0.00 s3, 5, 6details
#9ACCEPTED0.00 s3, 5, 6details
#10ACCEPTED0.00 s3, 5, 6details
#11ACCEPTED0.00 s3, 5, 6details
#12--4, 5, 6details
#13--4, 5, 6details
#14--4, 5, 6details
#15--4, 5, 6details
#16--5, 6details
#17--5, 6details
#18--5, 6details
#19--5, 6details
#200.46 s6details
#210.41 s6details
#220.46 s6details

Code

use std::{cmp, collections::HashMap, io, usize};

fn max_length_difference(
    mut lengths: Vec<(usize, usize)>,
    max_cuts: usize,
) -> Vec<usize> {
    let mut results = Vec::with_capacity(max_cuts);
    let largest = lengths.pop().unwrap();

    if lengths.len() != 0 {
        results.resize(cmp::min(max_cuts, largest.1-1), largest.0 - cmp::min(lengths.first().unwrap().0, largest.0/2));
        if max_cuts >= largest.1 {
            results.push(cmp::max(lengths.last().unwrap().0, largest.0-largest.0/2) - cmp::min(lengths.first().unwrap().0, largest.0/2));
        }
    } else {
        results.resize(cmp::min(max_cuts, largest.1-1), largest.0 - largest.0/2);
        if max_cuts >= largest.1 {
            results.push(0);
        }
    }

    if max_cuts <= largest.1 {
        assert_eq!(results.len(), max_cuts);
        return results;
    }

    let mut mins = vec![usize::MAX; max_cuts-largest.1];

    for p in 1..=largest.0/2 {
        let mut lenghts_copy = lengths.clone();
        insert_length(&mut lenghts_copy, p, largest.1);
        insert_length(&mut lenghts_copy, largest.0-p, largest.1);

        let inner_lengths = max_length_difference(lenghts_copy, max_cuts-largest.1);

        for i in 0..mins.len() {
            mins[i] = cmp::min(mins[i], inner_lengths[i]);
        }
    }

    results.append(&mut mins);
    assert_eq!(results.len(), max_cuts);
    return results;
}

fn insert_length(lengths: &mut Vec<(usize, usize)>, length: usize, count: usize) {
    for i in (0..lengths.len()).rev() {
        if lengths[i].0 < length {
            lengths.insert(i + 1, (length, count));
            return;
        } else if lengths[i].0 == length {
            lengths[i].1 += count;
            return;
        }
    }
    lengths.insert(0, (length, count));
}

fn main() {
    let mut in_str = String::new();
    io::stdin().read_line(&mut in_str).unwrap();
    in_str = in_str.trim().to_string();
    let (n_str, m_str) = in_str.trim().split_once(' ').unwrap();

    let _n: usize = n_str.parse().unwrap();
    let m: usize = m_str.parse().unwrap();

    let mut lengths_map: HashMap<usize, usize> = HashMap::new();

    let mut lengths_str = String::new();
    io::stdin().read_line(&mut lengths_str).unwrap();

    for l in lengths_str.split_whitespace() {
        let key = l.parse().unwrap();
        if lengths_map.contains_key(&key) {
            *lengths_map.get_mut(&key).unwrap() += 1;
        } else {
            lengths_map.insert(key, 1);
        }
    }

    let mut lengths_vec: Vec<(usize, usize)> = lengths_map.into_iter().collect();
    lengths_vec.sort_by_key(|k| k.0);

    println!("{}", max_length_difference(lengths_vec.clone(), m).iter().map(|v| v.to_string()).collect::<Vec<String>>().join(" "));
}

Test details

Test 1

Group: 1, 3, 4, 5, 6

Verdict: ACCEPTED

input
1 1
6

correct output

user output
0

Test 2

Group: 1, 4, 5, 6

Verdict: ACCEPTED

input
5 10
4 8 6 2 7

correct output
5 4 2 2 2 1 1 1 1 1 

user output
5 4 2 2 2 1 1 1 1 1

Test 3

Group: 1, 4, 5, 6

Verdict: ACCEPTED

input
5 10
5 5 8 6 7

correct output
3 3 2 3 2 2 1 1 1 2 

user output
3 3 2 3 2 2 1 1 1 2

Test 4

Group: 1, 4, 5, 6

Verdict: ACCEPTED

input
5 10
8 7 9 6 10

correct output
4 4 3 3 2 2 1 2 2 1 

user output
4 4 3 3 2 2 1 2 2 1

Test 5

Group: 2, 5, 6

Verdict: ACCEPTED

input
1000 1071
3 2 3 1 3 3 2 3 2 3 2 2 2 1 2 ...

correct output
2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 ...

user output
2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 ...

Test 6

Group: 2, 5, 6

Verdict: ACCEPTED

input
1000 1500
3 2 2 3 2 3 2 2 2 3 2 2 3 3 3 ...

correct output
2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 ...

user output
2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 ...

Test 7

Group: 3, 5, 6

Verdict: ACCEPTED

input
1000 2
15 710 210 347 398 66 318 277 ...

correct output
994 994 

user output
994 994

Test 8

Group: 3, 5, 6

Verdict: ACCEPTED

input
1000 2
743 890 592 942 736 969 616 50...

correct output
498 496 

user output
498 496

Test 9

Group: 3, 5, 6

Verdict: ACCEPTED

input
1000 2
987 968 920 994 988 918 914 95...

correct output
500 500 

user output
500 500

Test 10

Group: 3, 5, 6

Verdict: ACCEPTED

input
1000 2
996 1000 998 998 999 997 997 9...

correct output
500 500 

user output
500 500

Test 11

Group: 3, 5, 6

Verdict: ACCEPTED

input
1000 2
501 501 501 501 501 501 501 50...

correct output
1 168 

user output
1 168

Test 12

Group: 4, 5, 6

Verdict:

input
100 200
145 136 74 83 73 36 196 115 11...

correct output
194 190 189 183 182 181 181 17...

user output
(empty)

Test 13

Group: 4, 5, 6

Verdict:

input
100 200
157 110 168 155 192 107 146 15...

correct output
95 96 96 95 93 94 94 94 90 91 ...

user output
(empty)

Test 14

Group: 4, 5, 6

Verdict:

input
50 200
137 118 160 118 146 160 140 18...

correct output
98 98 98 96 90 91 88 88 84 86 ...

user output
(empty)

Test 15

Group: 4, 5, 6

Verdict:

input
100 200
147 174 186 148 155 128 158 18...

correct output
99 99 98 98 97 97 96 96 95 95 ...

user output
(empty)

Test 16

Group: 5, 6

Verdict:

input
1000 2000
928772177 816188227 216592201 ...

correct output
991676844 990940224 990685481 ...

user output
(empty)

Test 17

Group: 5, 6

Verdict:

input
1000 2000
665759876 597950008 615453266 ...

correct output
498801198 498681904 498504321 ...

user output
(empty)

Test 18

Group: 5, 6

Verdict:

input
500 2000
683288817 784230412 626685186 ...

correct output
497667621 498434895 495465990 ...

user output
(empty)

Test 19

Group: 5, 6

Verdict:

input
1000 2000
666667000 809309500 571572000 ...

correct output
499499500 499249250 498999000 ...

user output
(empty)

Test 20

Group: 6

Verdict:

input
100000 200000
861772559 734298084 983382252 ...

correct output
499973914 499985299 499985141 ...

user output
(empty)

Test 21

Group: 6

Verdict:

input
30000 200000
691834579 617419813 514778075 ...

correct output
499967533 499976270 499969810 ...

user output
(empty)

Test 22

Group: 6

Verdict:

input
100000 200000
820255000 960780000 741965000 ...

correct output
499995000 499992500 499990000 ...

user output
(empty)