結果

問題 No.78 クジ付きアイスバー
ユーザー titia
提出日時 2022-01-26 00:41:40
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
WA  
実行時間 -
コード長 678 bytes
コンパイル時間 261 ms
コンパイル使用メモリ 12,544 KB
実行使用メモリ 10,880 KB
最終ジャッジ日時 2024-12-17 22:12:09
合計ジャッジ時間 2,464 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 25 WA * 10
権限があれば一括ダウンロードができます

ソースコード

diff # 

N,K=map(int,input().split())
S=list(map(int,list(input().strip())))

SUM=[0]
for s in S:
    SUM.append(SUM[-1]+s)

if N<=SUM[-1]:
    OK=N
else:
    OK=K

def calc(x):
    ANS=(x//N)*SUM[-1]
    if x%N>0:
        ANS+=SUM[(x%N)]
    return ANS

NG=0
while OK>NG+1:
    #print(OK,NG)
    mid=(OK+NG)//2

    NOW=0
    PLUS=mid

    while True:
        #print(NOW,PLUS)
        k=calc(PLUS)-calc(NOW)

        NOW=PLUS
        PLUS=PLUS+k

        if PLUS>=K or PLUS>=mid+2*N:
            break
        if PLUS==NOW:
            break

    #print(mid,NOW,PLUS)

    if PLUS>=K or PLUS>=mid+2*N:
        OK=mid
    else:
        NG=mid

print(OK)
0