結果
問題 | No.2956 Substitute with Average |
ユーザー |
![]() |
提出日時 | 2024-11-08 23:28:22 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 698 ms / 3,000 ms |
コード長 | 414 bytes |
コンパイル時間 | 412 ms |
コンパイル使用メモリ | 82,560 KB |
実行使用メモリ | 306,684 KB |
最終ジャッジ日時 | 2024-11-08 23:28:35 |
合計ジャッジ時間 | 12,148 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 25 |
ソースコード
n = int(input())a = list(map(int,input().split()))m = max(a); s = n*(n+1)//2+1for i in range(1,m+1):b = [0]; d = {}for v in a: b.append(b[-1]+v-i)for i,v in enumerate(b):if v in d:if d[v][2]==i-1: d[v][1] += 1; d[v][2] = ielse: d[v][0] += 1; d[v][1] += 1; d[v][2] = ielse: d[v] = [1,1,i]for p,q,_ in d.values(): s -= q*(q-1)//2-p*(p-1)//2print(s)