アルゴリズムと数学 031

https://atcoder.jp/contests/math-and-algorithm/tasks/math_and_algorithm_ac

これもDPですが、一つ前とその前で最大を覚えておくとよいです。
手元では、

(a, b) = (max(a, b), a + c)

が通ったのですが、AtCoderではダメでした。

// Taro's Vacation
#![allow(non_snake_case)]

use std::cmp::max;

fn read<T: std::str::FromStr>() -> T {
    let mut line = String::new();
    std::io::stdin().read_line(&mut line).ok();
    line.trim().parse().ok().unwrap()
}

fn read_vec<T: std::str::FromStr>() -> Vec<T> {
    read::<String>().split_whitespace()
            .map(|e| e.parse().ok().unwrap()).collect()
}

fn read_input() -> Vec<i64> {
    let _N: usize = read();
    let A = read_vec();
    A
}

fn f(A: Vec<i64>) -> i64 {
    let (mut a, mut b) = (A[0], A[1]);
    for c in A[2..].iter() {
        let tmp = a;
        a = max(a, b);
        b = tmp + c
//      (a, b) = (max(a, b), a + c)
    }
    max(a, b)
}

fn main() {
    let A = read_input();
    println!("{}", f(A))
}