| Task: | Tikut | 
| Sender: | EmuBird | 
| Submission time: | 2024-11-09 17:20:21 +0200 | 
| Language: | Rust (2021) | 
| Status: | READY | 
| Result: | 27 | 
| group | verdict | score | 
|---|---|---|
| #1 | ACCEPTED | 7 | 
| #2 | ACCEPTED | 8 | 
| #3 | ACCEPTED | 12 | 
| #4 | WRONG ANSWER | 0 | 
| #5 | WRONG ANSWER | 0 | 
| #6 | WRONG ANSWER | 0 | 
| test | verdict | time | group | |
|---|---|---|---|---|
| #1 | ACCEPTED | 0.00 s | 1, 3, 4, 5, 6 | details | 
| #2 | ACCEPTED | 0.00 s | 1, 4, 5, 6 | details | 
| #3 | ACCEPTED | 0.00 s | 1, 4, 5, 6 | details | 
| #4 | ACCEPTED | 0.00 s | 1, 4, 5, 6 | details | 
| #5 | ACCEPTED | 0.01 s | 2, 5, 6 | details | 
| #6 | ACCEPTED | 0.01 s | 2, 5, 6 | details | 
| #7 | ACCEPTED | 0.00 s | 3, 5, 6 | details | 
| #8 | ACCEPTED | 0.00 s | 3, 5, 6 | details | 
| #9 | ACCEPTED | 0.00 s | 3, 5, 6 | details | 
| #10 | ACCEPTED | 0.00 s | 3, 5, 6 | details | 
| #11 | ACCEPTED | 0.00 s | 3, 5, 6 | details | 
| #12 | ACCEPTED | 0.00 s | 4, 5, 6 | details | 
| #13 | WRONG ANSWER | 0.00 s | 4, 5, 6 | details | 
| #14 | WRONG ANSWER | 0.00 s | 4, 5, 6 | details | 
| #15 | WRONG ANSWER | 0.00 s | 4, 5, 6 | details | 
| #16 | ACCEPTED | 0.01 s | 5, 6 | details | 
| #17 | WRONG ANSWER | 0.03 s | 5, 6 | details | 
| #18 | WRONG ANSWER | 0.02 s | 5, 6 | details | 
| #19 | ACCEPTED | 0.02 s | 5, 6 | details | 
| #20 | TIME LIMIT EXCEEDED | -- | 6 | details | 
| #21 | TIME LIMIT EXCEEDED | -- | 6 | details | 
| #22 | TIME LIMIT EXCEEDED | -- | 6 | details | 
Code
use std::{cmp, io};
use std::collections::{HashSet, VecDeque};
struct Stick {
    length: u32, // Original length. Shall never be modified.
    parts: Vec<u32> // Lengths of each part. Should always be in descending order.
}
fn main() {
    let stdin = io::stdin();
    let (n, m) = {
        let mut input: String = String::new();
        stdin.read_line(&mut input).unwrap();
        let mut iter = input.trim().split_whitespace().map(|x| x.parse::<u32>().unwrap());
        (iter.next().unwrap(), iter.next().unwrap())
    };
    let mut sticks: Vec<Stick> = {
        let mut input: String = String::new();
        stdin.read_line(&mut input).unwrap();
        input.trim().split_whitespace().map(|x| {
            let length = x.parse::<u32>().unwrap();
            Stick {
                length,
                parts: vec![length]
            }
        }).collect()
    };
    assert_eq!(sticks.len(), n as usize);
    sticks.sort_by(|a, b| b.length.cmp(&a.length));
    let mut min = (sticks.last().unwrap().length, sticks.len() - 1);
    let mut max_list: VecDeque<(u32, HashSet<usize>)> = { // VecDeque<(len, HashSet<indices>)>
        let mut max_list: VecDeque<(u32, HashSet<usize>)> = VecDeque::with_capacity(sticks.len());
        for i in 0..sticks.len() {
            let stick_max = sticks[i].parts[0];
            if !max_list.is_empty() && max_list[max_list.len() - 1].0 == stick_max {
                let last_index = max_list.len() - 1;
                max_list[last_index].1.insert(i);
            } else {
                max_list.push_back((stick_max, HashSet::from([i])));
            }
        }
        max_list
    };
    
    let mut answers = Vec::<u32>::with_capacity(m as usize);
    let mut k = 0;
    while k < m {
        let index: usize = {
            // Splitting a max length stick is usually a good idea.
            let max_value = max_list[0].0;
            let some_max_occurrence = *max_list[0].1.iter().next().unwrap();
            let (max_stick_split_min, max_stick_split_max) = get_extremes_after_next_cut(&sticks[some_max_occurrence]);
            if max_stick_split_min >= min.0 {
                some_max_occurrence
            } else {
                // If splitting the max stick would cause the min value to drop, check if something better could be done.
                let mut next_index = some_max_occurrence;
                let mut next_delta = {
                    let mut comparison_max = max_value;
                    if max_list[0].1.len() <= 1 && max_list.len() > 1 {
                        comparison_max = cmp::max(max_stick_split_max, max_list[1].0);
                    }
                    comparison_max - max_stick_split_min
                };
                for i in 0..sticks.len() {
                    if i == some_max_occurrence {
                        continue;
                    }
                    let (split_min, _) = get_extremes_after_next_cut(&sticks[i]);
                    let delta = max_value - cmp::min(min.0, split_min);
                    if delta < next_delta {
                        next_delta = delta;
                        next_index = i;
                    }
                    if delta == 0 {
                        break;
                    }
                }
                next_index
            }
        };
        let previous_part_count = sticks[index].parts.len() as u32;
        let denominator = previous_part_count + 1;
        let new_parts = divide_stick(sticks[index].length, denominator);
        assert!(new_parts[0] <= max_list[0].0);
        let old_parts = &sticks[index].parts;
        let required_cuts = new_parts.len() - old_parts.len();
        assert!(required_cuts <= 1);
        if required_cuts > 0 {
            // Change max if needed
            if new_parts[0] != old_parts[0] {
                // Add the maximum length of this stick to max_list after splitting.
                match max_list.binary_search_by_key(&-(new_parts[0] as i32), |pair| -(pair.0 as i32)) {
                    Ok(i) => {
                        max_list[i].1.insert(index);
                    }
                    Err(i) => {
                        max_list.insert(i, (new_parts[0], HashSet::from([index])));
                    }
                };
                // Remove the old maximum length of this stick from max_list prior to the split.
                match max_list.binary_search_by_key(&-(old_parts[0] as i32), |pair| -(pair.0 as i32)) {
                    Ok(i) => {
                        max_list[i].1.remove(&index);
                        if max_list[i].1.is_empty() {
                            max_list.remove(i);
                        }
                    }
                    Err(_) => {
                        panic!("Old maximum value {} of stick {} was not in max_list.", &old_parts[0], index);
                    }
                };
            }
            // Change min if needed
            if new_parts.last().unwrap() < &min.0 {
                min = (*new_parts.last().unwrap(), index);
            }
            sticks[index].parts = new_parts;
            let delta = max_list[0].0 - min.0;
            answers.push(delta);
            k += 1;
            {
                let mut calculated_max = *sticks.first().unwrap().parts.first().unwrap();
                let mut calculated_min = *sticks.last().unwrap().parts.last().unwrap();
                for stick in &sticks {
                    for part in &stick.parts {
                        if *part < calculated_min {
                            calculated_min = *part;
                        }
                        if *part > calculated_max {
                            calculated_max = *part;
                        }
                    }
                }
                assert_eq!(calculated_min, min.0);
                assert_eq!(calculated_max, max_list[0].0);
            }
        } else {
            assert_eq!(max_list[0].0, 1);
            assert_eq!(min.0, 1);
            break;
        }
    }
    let answers: Vec<String> = answers.iter().take(m as usize).map(|a| a.to_string()).collect();
    println!("{}", answers.join(" "))
}
/// Divides a stick of given length into `denominator` parts.
/// The returned Vec is in descending size order. max - min <= 1 inside the Vec.
fn divide_stick(length: u32, denominator: u32) -> Vec<u32> {
    let mut parts = Vec::with_capacity(denominator as usize);
    let div = length / denominator;
    if div > 0 {
        let mut extra = length % denominator;
        for _ in 0..denominator {
            if extra > 0 {
                extra -= 1;
                parts.push(div + 1);
            } else {
                parts.push(div);
                assert!(div > 0);
            }
        }
    } else {
        for _ in 0..length {
            parts.push(1);
        }
    }
    assert_eq!(parts.iter().sum::<u32>(), length);
    assert!(parts.iter().max().unwrap() - parts.iter().min().unwrap() <= 1);
    parts
}
/// (min, max)
fn get_extremes_after_next_cut(stick: &Stick) -> (u32, u32) {
    let denominator = stick.parts.len() as u32 + 1;
    let min = cmp::max(stick.length / denominator, 1);
    let max = min + if stick.length % denominator == 0 { 0 } else { 1 };
    (min, max)
}Test details
Test 1
Group: 1, 3, 4, 5, 6
Verdict: ACCEPTED
| input | 
|---|
| 1 1 6  | 
| correct output | 
|---|
| 0 | 
| 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: ACCEPTED
| input | 
|---|
| 100 200 145 136 74 83 73 36 196 115 11...  | 
| correct output | 
|---|
| 194 190 189 183 182 181 181 17... | 
| user output | 
|---|
| 194 190 189 183 182 181 181 17... | 
Test 13
Group: 4, 5, 6
Verdict: WRONG ANSWER
| 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 | 
|---|
| 95 96 96 95 93 94 94 94 90 91 ... | 
Test 14
Group: 4, 5, 6
Verdict: WRONG ANSWER
| 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 | 
|---|
| 98 98 98 96 90 91 88 88 84 86 ... | 
Test 15
Group: 4, 5, 6
Verdict: WRONG ANSWER
| 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 | 
|---|
| 99 99 98 98 97 97 96 96 95 95 ... | 
Test 16
Group: 5, 6
Verdict: ACCEPTED
| input | 
|---|
| 1000 2000 928772177 816188227 216592201 ...  | 
| correct output | 
|---|
| 991676844 990940224 990685481 ... | 
| user output | 
|---|
| 991676844 990940224 990685481 ... | 
Test 17
Group: 5, 6
Verdict: WRONG ANSWER
| input | 
|---|
| 1000 2000 665759876 597950008 615453266 ...  | 
| correct output | 
|---|
| 498801198 498681904 498504321 ... | 
| user output | 
|---|
| 498801198 498681904 498504321 ... | 
Test 18
Group: 5, 6
Verdict: WRONG ANSWER
| input | 
|---|
| 500 2000 683288817 784230412 626685186 ...  | 
| correct output | 
|---|
| 497667621 498434895 495465990 ... | 
| user output | 
|---|
| 497667621 498434895 495465990 ... | 
Test 19
Group: 5, 6
Verdict: ACCEPTED
| input | 
|---|
| 1000 2000 666667000 809309500 571572000 ...  | 
| correct output | 
|---|
| 499499500 499249250 498999000 ... | 
| user output | 
|---|
| 499499500 499249250 498999000 ... | 
Test 20
Group: 6
Verdict: TIME LIMIT EXCEEDED
| input | 
|---|
| 100000 200000 861772559 734298084 983382252 ...  | 
| correct output | 
|---|
| 499973914 499985299 499985141 ... | 
| user output | 
|---|
| (empty) | 
Test 21
Group: 6
Verdict: TIME LIMIT EXCEEDED
| input | 
|---|
| 30000 200000 691834579 617419813 514778075 ...  | 
| correct output | 
|---|
| 499967533 499976270 499969810 ... | 
| user output | 
|---|
| (empty) | 
Test 22
Group: 6
Verdict: TIME LIMIT EXCEEDED
| input | 
|---|
| 100000 200000 820255000 960780000 741965000 ...  | 
| correct output | 
|---|
| 499995000 499992500 499990000 ... | 
| user output | 
|---|
| (empty) | 
