Cho dãy số nguyên dương và số nguyên dương Một đoạn con liên tiếp của dãy A có điểm đầu , điểm cuối với là tập hợp tất cả các phần tử với . Đếm số đoạn con có tổng tất cả các phần tử bằng 0 Input Dòng đầu ghi hai số nguyên dương Dòng tiếp theo ghi số nguyên Output Ghi số lượng đoạn con tìm được EXAMPLE INPUT Copy 5 OUTPUT1 Copy 2 1 -1 -2 0

#include <bits/stdc++.h>

using namespace std;
map<long long, int> m;
long long n, a, dp[1000005];
int main()
{
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    cin >> n;
    for (int i = 1; i <= n; i++)
    {
        cin >> a;
        dp[i] = dp[i - 1] + a;
    }
    long long res = 0;
    for (int i = 1; i <= n; i++)
    {
        res += m[dp[i]] + (dp[i] == 0);
        ++m[dp[i]];
    }
    cout << res;
    return 0;
}