結果

問題 No.3257 +|+
コンテスト
ユーザー nikoro256
提出日時 2025-09-05 23:13:07
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 474 bytes
コンパイル時間 213 ms
コンパイル使用メモリ 82,744 KB
実行使用メモリ 330,260 KB
最終ジャッジ日時 2025-09-05 23:13:13
合計ジャッジ時間 5,590 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 4 TLE * 1 -- * 28
権限があれば一括ダウンロードができます

ソースコード

diff # 

from math import isqrt
from collections import defaultdict
N=int(input())
A=list(map(int,input().split()))
n=isqrt(4*(10**5))
ans=0
d = defaultdict(int)
for ni in range(1,n+1):
    for j in range(N):
        ans+=d[-(A[j]-(j+1)*ni)]
        d[(A[j]-(j+1)*ni)]+=1
    d.clear()

for ij in range(1,n+1):
    for j in range(N):
        i = ij-j-2
        if 0<=i<j<N:
            ni = (A[i]+A[j])//ij
            if (A[i]+A[j])%ij==0 and ni>n:
                ans+=1
print(ans)
0