結果

問題 No.2956 Substitute with Average
ユーザー startcpp
提出日時 2024-10-25 00:16:36
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
(最新)
AC  
(最初)
実行時間 -
コード長 978 bytes
コンパイル時間 1,303 ms
コンパイル使用メモリ 94,860 KB
実行使用メモリ 13,636 KB
最終ジャッジ日時 2024-10-25 19:45:44
合計ジャッジ時間 5,799 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 7 TLE * 1 -- * 17
権限があれば一括ダウンロードができます

ソースコード

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;

signed main() {
	cin.tie(nullptr);
 	ios_base::sync_with_stdio(false);
 	
	int i, j;
	int n;
	int a[1 << 17], ra[1 << 17] = {0};
	int b[1 << 17] = {0};
	
	cin >> n;
	rep(i, n) cin >> a[i];
	rep(i, n) ra[i + 1] = ra[i] + a[i];
	assert(1 <= n && n <= 100000);
	
	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 <= 30);
	assert(minA >= 1);

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

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