結果

問題 No.1555 Constructed Balancing Sequence
ユーザー chineristAC
提出日時 2021-01-11 03:58:00
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 978 bytes
コンパイル時間 167 ms
コンパイル使用メモリ 82,648 KB
実行使用メモリ 100,716 KB
最終ジャッジ日時 2024-06-22 19:29:30
合計ジャッジ時間 4,955 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample WA * 2
other WA * 42
権限があれば一括ダウンロードができます

ソースコード

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