結果

問題 No.973 余興
ユーザー shibh308
提出日時 2019-12-06 07:59:42
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
TLE  
実行時間 -
コード長 1,474 bytes
コンパイル時間 291 ms
コンパイル使用メモリ 12,928 KB
実行使用メモリ 277,592 KB
最終ジャッジ日時 2024-12-17 21:11:28
合計ジャッジ時間 257,024 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 5 TLE * 49
権限があれば一括ダウンロードができます

ソースコード

diff # 

import numpy as np

n, x = tuple(map(int, input().split()))
a = list(map(int, input().split()))


left = np.zeros(n, dtype=np.int32)
right = np.zeros(n, dtype=np.int32)
l = 0
r = 0
s = 0

while True:
    while r != n and  s + a[r] <= x:
        s += a[r]
        r += 1
    left[l] = r - l
    s -= a[l]
    l += 1
    if l == n:
        break

l = n - 1
r = n - 1
s = 0

while True:
    while l != -1 and s + a[l] <= x:
        s += a[l]
        l -= 1
    right[r] = r - l
    s -= a[r]
    r -= 1
    if r < 0:
        break

dp_l = np.zeros((n, n + 1), dtype=np.int32)
dp_r = np.zeros((n, n + 1), dtype=np.int32)


i = 0
j = 0
while i <= n:
    while j < n:
        if i == 0:
            dp_l[j,i] = 0
            dp_r[j,i] = 0
            j += 1
            continue
        l = j
        r = j + i - 1
        if r >= n:
            break
        cnt = min(left[l], i - 1)
        x = dp_r[r,i - 1] - dp_r[r,i - cnt - 1]
        if x != cnt:
            dp_l[l,i] = dp_l[l,i - 1] + 1
            dp_r[r,i] = dp_r[r,i - 1] + 1
            j += 1
            continue

        cnt = min(right[r], i - 1)
        x = dp_l[l,i - 1] - dp_l[l,i - cnt - 1]
        if x != cnt:
            dp_l[l,i] = dp_l[l,i - 1] + 1
            dp_r[r,i] = dp_r[r,i - 1] + 1
            j += 1
            continue
        dp_l[l,i] = dp_l[l,i - 1]
        dp_r[r,i] = dp_r[r,i - 1]
        j += 1
    i += 1
    j = 0

if dp_l[0,n] == dp_l[0,n - 1]:
    print("B")
else:
    print("A")
0