結果

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

ソースコード

diff # 

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

int N;

int solve1(vector<int> A){
  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]--;
  	  }
  	}
  }
  return ans;
}

int solve2(vector<int> A){
  int ans = 0;
  for(int i = 1; i <= N; i++){
    for(int j = i + 1; j <= N; j++){
      if((A[i]+A[j]) % (i+j) == 0){
        ans++;
      }
    }
  }
  return ans;
}

signed main(){
  cin>>N;
  vector<int> A(N+1);
  for(int i = 1; i <= N; i++) cin>>A[i];
  cout << solve1(A) << "\n";
}
0