Riippuliito

#[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(huippudata: &mut [HuippuData], korkeudet: &[usize], osoitin: 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(huippudata, mahd_vuoret, osoitin, 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();
    /*for kork in korkeudet_str.split_whitespace() {
        let korkeus = kork.trim().parse::<usize>().unwrap();
        
        // uusi korkein huippu
        if korkeus > nyk_huippu.korkeus {
            let o_nyk_huippu = nyk_huippu;
            nyk_huippu = Huippu{
                korkeus,
                vasen: Some(Rc::new(o_nyk_huippu)),
                oikea: None,
            };
        } else {
            lisaa_huippu_jalkeen(&mut nyk_huippu, Huippu{korkeus, vasen: None, oikea: None});
        }
    }*/
    let mut huippudata = std::iter::repeat(HuippuData(None, None)).take(lkm).collect::<Vec<HuippuData>>();
    let korkeudet = korkeudet_str.split_whitespace().map(|x| x.trim().parse::<usize>().unwrap()).collect::<Vec<usize>>();
    //keraa_huippudata(&mut huippudata, &korkeudet, 0);
    println!("{}", tee_hyppy(&mut huippudata, &korkeudet, 0, 0));
 
    /*
     * // Lista "kantamasta", jonka verran huipulta voi lentää
    let mut huipulta_paasee_vas = Vec::with_capacity(lkm);
    let mut huipulta_paasee_oik = Vec::with_capacity(lkm);
    for i in 0..lkm {
        let mut max_oikealle_huiput = 1;
        for j in (i+1)..lkm {
            if korkeudet[j] >= korkeudet[i] {
                break;
            }
            max_oikealle_huiput += 1;
        }
 
        let mut max_vasemmalle_huiput = 1;
 
        if i != 0 {
            for j in (0..=(i-1)).rev() {
                if korkeudet[j] >= korkeudet[i] {
                    break;
                }
                max_vasemmalle_huiput += 1;
            }
        }
 
        huipulta_paasee_vas.push(max_vasemmalle_huiput);
        huipulta_paasee_oik.push(max_oikealle_huiput);
    }*/
 
}

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