結果

問題 No.1211 円環はお断り
ユーザー gew1fw
提出日時 2025-06-12 21:48:03
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,356 bytes
コンパイル時間 331 ms
コンパイル使用メモリ 82,076 KB
実行使用メモリ 95,340 KB
最終ジャッジ日時 2025-06-12 21:54:26
合計ジャッジ時間 5,853 ms
ジャッジサーバーID
(参考情報)
judge1 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample -- * 2
other AC * 13 TLE * 1 -- * 27
権限があれば一括ダウンロードができます

ソースコード

diff #

def main():
    import sys
    input = sys.stdin.read().split()
    idx = 0
    N = int(input[idx])
    idx += 1
    K = int(input[idx])
    idx += 1
    A = list(map(int, input[idx:idx+N]))
    
    if K == 0:
        print(0)
        return
    
    total = sum(A)
    if K == 1:
        print(total)
        return
    
    low = 0
    high = total
    best = 0
    
    def is_possible(mid):
        if mid * K > total:
            return False
        B = A + A
        prefix = [0] * (2 * N + 1)
        for i in range(2 * N):
            prefix[i + 1] = prefix[i] + B[i]
        
        for i in range(N):
            end = i + N
            total_sub = prefix[end] - prefix[i]
            if total_sub < mid * K:
                continue
            
            cnt = 0
            current_sum = 0
            for j in range(i, i + N):
                current_sum += B[j]
                if current_sum >= mid:
                    cnt += 1
                    current_sum = 0
                    if cnt >= K:
                        break
            if cnt >= K:
                return True
        return False
    
    while low <= high:
        mid = (low + high) // 2
        if is_possible(mid):
            best = mid
            low = mid + 1
        else:
            high = mid - 1
    print(best)

if __name__ == "__main__":
    main()
0