結果

問題 No.1555 Constructed Balancing Sequence
ユーザー chineristACchineristAC
提出日時 2021-01-11 03:58:00
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 978 bytes
コンパイル時間 427 ms
コンパイル使用メモリ 86,660 KB
実行使用メモリ 101,328 KB
最終ジャッジ日時 2023-09-04 22:02:24
合計ジャッジ時間 7,332 ms
ジャッジサーバーID
（参考情報） 		judge15 / judge14
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 WA -
testcase_02 WA -
testcase_03 WA -
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 WA -
testcase_21 WA -
testcase_22 WA -
testcase_23 WA -
testcase_24 WA -
testcase_25 WA -
testcase_26 WA -
testcase_27 WA -
testcase_28 WA -
testcase_29 WA -
testcase_30 WA -
testcase_31 WA -
testcase_32 WA -
testcase_33 WA -
testcase_34 WA -
testcase_35 WA -
testcase_36 WA -
testcase_37 WA -
testcase_38 WA -
testcase_39 WA -
testcase_40 WA -
testcase_41 WA -
testcase_42 WA -
testcase_43 WA -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#maspyさんのコード(https://yukicoder.me/submissions/602281)をもとにつくったO(N)解

def check(N,K,A):
    import itertools
    MOD = 998_244_353
    def is_balanced(A, Acum):
        for a, b in zip(A[1:], Acum[1:]):
            if a > b:
                return False
        return True

    def main(N, K, A):
        S = [0] + list(itertools.accumulate(A))
        if not is_balanced(A, S):
            return 0
        MAX = K + 10
        # x 以下の数列は対策済
        dp = [0] * (MAX + MAX + 1)
        v = A[-1] - 1
        res = 1
        for n in range(N - 2, -1, -1):
            print(v)
            if v < S[n] + A[n] - 1:
                low = A[n]
            else:
                low = -K
            #print(A[n]+1-low)
            res *= (A[n]+1-low)
            res %= MOD
            v = max(A[n]-1,v//2)

        return res

    return main(N,K,A)

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