結果
問題 | No.2941 Sigma Music Game Score Problem |
ユーザー |
|
提出日時 | 2024-10-18 21:47:31 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 444 ms / 2,500 ms |
コード長 | 666 bytes |
コンパイル時間 | 407 ms |
コンパイル使用メモリ | 82,600 KB |
実行使用メモリ | 261,628 KB |
最終ジャッジ日時 | 2024-10-18 22:37:35 |
合計ジャッジ時間 | 7,153 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 29 |
ソースコード
import sys input = lambda :sys.stdin.readline()[:-1] ni = lambda :int(input()) na = lambda :list(map(int,input().split())) yes = lambda :print("yes");Yes = lambda :print("Yes");YES = lambda : print("YES") no = lambda :print("no");No = lambda :print("No");NO = lambda : print("NO") ####################################################################### m, n = na() mod = 998244353 x = [0] + na() + [m + 1] s = 0 ans = 0 def f(s, h):# s^2 + (s + 1) ^ 2 + ... + (s + h - 1) ^ 2 return g(s + h - 1) - g(s - 1) def g(x): return x * (x + 1) * (2 * x + 1) // 6 % mod for i in range(n + 1): h = x[i + 1] - x[i] - 1 ans += g(h) ans %= mod print(ans)