AOJ 0636 フェーン現象

問題へのリンク

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 5 1 2");
            WillReturn.Add("0");
            WillReturn.Add("4");
            WillReturn.Add("1");
            WillReturn.Add("8");
            WillReturn.Add("1 2 2");
            WillReturn.Add("1 1 -2");
            WillReturn.Add("2 3 5");
            WillReturn.Add("1 2 -1");
            WillReturn.Add("1 3 5");
            //-5
            //-7
            //-13
            //-13
            //-18
        }
        else if (InputPattern == "Input2") {
            WillReturn.Add("2 2 5 5");
            WillReturn.Add("0");
            WillReturn.Add("6");
            WillReturn.Add("-1");
            WillReturn.Add("1 1 4");
            WillReturn.Add("1 2 8");
            //5
            //-35
        }
        else if (InputPattern == "Input3") {
            WillReturn.Add("7 8 8 13");
            WillReturn.Add("0");
            WillReturn.Add("4");
            WillReturn.Add("-9");
            WillReturn.Add("4");
            WillReturn.Add("-2");
            WillReturn.Add("3");
            WillReturn.Add("10");
            WillReturn.Add("-9");
            WillReturn.Add("1 4 8");
            WillReturn.Add("3 5 -2");
            WillReturn.Add("3 3 9");
            WillReturn.Add("1 7 4");
            WillReturn.Add("3 5 -1");
            WillReturn.Add("5 6 3");
            WillReturn.Add("4 4 9");
            WillReturn.Add("6 7 -10");
            //277
            //277
            //322
            //290
            //290
            //290
            //290
            //370
        }
        else {
            string wkStr;
            while ((wkStr = Console.ReadLine()) != null) WillReturn.Add(wkStr);
        }
        return WillReturn;
    }

    static long mN;
    static long UB;
    static long mS;
    static long mT;

    static void Main()
    {
        List<string> InputList = GetInputList();

        long[] wkArr = { };
        Action<string> SplitAct = pStr =>
            wkArr = pStr.Split(' ').Select(pX => long.Parse(pX)).ToArray();

        SplitAct(InputList[0]);

        mN = wkArr[0];
        UB = mN;
        mS = wkArr[2];
        mT = wkArr[3];

        var InsDualSegmentTree = new DualSegmentTree(UB);

        long[] AArr = InputList.Skip(1).Take((int)mN + 1).Select(pX => long.Parse(pX)).ToArray();
        for (long I = 0; I <= UB; I++) {
            InsDualSegmentTree.RangeAdd(I, I, AArr[I]);
        }

        // 階差数列を求める
        var KaisaList = new List<long>();
        for (long I = 1; I <= UB; I++) {
            KaisaList.Add(AArr[I] - AArr[I - 1]);
        }
        long CurrSum = KaisaList.Sum(pX => GetOndoDiff(pX));

        var sb = new System.Text.StringBuilder();
        foreach (string EachStr in InputList.Skip(1 + (int)mN + 1)) {
            SplitAct(EachStr);
            long AddRangeSta = wkArr[0];
            long AddRangeEnd = wkArr[1];
            long AddVal = wkArr[2];

            long MinusVal = 0;
            if (0 < AddRangeSta) {
                long Val1 = InsDualSegmentTree.GetVal(AddRangeSta - 1);
                long Val2 = InsDualSegmentTree.GetVal(AddRangeSta);
                MinusVal += GetOndoDiff(Val2 - Val1);
            }
            if (AddRangeEnd < UB) {
                long Val1 = InsDualSegmentTree.GetVal(AddRangeEnd);
                long Val2 = InsDualSegmentTree.GetVal(AddRangeEnd + 1);
                MinusVal += GetOndoDiff(Val2 - Val1);
            }

            long PlusVal = 0;
            InsDualSegmentTree.RangeAdd(AddRangeSta, AddRangeEnd, AddVal);
            var AfterKaisaList = new List<long>();
            if (0 < AddRangeSta) {
                long Val1 = InsDualSegmentTree.GetVal(AddRangeSta - 1);
                long Val2 = InsDualSegmentTree.GetVal(AddRangeSta);
                PlusVal += GetOndoDiff(Val2 - Val1);
            }
            if (AddRangeEnd < UB) {
                long Val1 = InsDualSegmentTree.GetVal(AddRangeEnd);
                long Val2 = InsDualSegmentTree.GetVal(AddRangeEnd + 1);
                PlusVal += GetOndoDiff(Val2 - Val1);
            }

            CurrSum -= MinusVal;
            CurrSum += PlusVal;

            sb.Append(CurrSum);
            sb.AppendLine();
        }
        Console.Write(sb.ToString());
    }

    // 階差を引数として、温度の変更値を返す
    static long GetOndoDiff(long pKaisa)
    {
        long Result = 0;
        long AbsVal = Math.Abs(pKaisa);

        if (pKaisa <= 0) Result = AbsVal * mT;
        else Result = -AbsVal * mS;

        return Result;
    }
}

// 区間加算、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 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

解説

階差数列で考えると
階差数列の変化に応じて決まる値を総和が解だと分かります。

後は、双対セグ木で区間加算をシュミレーションしつつ、
階差数列の変更分だけ再計算してます。