ARC125-A Dial Up

問題へのリンク

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

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

        HashSet<int> SDistinctSet = new HashSet<int>(SArr.Distinct());
        HashSet<int> TDistinctSet = new HashSet<int>(TArr.Distinct());

        if (TDistinctSet.IsSubsetOf(SDistinctSet) == false) {
            Console.WriteLine(-1);
            return;
        }

        int Answer = 0;
        bool IsChanged = false;
        int CurrS = SArr[0];
        for (int I = 0; I <= TArr.GetUpperBound(0); I++) {
            if (CurrS != TArr[I]) {
                if (IsChanged) {
                    Answer += 2;
                }
                else {
                    Answer += ShiftCost(SArr) + 1;
                    IsChanged = true;
                }
                CurrS = TArr[I];
            }
            else {
                Answer++;
            }
        }
        Console.WriteLine(Answer);
    }

    static int ShiftCost(int[] pSArr)
    {
        var KouhoList = new List<int>();

        int UB = pSArr.GetUpperBound(0);

        // 右シフトする場合
        for (int I = 1; I <= UB; I++) {
            if (pSArr[I] != pSArr[0]) {
                KouhoList.Add(I);
                break;
            }
        }

        // 左シフトする場合
        for (int I = UB; 1 <= I; I--) {
            if (pSArr[I] != pSArr[0]) {
                KouhoList.Add(UB - I + 1);
                break;
            }
        }

        return KouhoList.Min();
    }
}

解説

最初に、数列Tの要素の集合が
数列Sの要素の集合の部分集合かを判定してます。

部分集合でない場合は、-1を出力して、終了です。

後は、Tを走査して、S[0]と一致してたらコスト1を加算。
S[0]と不一致なら、左シフトまたは右シフトの最小コストを加算してから、コスト1を加算。
といったシュミレーションを行ってます。