結果

問題 No.2956 Substitute with Average
ユーザー startcpp
提出日時 2024-10-24 23:51:00
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 827 bytes
コンパイル時間 848 ms
コンパイル使用メモリ 74,972 KB
実行使用メモリ 6,824 KB
最終ジャッジ日時 2024-10-25 19:45:34
合計ジャッジ時間 1,944 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 10 WA * 15
権限があれば一括ダウンロードができます

ソースコード

diff # 

//定数倍が軽そうなもの
#pragma GCC target("avx2")
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")
#include <iostream>
#include <vector>
#include <cassert>
#define rep(i, n) for(i = 0; i < n; i++)
using namespace std;

int n;
int a[70000], ra[70001];

signed main() {
	int i, j;

	cin >> n;
	rep(i, n) cin >> a[i];
	rep(i, n) ra[i + 1] = ra[i] + a[i];
	assert(1 <= n && n <= 70000);
	
	int maxA = a[0], minA = a[0];
	rep(i, n) maxA = max(maxA, a[i]);
	rep(i, n) minA = min(minA, a[i]);
	assert(maxA <= 50);
	assert(minA >= 1);

	long long ans = n * (n - 1) / 2 + 1;
	
	for (i = 2; i <= n; i++) {
		for (j = 0; j + i <= n; j++) { //区間[j, j + i)を確認
			if (i * a[j] == ra[j + i] - ra[j]) ans--;
			else if (i * a[j + i - 1] == ra[j + i] - ra[j]) ans--;
		}
	}

	cout << ans << endl;
	return 0;
}
0