AtCoder Beginner Contest 270 D

https://atcoder.jp/contests/abc270/tasks/abc270_d

残りの石の数に対して、どちらも最善を尽くしたとき、先番がいくつ取れるかをDPにします。
ふつうにtake_whileが使えますね。

// Stones
#![allow(non_snake_case)]


//////////////////// library ////////////////////

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()
}


//////////////////// process ////////////////////

fn read_input() -> (usize, Vec<usize>) {
    let v = read_vec();
    let N = v[0];
    let A = read_vec();
    (N, A)
}

fn f(N: usize, A: Vec<usize>) -> usize {
    let mut dp: Vec<usize> = (0..(N+1)).map(|_| 0).collect();
    for n in 1..(N+1) {
        dp[n] = A.iter().take_while(|&a| a <= &n).
                            map(|&a| n - dp[n-a]).max().unwrap();
    }
    dp[N]
}

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