Riippuliito

warning: unused variable: `lkm`
  --> input/code.rs:91:9
   |
91 |     let lkm = lkm_str.trim().parse::<usize>().unwrap();
   |         ^^^ help: consider prefixing with an underscore: `_lkm`
   |
   = note: #[warn(unused_variables)] on by default

#[derive(Clone, Copy, Debug)]
struct HuippuData(Option<usize>, Option<usize>);

// Jokaiselta huipulta voi liitaa kahteen suuntaan (ikaan kuin binaaripuu), ja se suunta, jossa on enemman vuoria, on parempi

/*struct Huippu {
    korkeus: usize,
    vasen: Option<Rc<Huippu>>,
    oikea: Option<Rc<Huippu>>,
}

fn korkein_mahd_huippu(kanta: &Huippu, max_korkeus: usize) -> Option<Rc<Huippu>> {
    let vasen_max = match &kanta.vasen {
        None => None,
        Some(vasen) => {
            if vasen.korkeus < max_korkeus {
                Some(kanta.vasen.as_ref().clone())
            } else {
                korkein_mahd_huippu(&vasen, max_korkeus)
            }
        }
    };

    let oikea_max = match &kanta.oikea {
        None => None,
        Some(oikea) => {
            if oikea.korkeus < max_korkeus {
                Some(kanta.oikea.as_ref().clone())
            } else {
                korkein_mahd_huippu(&oikea, max_korkeus)
            }
        }
    };

    match oikea_max {
        None => vasen_max,
        Some(oikea) => match vasen_max {
            None => Some(oikea),
            Some(vasen) => {
                if vasen.korkeus > oikea.korkeus {
                    return Some(vasen);
                }
                Some(oikea)
            },
        }
    }
}

fn lisaa_huippu_jalkeen(kanta: &mut Huippu, uusi: Huippu) {
    let nyt_kanta = kanta;
    while kanta.oikea.is_some() {
        nyt_kanta = &mut nyt_kanta.oikea.unwrap();
    }

    uusi.vasen = match kanta.oikea {
        Some(oikea) => korkein_mahd_huippu(&oikea, kanta.korkeus),
        None => None,
    };

    kanta.oikea = Some(Rc::new(uusi));
}*/

fn tee_hyppy(korkeudet: &[usize], pituus: usize) -> usize {
    if korkeudet.len() == 0 {
        return pituus;
    }

    let huippu_kork = korkeudet.iter().enumerate().fold((0, 0), |acc, x| {
        if *x.1 > acc.1 {
            return (x.0, *x.1);
        }
        return acc;
    }).1;

    let mut korkein_hyppy = 0;
    let mahd_hypyt = korkeudet.rsplit(|&x| x == huippu_kork).collect::<Vec<_>>();
    if mahd_hypyt.len() < 2 {
        return pituus;
    }

    for mahd_vuoret in mahd_hypyt {
        korkein_hyppy = std::cmp::max(korkein_hyppy, tee_hyppy(mahd_vuoret, pituus + 1));
    }

    return korkein_hyppy;
}

fn main() {
    let mut lkm_str = String::new();
    std::io::stdin().read_line(&mut lkm_str).unwrap();
    let lkm = lkm_str.trim().parse::<usize>().unwrap();

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

    let korkeudet = korkeudet_str.split_whitespace().map(|x| x.trim().parse::<usize>().unwrap()).collect::<Vec<usize>>();
    println!("{}", tee_hyppy(&korkeudet, 0));
}

Test 1

Group: 1, 2, 3

Verdict: ACCEPTED

Test 2

Group: 1, 2, 3

Verdict: ACCEPTED

Test 3

Group: 1, 2, 3

Verdict: ACCEPTED

Test 4

Group: 1, 2, 3

Verdict: ACCEPTED

Test 5

Group: 1, 2, 3

Verdict: ACCEPTED

Test 6

Group: 1, 2, 3

Verdict: ACCEPTED

Test 7

Group: 1, 2, 3

Verdict: ACCEPTED

Test 8

Group: 2, 3

Verdict: ACCEPTED

Test 9

Group: 2, 3

Verdict: ACCEPTED

Test 10

Group: 2, 3

Verdict: ACCEPTED

Test 11

Group: 2, 3

Verdict: ACCEPTED

Test 12

Group: 2, 3

Verdict: ACCEPTED

Test 13

Group: 2, 3

Verdict: ACCEPTED

Test 14

Group: 2, 3

Verdict: ACCEPTED

Test 15

Group: 3

Verdict: ACCEPTED

Test 16

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 17

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 18

Group: 3

Verdict: ACCEPTED

Test 19

Group: 3

Verdict: ACCEPTED

Test 20

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 21

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 22

Group: 1, 2, 3

Verdict: ACCEPTED

Test 23

Group: 1, 2, 3

Verdict: ACCEPTED

Test 24

Group: 1, 2, 3

Verdict: ACCEPTED

Test 25

Group: 1, 2, 3

Verdict: ACCEPTED

Test 26

Group: 1, 2, 3

Verdict: ACCEPTED

Test 27

Group: 1, 2, 3

Verdict: ACCEPTED

Test 28

Group: 1, 2, 3

Verdict: ACCEPTED

Test 29

Group: 1, 2, 3

Verdict: ACCEPTED