leetcode 930 和相同的二元子数组

在由若干 0 和 1 组成的数组 A 中，有多少个和为 S 的非空子数组。

示例：

输入：A = [1,0,1,0,1], S = 2
输出：4
解释：
如下面黑体所示，有 4 个满足题目要求的子数组：
[1,0,1,0,1]
[1,0,1,0,1]
[1,0,1,0,1]
[1,0,1,0,1]

这道题我的处理方式就是先把原来的数组处理成这样一个数组：连续 0的个数，1，连续0的个数，1，连续0的个数，1…
比如数组1 0 1 0 1 处理完成之后变成 0 -1 1 -1 1 -1 0 (为了避免歧义 这里-1就是表示1的位置）
比如数组 10001 处理完成之后变成 0 -1 3 -1 0
之后再根据S分类讨论：
如果S=0 那就对于这个数组中每个不为-1的数 也就是每个连续0的个数x ,求1+2+3+…+x 然后再加入总和中
比如0000 ，S=0，那么一共有4+3+2+1=10种可能，0010000,一共有2+1+4+3+2+1=13种可能。
如果S不等于0 那么左下标就从0起 右下标从2s起，依次累加(zeros[left]+1)(zeros[i]+1)
代码如下
为了处理方便，在1开头的数组中，处理后的数组头部中加了一个0，表示开头有0个0
在1结尾的数组中，处理后的数组尾部加了一个0，表示尾巴有0个0

int numSubarraysWithSum(vector<int>& A, int S) {
	int ret = 0;
	int len = A.size();
	vector<int> zeros;
	zeros.clear();
	int cnt = 0;
	for (int i = 0; i < len; i++)
	{
		if (A[i] == 1)
		{
			zeros.push_back(cnt);
			zeros.push_back(-1);
			cnt = 0;
		}
		else cnt++;
	}
	zeros.push_back(cnt);
	if (S == 0)
	{
		for (int i = 0; i < zeros.size(); i++)
		{
			if (zeros[i] != -1)
			{
				ret += zeros[i] * (zeros[i] + 1) / 2;
			}
		}
		return ret;
	}
	int i;
	int left = 0;
	if ((i=S*2)>zeros.size()) return 0;
	while (i < zeros.size())
	{
		ret += (zeros[left] + 1)*(zeros[i] + 1);
		left += 2;
		i += 2;
	}
	return ret;
}