比赛 EYOI暨SBOI暑假快乐赛6th 评测结果 WWWWWTTTTT
题目名称 Count 1s 最终得分 0
用户昵称 HeSn 运行时间 5.475 s
代码语言 C++ 内存使用 5.75 MiB
提交时间 2022-06-30 08:52:12
显示代码纯文本
#include<bits/stdc++.h>
using namespace std;
int n, a[200100], a1[200100], a0[200100], ans = 0;
bool v[200100] = {0};
int main() {
	freopen("count1s.in", "r", stdin);
	freopen("count1s.out", "w", stdout);
	cin >> n;
	for(int i = 1; i <= n; i ++) {
		cin >> a[i];
		a1[i] = a1[i - 1] + a[i];
		a0[i] = a0[i - 1] + (!a[i]);
	}
	v[0] = 1;
	ans ++;
	for(int i = 1; i <= n; i ++) {
		for(int j = i; j <= n; j ++) {
			ans += (!v[a1[n] - (a1[j] - a1[i - 1]) + (a0[j] - a0[i - 1])]);
			v[a1[n] - (a1[j] - a1[i - 1]) + (a0[j] - a0[i - 1])] = 1;
		}
	}
	cout << ans << endl;
	return 0;
}