AGC023-A Zero-Sum Ranges

問題へのリンク

C#のソース

using System;
using System.Collections.Generic;
using System.Linq;

class Program
{
    static string InputPattern = "InputX";

    static List<string> GetInputList()
    {
        var WillReturn = new List<string>();

        if (InputPattern == "Input1") {
            WillReturn.Add("6");
            WillReturn.Add("1 3 -4 2 2 -2");
            //3
        }
        else if (InputPattern == "Input2") {
            WillReturn.Add("7");
            WillReturn.Add("1 -1 1 -1 1 -1 1");
            //12
        }
        else if (InputPattern == "Input3") {
            WillReturn.Add("5");
            WillReturn.Add("1 -2 3 -4 5");
            //0
        }
        else {
            string wkStr;
            while ((wkStr = Console.ReadLine()) != null) WillReturn.Add(wkStr);
        }
        return WillReturn;
    }

    static void Main()
    {
        List<string> InputList = GetInputList();
        long[] AArr = InputList[1].Split(' ').Select(pX => long.Parse(pX)).ToArray();

        var RunSumList = new List<long>();
        RunSumList.Add(0);
        for (int I = 0; I <= AArr.GetUpperBound(0); I++) {
            RunSumList.Add(RunSumList[I] + AArr[I]);
        }

        var CntList = new List<long>();
        foreach (var EachItem in RunSumList.GroupBy(pX => pX)) {
            CntList.Add(EachItem.Count());
        }

        long Answer = 0;
        foreach (long EachCnt in CntList) {
            long CurrPattern = EachCnt * (EachCnt - 1) / 2;
            Answer += CurrPattern;
        }
        Console.WriteLine(Answer);
    }
}

解説

1 3 -4 2 2 -2
というサンプルで考察すると

まず累積和を作ると下記になります。
0 1 4 0 2 4 2

同じ値のペアの個数の総和が解になので
値ごとの登場数を求めて、
nC2の総和を求めてます。