結果
問題 |
No.45 回転寿司
|
ユーザー |
|
提出日時 | 2020-10-08 16:29:32 |
言語 | Rust (1.83.0 + proconio) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,615 bytes |
コンパイル時間 | 14,930 ms |
コンパイル使用メモリ | 396,456 KB |
実行使用メモリ | 6,944 KB |
最終ジャッジ日時 | 2024-07-20 05:57:09 |
合計ジャッジ時間 | 15,479 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 23 WA * 7 |
ソースコード
use std::io::{stdin, Read, StdinLock}; use std::str::FromStr; struct Input { n: usize, v: Vec<usize>, } fn read_input(cin_lock: &mut StdinLock) -> Input { let n = next_token(cin_lock); let v = (0..n).map(|_| next_token(cin_lock)).collect(); Input { n, v } } fn solve(input: Input) -> usize { let mut dp = vec![0; input.n]; for i in 0..input.n { dp[i] = input.v[i].max(dp[i]); for j in i + 2..input.n { dp[j] = (dp[i] + input.v[j]).max(dp[j]); } } return dp[input.n - 1]; } #[cfg(test)] mod tests { use super::*; #[test] fn test_solve() { assert_eq!( solve(Input { n: 4, v: vec![1, 2, 3, 4] }), 6 ); assert_eq!( solve(Input { n: 4, v: vec![5, 4, 4, 9] }), 14 ); assert_eq!( solve(Input { n: 7, v: vec![1, 2, 9, 10, 1, 1, 4] }), 16 ); assert_eq!(solve(Input { n: 1, v: vec![100] }), 100); } } fn next_token<T: FromStr>(cin_lock: &mut StdinLock) -> T { cin_lock .by_ref() .bytes() .map(|c| c.unwrap() as char) .skip_while(|c| c.is_whitespace()) .take_while(|c| !c.is_whitespace()) .collect::<String>() .parse::<T>() .ok() .unwrap() } fn main() { let cin = stdin(); let mut cin_lock = cin.lock(); let input = read_input(&mut cin_lock); println!("{}", solve(input)); }