AOJ 0572 たのしいカードゲーム

問題へのリンク

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("5 4");
            WillReturn.Add("1 2 3 4 5");
            WillReturn.Add("3 1 4 1");
            //2
        }
        else if (InputPattern == "Input2") {
            WillReturn.Add("6 5");
            WillReturn.Add("4 1 5 2 3 4");
            WillReturn.Add("4 5 4 2 3");
            //3
        }
        else if (InputPattern == "Input3") {
            WillReturn.Add("10 10");
            WillReturn.Add("1 2 1 5 4 5 1 4 7 6");
            WillReturn.Add("2 2 4 1 7 5 2 2 1 3");
            //4
        }
        else {
            string wkStr;
            while ((wkStr = Console.ReadLine()) != null) WillReturn.Add(wkStr);
        }
        return WillReturn;
    }

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

        var IndListDict = new Dictionary<int, List<int>>();
        for (int I = 0; I <= AArr.GetUpperBound(0); I++) {
            if (IndListDict.ContainsKey(AArr[I]) == false) {
                IndListDict[AArr[I]] = new List<int>();
            }
            IndListDict[AArr[I]].Add(I);
        }

        var AnswerList = new List<int>();
        for (int I = 0; I <= BArr.GetUpperBound(0); I++) {
            int CurrAInd = 0;
            int MatchCnt = 0;
            for (int J = I; J <= BArr.GetUpperBound(0); J++) {
                if (IndListDict.ContainsKey(BArr[J]) == false) break;
                var IndList = IndListDict[BArr[J]];

                int ResultInd = ExecNibunhou_LowerBound(CurrAInd, IndList);
                if (ResultInd == -1) break;

                MatchCnt++;
                CurrAInd = IndList[ResultInd] + 1;
                AnswerList.Add(MatchCnt);
            }
        }
        Console.WriteLine(AnswerList.Max());
    }

    // 二分法で、Val以上で最小の値を持つ、添字を返す
    static int ExecNibunhou_LowerBound(int pVal, List<int> pList)
    {
        if (pList.Count == 0) return -1;

        // 最後の要素がVal未満の特殊ケース
        if (pVal > pList.Last()) {
            return -1;
        }
        // 最初の要素がVal以上の特殊ケース
        if (pVal <= pList[0]) {
            return 0;
        }

        int L = 0;
        int R = pList.Count - 1;

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

            if (pList[Mid] >= pVal) {
                R = Mid;
            }
            else {
                L = Mid;
            }
        }
        return R;
    }
}

解説

配列Bは連続した区間になるので、
その区間の左端を全探索し、
配列Bの要素ごとの配列Aの対応Indは、二分探索してます。