51Nod 2518 和为S c/c++题解

题目描述

小b有一个01序列A，她想知道A有多少个非空连续子序列和为S。
你能帮帮她吗？
输入
第一行输入一个数n，表示A的长度；
第二行输入n个数‘0’或‘1’，表示A中的元素，以空格隔开；
第三行输入一个非负整数S；
其中0≤S≤n≤30000。
输出
输出一个数，表示子数组的个数
输入样例
5
1 0 1 0 1
2
输出样例
4

代码：

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <climits>
#include <cstring>
#include <string>
#include <algorithm>
#include <vector>
#include <deque>
#include <list>
#include <utility>
#include <set>
#include <map>
#include <stack>
#include <queue>
#include <bitset>
#include <iterator>
using namespace std;

typedef long long ll;
const int inf = 0x3f3f3f3f;
const ll  INF = 0x3f3f3f3f3f3f3f3f;
const double PI = acos(-1.0);
const double E = exp(1.0);
const int MOD = 1e9+7;
const int MAX = 3e4+5;
ll n,s;
ll a[MAX];

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);

    while(cin >> n)
    {
        int x;
        for(int i = 1; i <= n; i++)
        {
            cin >> x;
            a[i] = a[i-1] + x;// 0 1
        }
        cin >> s;
        ll sum = 0;
        for(int i = 1; i <= n; i++)
        {
            for(int j = i; j <= n; j++)
            {
                if(a[j] - a[i-1] == s)
                    sum++;
            }
        }
        cout << sum << endl;
    }

    return 0;
}