結果
問題 | No.2956 Substitute with Average |
ユーザー |
![]() |
提出日時 | 2024-11-08 22:12:36 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,156 ms / 3,000 ms |
コード長 | 806 bytes |
コンパイル時間 | 512 ms |
コンパイル使用メモリ | 82,264 KB |
実行使用メモリ | 246,620 KB |
最終ジャッジ日時 | 2024-11-08 22:13:01 |
合計ジャッジ時間 | 18,783 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 25 |
ソースコード
from collections import defaultdictN=int(input())A=list(map(int,input().split()))def solve(N,A):ans=N*(N+1)//2for m in range(1,31):C=[0]+[a-m for a in A]for i in range(1,N+1):C[i]+=C[i-1]dct=defaultdict(int)for i in range(N+1):dct[C[i]]+=1if i<N and A[i]==m:ans-=dct[C[i]]dct=defaultdict(int)for i in range(N,-1,-1):dct[C[i]]+=1if i and A[i-1]==m:ans-=dct[C[i]]C=[0]+Afor i in range(1,N+1):C[i]+=C[i-1]dct=defaultdict(list)for i in range(N):dct[(A[i],C[i]-i*A[i])].append(i)for lst in dct.values():le=len(lst)ans+=le*(le+1)//2ans+=1return ansans=solve(N,A)print(ans)