結果

問題 No.2698 Many Asakatsu
ユーザー 👑 rin204
提出日時 2024-03-22 22:32:01
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 868 ms / 2,000 ms
コード長 1,381 bytes
コンパイル時間 1,181 ms
コンパイル使用メモリ 82,604 KB
実行使用メモリ 111,064 KB
最終ジャッジ日時 2024-09-30 12:00:07
合計ジャッジ時間 14,409 ms
ジャッジサーバーID
(参考情報)
judge1 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 32
権限があれば一括ダウンロードができます

ソースコード

diff #

def solve(n, m, A, B, Q, x, C):
    def f(i, x):
        a, b = A[i], B[i]
        x -= a
        if b == 0:
            return abs(x) * m
        p = x // b
        if p < 0:
            return m * (b * (m - 1)) // 2 - x * m
        elif p >= m:
            return x * m - m * (b * (m - 1)) // 2
        else:
            ret = x * (p + 1) - (p + 1) * (b * p) // 2
            ret += (m - 1 - p) * (b * (p + 1) + b * (m - 1)) // 2 - x * (m - 1 - p)
            return ret

    P = [0]
    thr = []
    i = 1
    while i < n:
        bi = P[-1]
        l = 0
        r = 1 << 60

        while r - l > 1:
            mid = (l + r) // 2
            if f(bi, mid) > f(i, mid):
                r = mid
            else:
                l = mid

        if not thr or thr[-1] <= r:
            P.append(i)
            thr.append(r)
            i += 1
        else:
            P.pop()
            thr.pop()

    p = 0
    ans = []

    for _ in range(Q):
        while p < len(P) - 1 and x >= thr[p]:
            p += 1
        ans.append(P[p] + 1)
        x += C[P[p]]
    return ans


if __name__ == "__main__":
    n, m = map(int, input().split())
    A = [0] * n
    B = [0] * n
    for i in range(n):
        A[i], B[i] = map(int, input().split())

    Q, x = map(int, input().split())
    C = list(map(int, input().split()))

    res = solve(n, m, A, B, Q, x, C)
    print(*res)
0