結果

問題 No.2694 The Early Bird Catches The Worm
ユーザー MM
提出日時 2024-02-15 20:42:50
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 26 ms / 2,000 ms
コード長 1,190 bytes
コンパイル時間 12,195 ms
コンパイル使用メモリ 401,896 KB
実行使用メモリ 10,624 KB
最終ジャッジ日時 2024-09-28 22:30:09
合計ジャッジ時間 15,361 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 72
権限があれば一括ダウンロードができます
コンパイルメッセージ
warning: unused import: `std::str::FromStr`
 --> src/main.rs:2:5
  |
2 | use std::str::FromStr;
  |     ^^^^^^^^^^^^^^^^^
  |
  = note: `#[warn(unused_imports)]` on by default

ソースコード

diff # 

use std::io::{self, Read};
use std::str::FromStr;

fn main() {
    let mut input = String::new();
    io::stdin().read_to_string(&mut input).unwrap();

    let mut iter = input.split_whitespace();
    let n: usize = iter.next().unwrap().parse().unwrap();
    let h: i64 = iter.next().unwrap().parse().unwrap();

    let a: Vec<i64> = (0..n)
        .map(|_| iter.next().unwrap().parse().unwrap())
        .collect();
    let b: Vec<i64> = (0..n)
        .map(|_| iter.next().unwrap().parse().unwrap())
        .collect();

    let mut s: Vec<i64> = vec![0; n + 1];
    for i in 0..n {
        s[i + 1] = s[i] + b[i];
    }

    let mut ans = 0;
    let mut tmp = 0;
    let mut need_hp = 0;
    let mut r = 0;
    for l in 0..n {
        while r < n && need_hp + b[r] * (r as i64 - l as i64 + 1) <= h {
            tmp += a[r];
            need_hp += b[r] * (r as i64 - l as i64 + 1);
            r += 1;
        }

        // println!("{} {} {} {}", l, r, need_hp, tmp);
        if tmp > ans {
            ans = tmp;
        }

        if l == r {
            r += 1;
        } else {
            tmp -= a[l];
            need_hp -= s[r] - s[l];
        }
    }

    println!("{}", ans);
}
0