#101249 (Go) No.17 2つの地点に泊まりたい

• 問題: No.17 2つの地点に泊まりたい
• ユーザー: fmhr
• 提出日時: 2016-07-02 16:41:38
• 言語: Go (1.22.1)
• 結果: WA
• 実行時間: -
• コード長: 1,410 bytes
• コンパイル時間: 10,705 ms
• コンパイル使用メモリ: 235,976 KB
• 実行使用メモリ: 6,824 KB
• 最終ジャッジ日時: 2024-10-12 00:59:25
• 合計ジャッジ時間: 11,638 ms
• ジャッジサーバーID （参考情報）: judge3 / judge2

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	1 ms; 5,248 KB
testcase_01	AC	1 ms; 5,248 KB
testcase_02	AC	1 ms; 5,248 KB
testcase_03	WA	-
testcase_04	AC	3 ms; 5,248 KB
testcase_05	WA	-
testcase_06	AC	8 ms; 5,248 KB
testcase_07	AC	6 ms; 5,248 KB
testcase_08	WA	-
testcase_09	WA	-
testcase_10	WA	-
testcase_11	WA	-
testcase_12	AC	1 ms; 5,248 KB
testcase_13	AC	1 ms; 5,248 KB
testcase_14	AC	1 ms; 5,248 KB
testcase_15	AC	1 ms; 5,248 KB
testcase_16	AC	1 ms; 5,248 KB
testcase_17	WA	-
testcase_18	WA	-
testcase_19	WA	-
testcase_20	WA	-
testcase_21	WA	-
testcase_22	WA	-
testcase_23	WA	-
testcase_24	WA	-
testcase_25	AC	1 ms; 5,248 KB
testcase_26	WA	-

ソースコード

package main

import (
	"fmt"
	"math"
)

/*
Warchal_Floyd
ワーシャル・フロイド
すべての２点間の最短距離を求めるO(N^3)
1. d[N}[N]をMAXでうめる
2. d[i][i] = 0
3. k, i, j　でまわす
テスト用問題

*/

type WarchalFloyd struct {
	n int
	r [][]int
}

func NewWarchalFloyd(n int) *WarchalFloyd {
	r := make([][]int, n)
	for i := range r {
		r[i] = make([]int, n)
		for j := 0; j < n; j++ {
			if i == j {
				r[i][j] = 0
			} else {
				r[i][j] = math.MaxInt64/3
			}
		}
	}
	return &WarchalFloyd{n: n, r: r}
}

func (wf *WarchalFloyd) conect(u, v, l int) {
	wf.r[u][v] = l
}

func (wf *WarchalFloyd) build() {
	for i := 0; i < wf.n; i++ {
		for j := 0; j < wf.n; j++ {
			for k := 0; k < wf.n; k++ {
				wf.r[i][j] = min(wf.r[i][j], wf.r[i][k]+wf.r[k][j])
			}
		}
	}
}

func (wf *WarchalFloyd) cost(u, v int) int {
	return wf.r[u][v]
}

func main() {
	var N, M int
	fmt.Scan(&N)
	S := make([]int, N)
	for i:=0;i<N;i++ {
		fmt.Scan(&S[i])
	}

	fmt.Scan(&M)
	wf := NewWarchalFloyd(N)
	var u, v, l int
	for i := 0; i < M; i++ {
		fmt.Scan(&u, &v, &l)
		wf.conect(u, v, l)
		wf.conect(v, u, l)
	}
	wf.build()
	ans := math.MaxInt64/3
	for i := 1; i < N-1; i++ {
		for j := 1; j < N-1; j++ {
			if i == j {
				continue
			}
			ans = min(ans, wf.r[0][i]+wf.r[i][j]+wf.r[j][N-1]+S[i]+S[j])
		}
	}
	fmt.Println(ans)
}

func min(a, b int) int {
	if a > b {
		return b
	}
	return a
}