思路

一个连续子序列（片段）可以由其起始位置 $i$ 和结束位置 $j$ 定义（其中 $1 \le i \le j \le n$）。 根据题意，一个从索引 $i$ 到 $j$ 的片段是“平衡片段”的条件是： $\sum_{k=i}^{j} A_k = j - i + 1$ 这里的 $\sum_{k=i}^{j} A_k$ 是片段中所有水晶的能量和，而 $j - i + 1$ 是该片段的长度。

直接暴力枚举所有的起始位置 $i$ 和结束位置 $j$，然后计算和并判断，时间复杂度会达到 $O(n^2)$，对于 $n=2 \times 10^5$ 的数据规模来说太慢了，会超时。我们需要一个更高效的算法。