题解 AT2828 【和がNの区間】

前缀和好题，但不知道为什么标签是 $\color{blue} \text{数论。}$

然后将前缀和加进去最后暴力枚举一下。但这里注意，暴力时如果已经超过了n那么就得终止循环，不然你会得到TLE。

代码如下:

#include <iostream>
#include <vector>
using namespace std;

vector<long long> sum(100005); //前缀和数组

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0); cout.tie(0); // 前2为读入输出优化，也可以用快读
    int n;
    cin >> n; // 读入数字的个数
    for(register int i = 1; i <= n; i++)
    {
        long long x;
        cin >> x; // 读入每个数，注意不要读入数组中
        sum[i] = sum[i - 1] + x; // 前缀和加入这个数
    }
    int cnt = 0; // 定义统计答案个数的变量
    for(register int i = 1; i <= n; i++)
    {
        if(sum[i] == n) {cnt++; continue;}
        for(register int j = i; j <= n; j++)
        {
            if(sum[j] - sum[i] == n) cnt++;
            else if(sum[j] - sum[i] > n) break; // 这里一定要加！不然会T
        }
    }
    cout << cnt << endl;
    return 0;
}