結果

問題 No.3257 +|+
コンテスト
ユーザー yu23578
提出日時 2025-09-05 22:11:03
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 695 bytes
コンパイル時間 1,779 ms
コンパイル使用メモリ 196,028 KB
実行使用メモリ 9,504 KB
最終ジャッジ日時 2025-09-05 22:11:35
合計ジャッジ時間 10,176 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 3 WA * 1 RE * 29
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#define int long long

int N;

signed main(){
  cin>>N;
  vector<int> A(N+1);
  for(int i = 1; i <= N; i++) cin>>A[i];
  int B = 447;
  int ans = 0;
  for(int i = 1; i <= 3*B; i++){
  	for(int j = 1; j <= 3*B; j++){
  		if(i >= N || j >= N) continue;
  		if((A[i] + A[j]) % (i + j) == 0){
  			ans++;
  			if((A[i] + A[j]) / (i + j) <= B){
  				ans--;
  			}
  		}
  	}
  }
  vector<int> cnt(600000);
  int ho = 300000;
  for(int i = 1; i <= B; ++i){
  	for(int j = 1; j <= N; ++j){
  		ans += cnt[j * i - A[j] + ho];
  		cnt[A[j] - j * i + ho]++;
  	}
  	for(int j = 1; j <= N; ++j){
  		cnt[A[j] - j * i + ho]--;
  	}
  }
  cout << ans << "\n";
}
0