情報オリンピック    次の情報オリンピックの問題へ

2010 春合宿 階段


問題へのリンク


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 350");
            WillReturn.Add("315");
            WillReturn.Add("191");
            WillReturn.Add("98");
            WillReturn.Add("70");
            WillReturn.Add("126");
            WillReturn.Add("200");
            //9
        }
        else {
            string wkStr;
            while ((wkStr = Console.ReadLine()) != null) WillReturn.Add(wkStr);
        }
        return WillReturn;
    }

    const long Hou = 1234567;

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

        long[] HArr = InputList.Skip(1).Select(pX => long.Parse(pX)).ToArray();
        long UB = HArr.GetUpperBound(0);

        // 大きすぎる段差がある場合
        if (P < HArr.Max()) {
            Console.WriteLine(0);
            return;
        }

        // 累積和
        long[] RunSumArr = (long[])HArr.Clone();
        for (long I = 1; I <= UB; I++) {
            RunSumArr[I] += RunSumArr[I - 1];
        }

        // 場合の数[どこから上るか]なインラインDP表
        var DPSegTree = new DualSegmentTree(UB, 1234567);
        DPSegTree.RangeAdd(0, 0, 1);

        long Answer = 0;
        for (long I = 0; I <= UB; I++) {
            long Limit = P;
            if (I > 0) {
                Limit += RunSumArr[I - 1];
            }

            int ResultInd = ExecNibunhou_LowerOrEqual_Max(Limit, RunSumArr);
            if (ResultInd == -1) continue;

            long RangeSta = I + 1;
            long RangeEnd = ResultInd + 1;
            //Console.WriteLine("I={0}から配れる区間は[{1},{2}]", I, RangeSta, RangeEnd);

            long CurrVal = DPSegTree.GetVal(I);
            if (RangeSta > UB) {
                Answer += CurrVal;
                Answer %= Hou;
                continue;
            }
            if (RangeEnd > UB) {
                Answer += CurrVal;
                Answer %= Hou;
            }
            RangeEnd = Math.Min(RangeEnd, UB);
            DPSegTree.RangeAdd(RangeSta, RangeEnd, CurrVal);
        }
        Console.WriteLine(Answer);
    }

    // 二分法で、Val以下で最大の値を持つ、添字を返す
    static int ExecNibunhou_LowerOrEqual_Max(long pVal, long[] pArr)
    {
        // 最後の要素がVal以下の特殊ケース
        if (pVal >= pArr.Last()) {
            return pArr.GetUpperBound(0);
        }
        // 最初の要素がVal超えの特殊ケース
        if (pVal < pArr[0]) {
            return -1;
        }

        int L = 0;
        int R = pArr.GetUpperBound(0);

        while (L + 1 < R) {
            int Mid = (L + R) / 2;

            if (pArr[Mid] <= pVal) {
                L = Mid;
            }
            else {
                R = Mid;
            }
        }
        return L;
    }
}

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

    // ノードの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, long pHou)
    {
        mExternalArrUB = pExternalArrUB;

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

        mHou = pHou;
    }

    // 双対セグメント木の機能
    // 区間加算
    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)
    {
        long Result = FenwickTree_GetSum(pEnd) - FenwickTree_GetSum(pSta - 1);
        Result %= mHou;
        if (Result < 0) Result += mHou;
        return Result;
    }

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

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


解説

場合の数[どこから上るか]なインラインDP表を
双対セグ木で管理してます。