寻找优美的密码

题目背景
在数字王国中，有一种特殊的密码被称为“优美串”。这种密码由一系列的0和1组成，且在一个优美串中，1的数量恰好比0的数量多一个。为了提高王国的安全性，国王命令大臣们找出所有可能的优美串。

题目描述
给定一个由0和1组成的数字串，你需要计算出这个数字串中所有可能的优美子串的数量。一个优美子串是指长度大于2的连续子串，其中1的数量比0的数量恰好多一个。

输入格式
第一行包含一个整数N，表示数字串的长度。 第二行包含一个长度为N的01数字串。

输出格式
输出一个整数，表示优美子串的数量。

输入样例
5
1 1 0 1 0
输出样例
3
说明
在样例中，满足条件的优美子串有：1 1 0, 1 0 1, 1 1 0 1 0，共3个。
约束条件
50%数据，$1\le N\le 100$
100%数据，$1\le N\le 10^5$
代码

#include <bits/stdc++.h>
using namespace std;
long long s[100010], v[200010];
long long a[100010];
long long t, ans;
int main() {
	int n;
	cin >> n;
	for (int i = 1; i <= n; i++)cin >> a[i];
	s[0] = 100001;
	v[100001] = 1;
	for (int i = 1; i <= n; i++) {
		if (a[i] == 1) t++;
		s[i] = s[i - 1] + (a[i] == 1 ? 1 : -1);
		ans += v[s[i] - 1];
		v[s[i]]++;
	}
	printf("%lld\n", ans - t);
	return 0;
}