結果
問題 | No.1325 Subsequence Score |
ユーザー |
|
提出日時 | 2020-12-22 00:15:38 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 267 ms / 2,000 ms |
コード長 | 1,119 bytes |
コンパイル時間 | 161 ms |
コンパイル使用メモリ | 81,916 KB |
実行使用メモリ | 205,940 KB |
最終ジャッジ日時 | 2024-09-21 13:28:39 |
合計ジャッジ時間 | 5,147 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 28 |
ソースコード
import sysinput = lambda : sys.stdin.readline().rstrip()sys.setrecursionlimit(2*10**5+10)write = lambda x: sys.stdout.write(x+"\n")n = int(input())a = list(map(int, input().split()))M = 998244353### 素数の逆元とCombinationN = n+10 # 必要なテーブルサイズg1 = [0] * (N+1) # 元テーブルg2 = [0] * (N+1) #逆元テーブルinverse = [0] * (N+1) #逆元テーブル計算用テーブルg1[0] = g1[1] = g2[0] = g2[1] = 1inverse[0], inverse[1] = [0, 1]for i in range( 2, N + 1 ):g1[i] = ( g1[i-1] * i ) % Minverse[i] = ( -inverse[M % i] * (M//i) ) % M # ai+b==0 mod M <=> i==-b*a^(-1) <=> i^(-1)==-b^(-1)*aよりg2[i] = (g2[i-1] * inverse[i]) % Mdef cmb(n, r, M):if ( r<0 or r>n ):return 0r = min(r, n-r)return ((g1[n] * g2[r] % M) * g2[n-r]) % Mdef perm(n, r, M):if (r<0 or r>n):return 0return (g1[n] * g2[n-r]) % Mans = 0vs = [1]v = 1for i in range(N):v *= 2v %= Mvs.append(v)# ans = vs[n-1] * a[0]for i in range(1,n+1):ans += a[i-1] * ((i-1) * vs[i-2] + vs[i-1]) * vs[n-i]ans %= Mprint(ans)