ABC229-D Longest X

問題へのリンク

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("XX...X.X.X.");
            WillReturn.Add("2");
            //5
        }
        else if (InputPattern == "Input2") {
            WillReturn.Add("XXXX");
            WillReturn.Add("200000");
            //4
        }
        else {
            string wkStr;
            while ((wkStr = Console.ReadLine()) != null) WillReturn.Add(wkStr);
        }
        return WillReturn;
    }

    static char[] mSArr;
    static long UB;

    static long mK;

    static Fenwick_Tree mIns_Fenwick_Tree;

    static void Main()
    {
        List<string> InputList = GetInputList();
        mSArr = InputList[0].ToCharArray();
        mK = long.Parse(InputList[1]);

        UB = mSArr.GetUpperBound(0);
        mIns_Fenwick_Tree = new Fenwick_Tree(UB);
        for (long I = 0; I <= UB; I++) {
            if (mSArr[I] == '.') {
                mIns_Fenwick_Tree.Add(I, 1);
            }
        }

        long L = 0;
        long R = mSArr.Length + 1;

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

            if (CanAchive(Mid)) {
                L = Mid;
            }
            else {
                R = Mid;
            }
        }
        Console.WriteLine(L);
    }

    // 長さNの区間をXにできるか
    static bool CanAchive(long pLen)
    {
        for (long I = 0; I <= UB; I++) {
            long Sta = I;
            long End = I + pLen - 1;
            if (End > UB) break;

            long RangeDotCnt = mIns_Fenwick_Tree.GetSum(Sta, End);
            if (RangeDotCnt <= mK) return true;
        }
        return false;
    }
}

// フェニック木
#region Fenwick_Tree
internal class Fenwick_Tree
{
    private long[] mBitArr;
    private long mExternalArrUB;

    // コンストラクタ
    internal Fenwick_Tree(long pExternalArrUB)
    {
        mExternalArrUB = pExternalArrUB;

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

    // [pSta,pEnd] のSumを返す
    internal long GetSum(long pSta, long pEnd)
    {
        return GetSum(pEnd) - GetSum(pSta - 1);
    }

    // [0,pEnd] のSumを返す
    internal long GetSum(long pEnd)
    {
        pEnd++; // 1オリジンに変更

        long Sum = 0;
        while (pEnd >= 1) {
            Sum += mBitArr[pEnd];
            pEnd -= pEnd & -pEnd;
        }
        return Sum;
    }

    // [I] に Xを加算
    internal void Add(long pI, long pX)
    {
        pI++; // 1オリジンに変更

        while (pI <= mBitArr.GetUpperBound(0)) {
            mBitArr[pI] += pX;
            pI += pI & -pI;
        }
    }
}
#endregion

解説

フェニック木で区間の.の数を高速に求めることができるようにしてから、
答えを二分探索してます。