src/bin/diving-board-lcci.rs

#![allow(dead_code, unused, unused_variables)]

fn main() {}

struct Solution;

impl Solution {
    pub fn diving_board(shorter: i32, longer: i32, k: i32) -> Vec<i32> {
        let mut v = Vec::<i32>::new();
        if k == 0 {
            return v;
        }

        if shorter == longer {
            v.push(shorter * k);
            return v;
        }

        for i in 0..=k {
            let s = i * longer + (k - i) * shorter;

            v.push(s);
        }

        v
    }
}