思路

当我们在某处空地发射弹球后，怪物受到攻击的顺序一定是左右或者右左交替，直到某一侧没有怪物时小球离开。

所以要恰好消灭所有怪物，空地左右的怪物生命值之和相差为 $0$ 或 $1$ 才可消灭怪物。

只要枚举空地位置，看当前发射弹球位置 $i$ 能否消灭怪物即可，即 $pre_{i-1}-pre_0$ 和 $pre_n-pre_{i}$ 的差值绝对值 $d \le 1$。

• $d=0$ 时，可以左右发射，贡献为 $2$。
• $d=1$ 时，只可以向一个方向发射，贡献为 $1$。

核心代码

for (int i = 1; i <= n; i++) {
	cin >> a[i];
	pre[i] = pre[i - 1] + a[i];
}
for (int i = 1; i <= n; i++) {
	if (a[i] == 0) {
		ll suml = pre[i - 1] - pre[0];
		ll sumr = pre[n] - pre[i];
		if (suml == sumr) {
			ans += 2;
		}
		if (abs(suml - sumr) == 1) {
			ans++;
		}
	}
}