結果

問題 No.45 回転寿司
ユーザー iwot
提出日時 2019-12-17 14:49:13
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 2 ms / 5,000 ms
コード長 1,366 bytes
コンパイル時間 14,606 ms
コンパイル使用メモリ 383,380 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-07-02 21:26:36
合計ジャッジ時間 16,275 ms
ジャッジサーバーID
(参考情報)
judge1 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff #

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

fn read_vec<T: std::str::FromStr>() -> Vec<T> {
    let mut s = String::new();
    std::io::stdin().read_line(&mut s).ok();
    s.trim()
        .split_whitespace()
        .map(|e| e.parse().ok().unwrap())
        .collect()
}

fn main() {
  let n: usize = read();
  let ds: Vec<u32> = read_vec();
  println!("{}", no45(n, ds));
}

fn no45(max: usize, ds: Vec<u32>) -> u32 {
  
  fn get_next(start_index: usize, current_index: usize, ds: &[u32]) -> Option<(usize, u32)> {
    let next_index = if start_index == current_index {
      current_index + 2
    } else {
      current_index + 1
    };

    if next_index >= ds.len() {
      None
    } else {
      Some((next_index, ds[next_index]))
    }
  }

  let mut start_index = 0;
  let mut current_index = 0;
  let mut values = ds.clone();
  loop {
    if let Some((next_index, value)) = get_next(start_index, current_index, &ds) {
      if values[start_index] + value > values[next_index] {
        values[next_index] = values[start_index] + value;
      }
      current_index = next_index;
    } else {
      if start_index + 1 >= max {
        break;
      }

      start_index += 1;
      current_index = start_index;
    }
  }

  *values.iter().max().unwrap()
}
0