Submission #7845589

---

Source Code Expand

using System;
using System.Collections.Generic;
using System.Linq;
using static System.Console;
using System.Runtime.CompilerServices;
using static MyUtil;

class MyUtil
{
    public static long[] ReadLongArray()
    {
        return ReadLine().Split().Select(x => long.Parse(x)).ToArray();
    }
}

class Program
{
    public static void Main()
    {
        int n = int.Parse(ReadLine());
        long[] a = ReadLongArray();

        long[] b = new long[n+1];
        for (int i = 1; i <= n; i++)
            b[i] = b[i-1] + a[i-1];

        Array.Sort(b);

        long ans = 0;
        long curr = b[0] - 1;
        long num = 1;
        for (int i = 0; i <= n; i++)
        {
            if (b[i] == curr) num++;
            else
            {
                ans += num * (num - 1) / 2;
                curr = b[i];
                num = 1;
            }
        }
        ans += num * (num - 1) / 2;
        WriteLine(ans);
    }
}

Submission Info

Judge Result