結果

問題 No.2956 Substitute with Average
ユーザー vwxyz
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

from collections import defaultdict
N=int(input())
A=list(map(int,input().split()))
def solve(N,A):
ans=N*(N+1)//2
for 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]]+=1
if i<N and A[i]==m:
ans-=dct[C[i]]
dct=defaultdict(int)
for i in range(N,-1,-1):
dct[C[i]]+=1
if i and A[i-1]==m:
ans-=dct[C[i]]
C=[0]+A
for 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)//2
ans+=1
return ans
ans=solve(N,A)
print(ans)
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0