#1022168 (C++14) No.2956 Substitute with Average

• 問題: No.2956 Substitute with Average
• ユーザー: startcpp
• 提出日時: 2024-10-24 23:57:00
• 言語: C++14 (gcc 12.3.0 + boost 1.83.0)
• 結果: WA
• 実行時間: -
• コード長: 896 bytes
• コンパイル時間: 846 ms
• コンパイル使用メモリ: 76,028 KB
• 実行使用メモリ: 6,824 KB
• 最終ジャッジ日時: 2024-10-25 19:45:38
• 合計ジャッジ時間: 1,962 ms
• ジャッジサーバーID （参考情報）: judge3 / judge4

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 6,816 KB
testcase_01	AC	2 ms; 6,816 KB
testcase_02	AC	2 ms; 6,816 KB
testcase_03	AC	2 ms; 6,820 KB
testcase_04	AC	2 ms; 6,816 KB
testcase_05	AC	2 ms; 6,816 KB
testcase_06	AC	2 ms; 6,820 KB
testcase_07	AC	2 ms; 6,816 KB
testcase_08	AC	2 ms; 6,816 KB
testcase_09	AC	2 ms; 6,816 KB
testcase_10	WA	-
testcase_11	WA	-
testcase_12	WA	-
testcase_13	WA	-
testcase_14	WA	-
testcase_15	WA	-
testcase_16	WA	-
testcase_17	WA	-
testcase_18	AC	13 ms; 6,816 KB
testcase_19	WA	-
testcase_20	WA	-
testcase_21	WA	-
testcase_22	AC	15 ms; 6,820 KB
testcase_23	AC	12 ms; 6,816 KB
testcase_24	WA	-
testcase_25	WA	-
testcase_26	WA	-
testcase_27	WA	-

ソースコード

//定数倍が軽そうなもの
#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];
int b[70000];

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 = ((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;
}