ABC379-E Sum of All Substrings

問題へのリンク

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("3");
            WillReturn.Add("379");
            //514
        }
        else if (InputPattern == "Input2") {
            WillReturn.Add("30");
            WillReturn.Add("314159265358979323846264338327");
            //369673254065355789035427227741
        }
        else {
            string wkStr;
            while ((wkStr = Console.ReadLine()) != null) WillReturn.Add(wkStr);
        }
        return WillReturn;
    }

    static long[] GetSplitArr(string pStr)
    {
        return (pStr == "" ? new string[0] : pStr.Split(' ')).Select(pX => long.Parse(pX)).ToArray();
    }

    static void Main()
    {
        List<string> InputList = GetInputList();
        string S = InputList[1];

        char[] NumArr = S.ToCharArray();
        long UB = NumArr.GetUpperBound(0);

        var InsDualSegmentTree = new DualSegmentTree(NumArr.Length);

        for (long I = 0; I <= UB; I++) {
            long CurrNum = NumArr[I] - '0';

            long LeftCnt = I;
            long RightCnt = NumArr.Length - LeftCnt - 1;

            long RangeSta = 0;
            long RangeEnd = RightCnt;
            InsDualSegmentTree.RangeAdd(RangeSta, RangeEnd, CurrNum * (LeftCnt + 1));
        }

        var ValDict = new Dictionary<long, long>();
        for (long I = 0; I <= InsDualSegmentTree.GetUB(); I++) {
            ValDict[I] = InsDualSegmentTree[I];
        }

        var AnswerList = new List<long>();
        long[] KeysArr = ValDict.Keys.ToArray();
        Array.Sort(KeysArr);

        for (long I = 0; I <= KeysArr.GetUpperBound(0); I++) {
            long CurrVal = ValDict[I];

            AnswerList.Add(CurrVal % 10);

            long NewKey = I + 1;
            if (ValDict.ContainsKey(NewKey) == false) {
                ValDict[NewKey] = 0;
            }

            ValDict[NewKey] += CurrVal / 10;
            ValDict.Remove(I);
        }

        if (ValDict.Count > 0) {
            long CurrKey = ValDict.Keys.Min();
            while (true) {
                if (ValDict.ContainsKey(CurrKey) == false) {
                    break;
                }
                long CurrVal = ValDict[CurrKey];
                AnswerList.Add(CurrVal % 10);
                CurrKey++;
                if (ValDict.ContainsKey(CurrKey) == false) {
                    ValDict[CurrKey] = 0;
                }
                if (CurrVal / 10 > 0) {
                    ValDict[CurrKey] += CurrVal / 10;
                }
                else {
                    break;
                }
            }
        }

        AnswerList.Reverse();
        var sb = new System.Text.StringBuilder();
        AnswerList.ForEach(pX => sb.Append(pX));
        string AnswerStr = sb.ToString();
        if (AnswerStr.Length > 1) {
            AnswerStr = AnswerStr.TrimStart('0');
        }
        Console.WriteLine(AnswerStr);
    }
}

// 区間加算、1点取得な双対セグ木(フェニック木使用)
#region DualSegmentTree
internal class DualSegmentTree
{
    private long[] mBitArr; // 内部配列(1オリジンなため、添字0は未使用)
    private long mExternalArrUB;

    // ノードのIndexの列挙を返す
    internal IEnumerable<long> GetNodeIndEnum()
    {
        for (long I = 0; I <= GetUB(); I++) {
            yield return I;
        }
    }

    // ノードのUBを返す
    internal long GetUB()
    {
        return mExternalArrUB;
    }

    // コンストラクタ
    // フェニック木の外部配列(0オリジン)のUBを指定
    internal DualSegmentTree(long pExternalArrUB)
    {
        mExternalArrUB = pExternalArrUB;

        // フェニック木の外部配列は0オリジンで、
        // フェニック木の内部配列は1オリジンなため、2を足す
        mBitArr = new long[pExternalArrUB + 2];
    }

    // インデクサ
    internal long this[long pInd]
    {
        get { return GetVal(pInd); }
        set { RangeAdd(pInd, pInd, value - GetVal(pInd)); }
    }

    // 双対セグメント木の機能
    // 区間加算
    internal void RangeAdd(long pSta, long pEnd, long AddVal)
    {
        pSta++; // 1オリジンに変更
        pEnd++; // 1オリジンに変更

        long ImosSta = pSta;
        long ImosEnd = pEnd + 1;

        // いもす法
        FenwickTree_Add(ImosSta, AddVal);
        if (ImosEnd <= mBitArr.GetUpperBound(0)) {
            FenwickTree_Add(ImosEnd, -AddVal);
        }
    }

    // 双対セグメント木の機能
    // 1点取得
    internal long GetVal(long pInd)
    {
        pInd++; // 1オリジンに変更
        return FenwickTree_GetSum(1, pInd);
    }

    // フェニック木の機能
    // [pSta,pEnd] のSumを返す
    private long FenwickTree_GetSum(long pSta, long pEnd)
    {
        return FenwickTree_GetSum(pEnd) - FenwickTree_GetSum(pSta - 1);
    }

    // フェニック木の機能
    // [0,pEnd] のSumを返す
    private long FenwickTree_GetSum(long pEnd)
    {
        long Sum = 0;
        while (pEnd >= 1) {
            Sum += mBitArr[pEnd];
            pEnd -= pEnd & -pEnd;
        }
        return Sum;
    }

    // フェニック木の機能
    // [I] に Xを加算
    private void FenwickTree_Add(long pI, long pX)
    {
        while (pI <= mBitArr.GetUpperBound(0)) {
            mBitArr[pI] += pX;
            pI += pI & -pI;
        }
    }
}
#endregion

解説

1 2 3 4 5 9 1 2
で9の寄与度を考えます。
1桁目の   9になるのが3通り
2桁目で  90になるのが3通り
3桁目で 900になるのが3通り
4桁目で9000になるのが3通り
なので、
筆算を意識して、
総和[10の指数]な双対セグ木を用意し、
下位桁から筆算をして解けます。

繰り上がり対策で、Dictを用意し、
最上位桁が0になるまで、計算してます。