結果

問題 No.2694 The Early Bird Catches The Worm
ユーザー VvyLw
提出日時 2024-03-23 17:04:44
言語 D
(dmd 2.109.1)
結果
AC  
実行時間 82 ms / 2,000 ms
コード長 1,478 bytes
コンパイル時間 1,909 ms
コンパイル使用メモリ 149,632 KB
実行使用メモリ 16,812 KB
最終ジャッジ日時 2024-09-30 13:32:10
合計ジャッジ時間 7,503 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 72
権限があれば一括ダウンロードができます

ソースコード

diff # 

import std.stdio: readf, readln, writeln;
import std.conv: to;
import std.string: split;
struct PrefixSum(T) {
import std.algorithm.iteration: cumulativeFold;
import std.array: array, insertInPlace;
private:
    int n;
    bool not_built;
    T[] s;
public:
    this(const int n) {
        this.n = n;
        not_built = true;
        s = new T[n + 1];
    }
    this(const T[] a) {
        s = cumulativeFold!"a + b"(a, cast(T)0).array;
        insertInPlace(s, 0, 0);
    }
    T[] get(){ return s; }
    T sum(const int l, const int r) const { return s[r] - s[l]; }
    void add(const int l, const int r, const T x) {
        assert(not_built);
        s[l] += x;
        s[r] -= x;
    }
    T[] build() {
        assert(not_built);
        auto res = cumulativeFold!"a + b"(s).array;
        not_built = false;
        return res[0..n];
    }
}
void chmax(T, U)(ref T a, const U b) {
    if(a < b) {
		a = b;
	}
}
void main() {
    int n, h;
    readf("%d %d\n", &n, &h);
    const a = readln.split.to!(int[]);
    const b = readln.split.to!(long[]);
    auto ps = new PrefixSum!long(b);
    long ans = 0, s = 0, t = 0;
    for(int l = 0, r = 0; l < n; ++l) {
        while(r < n && t + (r - l + 1) * b[r] <= h) {
            s += a[r];
            t += (r - l + 1) * b[r];
            r++;
        }
        if(l == r) {
            r++;
        } else {
            chmax(ans, s);
            s -= a[l];
            t -= ps.sum(l, r);
        }
    }
    writeln(ans);
}
0