yukicoder 0017 2つの地点に泊まりたい

■■■問題■■■

0からN-1までのN個の地点がある。
地点から地点の移動コストがM個与えられる。
また各地点に滞在コストがある。

0地点からN-1地点にたどり着くまでに、0地点とN-1地点以外の異なる2つの地点に滞在しなければならない。
滞在する地点は自由に決めて良い。
この条件での移動コストと滞在コストの合計の最小値を求めよ。

都市を通るけど、滞在しないこともできます。

■■■入力■■■

N
S0
S1
・・・
Si
・・・
SN-1
M
A1 B1 C1
A2 B2 C2
・・・
Aj Bj Cj
・・・
AM BM CM

1行目は地点の数N (4 <= N <= 50)が与えられる。
その後のN行は0地点からN-1地点までの
N個の地点の滞在コストSi (1 <= Si <= 1000,0 <= i <= N-1) が与えられる。
ただし、0地点とN-1地点の滞在コストは0で与えられる。

その次の行は、移動についての情報がM個 (1 <= M <= N(N-1)/2) が与えられる。
その後のM行は
移動元Aj (0 <= Aj <= N-1)、
移動先Bj (0 <= Bj <= N-1)、
移動コストCj (1 <= Cj <= 1000)の順でM個の情報が与えられる。(1 <= j <= M)

「AjからBj」また「BjからAj」のどちらも移動可能とする。(つまり無向グラフ)
Aj != Bjは保証されている。
AjからBj(その逆も含む)への遷移は複数ないとする。

0地点からN-1地点には必ずたどり着ける経路があることが保証される。
さらに0地点とN-1地点以外の異なる2つの地点に滞在することができることが保証される。

0地点から辿りつけない地点もあることに注意。

移動コストと滞在コストの単位は同一とし、加算できるものとする。

■■■出力■■■

移動コストと滞在コストの合計の最小値を一行で出力してください。
最後に改行してください。

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("4");
            WillReturn.Add("0");
            WillReturn.Add("2");
            WillReturn.Add("100");
            WillReturn.Add("0");
            WillReturn.Add("5");
            WillReturn.Add("0 1 1");
            WillReturn.Add("0 2 1");
            WillReturn.Add("1 2 1");
            WillReturn.Add("1 3 1");
            WillReturn.Add("2 3 2");
            //105
            //地点は全部で4個。
            //地点1と地点2にはかならず滞在しなければならない。
            //0→2→1→3 の順番で移動し、
            //地点1と地点2に滞在したとき最小コストになる。
        }
        else if (InputPattern == "Input2") {
            WillReturn.Add("4");
            WillReturn.Add("0");
            WillReturn.Add("3");
            WillReturn.Add("3");
            WillReturn.Add("0");
            WillReturn.Add("3");
            WillReturn.Add("0 1 1");
            WillReturn.Add("1 3 2");
            WillReturn.Add("2 3 4");
            //17
            //地点1と地点2にはかならず滞在しなければならない。
            //よって、0→1→3→2→3 のような移動が必要。
            //同じ地点を2度通っても良い。
        }
        else if (InputPattern == "Input3") {
            WillReturn.Add("16");
            WillReturn.Add("0");
            WillReturn.Add("15");
            WillReturn.Add("6");
            WillReturn.Add("23");
            WillReturn.Add("8");
            WillReturn.Add("193");
            WillReturn.Add("14");
            WillReturn.Add("13");
            WillReturn.Add("53");
            WillReturn.Add("16");
            WillReturn.Add("85");
            WillReturn.Add("12");
            WillReturn.Add("15");
            WillReturn.Add("5");
            WillReturn.Add("14");
            WillReturn.Add("0");
            WillReturn.Add("15");
            WillReturn.Add("0 1 17");
            WillReturn.Add("14 15 18");
            WillReturn.Add("7 8 87");
            WillReturn.Add("4 5 137");
            WillReturn.Add("8 9 17");
            WillReturn.Add("11 12 33");
            WillReturn.Add("5 6 177");
            WillReturn.Add("9 10 47");
            WillReturn.Add("10 11 27");
            WillReturn.Add("1 2 77");
            WillReturn.Add("6 7 114");
            WillReturn.Add("12 13 15");
            WillReturn.Add("2 3 127");
            WillReturn.Add("13 14 11");
            WillReturn.Add("3 4 85");
            //1000
        }
        else {
            string wkStr;
            while ((wkStr = Console.ReadLine()) != null) WillReturn.Add(wkStr);
        }
        return WillReturn;
    }

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

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

        long N = long.Parse(InputList[0]);
        long UB = N - 1;

        long[,] KyoriArr = new long[UB + 1, UB + 1];

        const long INFTY = long.MaxValue;

        // 初期化処理
        for (long I = 0; I <= UB; I++) {
            for (long J = 0; J <= UB; J++) {
                KyoriArr[I, J] = (I == J) ? 0 : INFTY;
            }
        }

        var SList = new List<long>();
        foreach (string EachStr in InputList.Skip(1).Take((int)N)) {
            SList.Add(int.Parse(EachStr));
        }
        long[] SArr = SList.ToArray();

        foreach (string EachStr in InputList.Skip(1 + (int)N + 1)) {
            SplitAct(EachStr);
            KyoriArr[wkArr[0], wkArr[1]] = wkArr[2];
            KyoriArr[wkArr[1], wkArr[0]] = wkArr[2];
        }

        // ワーシャルフロイド法
        for (long K = 0; K <= UB; K++) {
            for (long I = 0; I <= UB; I++) {
                if (KyoriArr[I, K] == INFTY) continue;
                for (long J = 0; J <= UB; J++) {
                    if (KyoriArr[K, J] == INFTY) continue;
                    long CurrKouho = KyoriArr[I, K] + KyoriArr[K, J];
                    if (KyoriArr[I, J] > CurrKouho) {
                        KyoriArr[I, J] = CurrKouho;
                    }
                }
            }
        }

        // 2つの滞在地点を全探索
        var AnswerKouhoList = new List<long>();
        for (long I = 1; I <= N - 2; I++) {
            for (long J = 1; J <= N - 2; J++) {
                if (I == J) continue;
                long CurrKouho = 0;
                CurrKouho += SArr[I];
                CurrKouho += SArr[J];

                if (KyoriArr[0, I] != INFTY && KyoriArr[I, J] != INFTY && KyoriArr[J, N - 1] != INFTY) {
                    AnswerKouhoList.Add(CurrKouho + KyoriArr[0, I] + KyoriArr[I, J] + KyoriArr[J, N - 1]);
                }
            }
        }
        Console.WriteLine(AnswerKouhoList.Min());
    }
}

解説

ワーシャルフロイド法で、
全点対間最短距離を求めてから、
2つの泊まる地点を全探索してます。