結果

問題 No.2171 OR Assignment
コンテスト
ユーザー tute7627
提出日時 2022-11-21 23:12:21
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 2,168 ms / 3,500 ms
コード長 925 bytes
コンパイル時間 450 ms
コンパイル使用メモリ 82,220 KB
実行使用メモリ 117,240 KB
最終ジャッジ日時 2024-11-18 06:02:38
合計ジャッジ時間 28,103 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

import typing

MOD = 998244353

#https://ikatakos.com/pot/programming_algorithm/data_structure/binary_indexed_tree

class Bit:
    def __init__(self, n):
        self.size = n
        self.tree = [0] * (n + 1)
 
    def sum(self, i):
        s = 0
        while i > 0:
            s += self.tree[i]
            if s >= MOD:
                s -= MOD
            i -= i & -i
        return s
 
    def add(self, i, x):
        i += 1
        while i <= self.size:
            self.tree[i] += x
            if self.tree[i] >= MOD:
                self.tree[i] -= MOD
            i += i & -i

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

dp = Bit(N)
dp.add(0, 1)

prev = [-1] * 30

for i in range(N):
    for j in range(30):
        if (A[i] >> j) & 1:
            prev[j] = i 
    idx = sorted(list(set(prev)), reverse=True)
    for j in idx:
        if j != i:
            dp.add(j + 1, dp.sum(j + 1))

print(dp.sum(N))
0