結果

問題 No.1555 Constructed Balancing Sequence
ユーザー chineristAC
提出日時 2021-01-12 04:02:18
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 78 ms / 2,000 ms
コード長 963 bytes
コンパイル時間 422 ms
コンパイル使用メモリ 82,144 KB
実行使用メモリ 97,776 KB
最終ジャッジ日時 2024-06-22 19:30:08
合計ジャッジ時間 4,252 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 42
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

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

        res = 1

        for n in range(N-2,-1,-1):
            tmp = A[n] - 1 + S[n]
            flag = False
            for j in range(n+1,N):
                if tmp > K:
                    break
                if A[j]-1 >= tmp:
                    flag = True
                    break
                tmp *= 2
            if flag:
                res *= (A[n]+1+K)
                res %= MOD

        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