結果

問題 No.1008 Bench Craftsman
ユーザー maspymaspy
提出日時 2020-03-06 22:56:54
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 801 ms / 2,000 ms
コード長 1,321 bytes
コンパイル時間 141 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 55,048 KB
最終ジャッジ日時 2024-10-14 09:23:35
合計ジャッジ時間 23,582 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 27
権限があれば一括ダウンロードができます

ソースコード

diff # 

#!/usr/bin/env python3
# %%
import sys
read = sys.stdin.buffer.read
readline = sys.stdin.buffer.readline
readlines = sys.stdin.buffer.readlines
import numpy as np

# %%
N, M = map(int, readline().split())
A = np.array(readline().split(), np.int64)
XW = np.array(read().split(), np.int64)
X = XW[::2] - 1
W = XW[1::2]


# %%
def calc_cost(N, c, X, W):
    B1_d = np.zeros(N + 1, np.int64)
    B0_d = np.zeros(N + 1, np.int64)
    L = X - (W + c - 1) // c + 1
    L = np.maximum(0, L)
    const_term = W - c * X
    np.add.at(B1_d, L, c)
    np.add.at(B0_d, L, const_term)
    np.add.at(B1_d, X+1, -c)
    np.add.at(B0_d, X + 1, -const_term)

    R = X + (W + c - 1) // c - 1
    R = np.minimum(R, N - 1)
    const_term = W + c * X
    np.add.at(B1_d, X + 1, -c)
    np.add.at(B0_d, X + 1, const_term)
    np.add.at(B1_d, R + 1, c)
    np.add.at(B0_d, R + 1, -const_term)
    B0 = np.cumsum(B0_d)[:-1]
    B1 = np.cumsum(B1_d)[:-1]
    return B0 + B1 * np.arange(N, dtype=np.int64)


def test(c):
    if c == 0:
        return W.sum() <= A.min()
    cost = calc_cost(N, c, X, W)
    return np.all(cost < A)


# %%
left = -1
right = 10 ** 5 + 10
while left + 1 < right:
    mid = (left + right) // 2
    if test(mid):
        right = mid
    else:
        left = mid

if right > 10 ** 5:
    print(-1)
else:
    print(right)
0