結果

問題 No.1555 Constructed Balancing Sequence
ユーザー chineristACchineristAC
提出日時 2021-01-06 07:32:12
言語 PyPy3
(7.3.15)
結果
RE  
(最新)
AC  
(最初)
実行時間 -
コード長 1,552 bytes
コンパイル時間 627 ms
コンパイル使用メモリ 86,952 KB
実行使用メモリ 131,444 KB
最終ジャッジ日時 2023-09-04 22:00:54
合計ジャッジ時間 9,234 ms
ジャッジサーバーID
（参考情報） 		judge12 / judge13
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 75 ms
71,220 KB
testcase_01 AC 125 ms
77,404 KB
testcase_02 AC 77 ms
71,292 KB
testcase_03 AC 79 ms
71,116 KB
testcase_04 AC 77 ms
71,140 KB
testcase_05 AC 76 ms
71,272 KB
testcase_06 AC 75 ms
71,220 KB
testcase_07 AC 76 ms
71,252 KB
testcase_08 AC 76 ms
71,272 KB
testcase_09 AC 85 ms
76,120 KB
testcase_10 AC 77 ms
71,288 KB
testcase_11 AC 88 ms
75,948 KB
testcase_12 RE -
testcase_13 RE -
testcase_14 RE -
testcase_15 RE -
testcase_16 RE -
testcase_17 AC 683 ms
131,444 KB
testcase_18 RE -
testcase_19 RE -
testcase_20 RE -
testcase_21 RE -
testcase_22 RE -
testcase_23 RE -
testcase_24 RE -
testcase_25 RE -
testcase_26 RE -
testcase_27 RE -
testcase_28 RE -
testcase_29 RE -
testcase_30 RE -
testcase_31 RE -
testcase_32 RE -
testcase_33 RE -
testcase_34 RE -
testcase_35 RE -
testcase_36 RE -
testcase_37 RE -
testcase_38 AC 78 ms
71,092 KB
testcase_39 AC 76 ms
71,136 KB
testcase_40 AC 77 ms
71,344 KB
testcase_41 AC 91 ms
76,168 KB
testcase_42 RE -
testcase_43 RE -
権限があれば一括ダウンロードができます

ソースコード

diff # 

def perferct_solve(N,K,A):
    mod = 998244353

    diff = [A[i] for i in range(N)]
    cum = [A[i] for i in range(N)]
    S = A[0]
    for i in range(1,N):
        diff[i] = S - A[i]
        if diff[i] < 0:
            return 0
        S += A[i]
        cum[i] = S

    dp = [[0 for minus in range(3*K+1)] for i in range(N)]

    for minus in range(3*K+1):
        first = A[0] - minus
        if -K<=first<=K:
            dp[0][minus] = 1
        if minus:
            dp[0][minus] += dp[0][minus-1]
            dp[0][minus] %= mod

    for i in range(1,N):
        for minus in range(3*K+1):

            if diff[i] and -K<=A[i]-minus<=K:
                dp[i][minus] += dp[i-1][0]
                dp[i][minus] %= mod

            if minus%2==diff[i] and -K<=(cum[i]-minus)//2<=K:
                dp[i][minus] += dp[i-1][3*K] - dp[i-1][minus//2+diff[i]]
                dp[i][minus] %= mod

            if diff[i]<=1:
                pre_minus_L = max(diff[i],-K+minus-A[i])
                pre_minus_R = min((minus+diff[i])//2,K+minus-A[i])
                if pre_minus_L<=pre_minus_R:
                    dp[i][minus] += dp[i-1][pre_minus_R] - dp[i-1][pre_minus_L-1] * (pre_minus_L>0)
                    dp[i][minus] %= mod

            if minus:
                dp[i][minus] += dp[i][minus-1]
                dp[i][minus] %= mod

    return dp[N-1][0]

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

assert 2<=N<=1500
assert 1<=K<=1500
assert len(A)==N
for i in range(N):
    assert abs(A[i])<=K

print(perferct_solve(N,K,A))
0