結果
問題 | No.1239 Multiplication -2 |
ユーザー |
![]() |
提出日時 | 2020-09-25 23:13:15 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
TLE
(最新)
AC
(最初)
|
実行時間 | - |
コード長 | 1,823 bytes |
コンパイル時間 | 87 ms |
コンパイル使用メモリ | 12,800 KB |
実行使用メモリ | 21,716 KB |
最終ジャッジ日時 | 2024-06-28 07:21:14 |
合計ジャッジ時間 | 11,982 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 31 TLE * 3 |
ソースコード
import sys input = sys.stdin.readline N=int(input()) A=list(map(int,input().split())) mod=998244353 PLUS=[] MINUS=[] ANS=0 for i in range(N): if A[i]==2 or A[i]==-2: if i==0: LEFTP=1 LEFTM=0 else: LEFTP=pow(2,mod-2,mod) LEFTM=0 LEFT=1 for j in range(i-1,-1,-1): LEFT*=A[j] if j!=0: if LEFT==1: LEFTP+=pow(pow(2,mod-2,mod),i-j+1,mod) elif LEFT==-1: LEFTM+=pow(pow(2,mod-2,mod),i-j+1,mod) else: break else: if LEFT==1: LEFTP+=pow(pow(2,mod-2,mod),i-j,mod) elif LEFT==-1: LEFTM+=pow(pow(2,mod-2,mod),i-j,mod) else: break if i==N-1: RIGHTP=1 RIGHTM=0 else: RIGHTP=pow(2,mod-2,mod) RIGHTM=0 RIGHT=1 for j in range(i+1,N): RIGHT*=A[j] if j!=N-1: if RIGHT==1: RIGHTP+=pow(pow(2,mod-2,mod),j-i+1,mod) elif RIGHT==-1: RIGHTM+=pow(pow(2,mod-2,mod),j-i+1,mod) else: break else: if RIGHT==1: RIGHTP+=pow(pow(2,mod-2,mod),j-i,mod) elif RIGHT==-1: RIGHTM+=pow(pow(2,mod-2,mod),j-i,mod) else: break if A[i]==2: ANS+=LEFTP*RIGHTM+LEFTM*RIGHTP else: ANS+=LEFTP*RIGHTP+LEFTM*RIGHTM ANS%=mod print(ANS)