結果

問題 No.3389 k-Days Later
コンテスト
ユーザー urunea
提出日時 2025-12-02 01:45:43
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 421 ms / 2,000 ms
コード長 970 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 427 ms
コンパイル使用メモリ 82,908 KB
実行使用メモリ 97,128 KB
最終ジャッジ日時 2025-12-02 01:46:04
合計ジャッジ時間 17,230 ms
ジャッジサーバーID
(参考情報)
judge5 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 1
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code

from itertools import accumulate
import bisect

N=int(input())
D=list(map(int, input().split()))
acc_D = list(accumulate(D))
Q=int(input())
ans = []
for i in range(Q):
    y, m, d, k = (int(x) for x in input().split())
    if k <= D[m-1] - d:
        ans.append([y, m, d+k])
    else:
        k -= D[m-1]-d
        left = acc_D[N-1] - acc_D[m-1]
        m += 1
        if k <= left:
            k += acc_D[m-2]
            idx = bisect.bisect_left(acc_D, k)
            m = idx+1
            if 0 <= idx-1:
                k -= acc_D[idx-1]
            ans.append([y, m, k])
        else:
            y += 1
            k -= left
            y += k//acc_D[-1]
            k %= acc_D[-1]
            idx = bisect.bisect_left(acc_D, k)
            m = idx + 1
            if 0 <= idx-1:
                k -= acc_D[idx-1]
            if k != 0:
                ans.append([y, m, k])
            else:
                ans.append([y-1, N, D[-1]])

for i in ans:
    print(*i)
0