結果

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

ソースコード

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 = i + 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){
  	  if(j * i - A[j] + ho < 600000){
  	    ans += cnt[j * i - A[j] + ho];
  	  }
  	  if(A[j] - j * i + ho >= 0){
  	    cnt[A[j] - j * i + ho]++;
  	  }
  	}
  	for(int j = 1; j <= N; ++j){
  	  if(A[j] - j * i + ho >= 0){
  	    cnt[A[j] - j * i + ho]--;
  	  }
  	}
  }
  cout << ans << "\n";
}
0