結果

問題 No.1239 Multiplication -2
ユーザー chineristAC
提出日時 2020-10-01 16:21:08
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,434 bytes
コンパイル時間 139 ms
コンパイル使用メモリ 82,040 KB
実行使用メモリ 104,492 KB
最終ジャッジ日時 2024-07-07 01:21:25
合計ジャッジ時間 6,215 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1 WA * 2
other AC * 4 WA * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

N = int(input())
A = list(map(int,input().split()))
mod = 998244353
inv = pow(2,mod-2,mod)

res = 0

for i in range(N):
    if abs(A[i])==2:
        base = pow(2,max(i-1,0),mod)
        left_pos = base
        left_nega = 0
        prod = 1
        if i!=1:
            base = (base * inv) % mod
        for j in range(1,i+1):
            if abs(A[i-j])==1:
                prod *= A[i-j]
                if prod>0:
                    left_pos += base
                else:
                    left_nega += base
            else:
                break
            if j!=i-1:
                base = (base * inv) % mod

        base = pow(2,max(0,N-i-2),mod)
        right_pos = base
        right_nega = 0
        prod = 1
        if i!=N-2:
            base = (base * inv) % mod
        for j in range(1,N-i):
            print(base)
            if abs(A[i+j])==1:
                prod *= A[i+j]
                if prod>0:
                    right_pos += base
                else:
                    right_nega += base
            else:
                break
            if j!=N-i-2:
                base = (base * inv) % mod

        if A[i]==2:
            res += left_pos * right_nega + left_nega * right_pos
        else:
            res += left_pos * right_pos + left_nega * right_nega
        res %= mod
        #print(left_pos,left_nega,right_pos,right_nega)

#print(res)
ans = res * pow(2,(N-1)*(mod-2),mod)
print(ans%mod)
0