結果

問題 No.17 2つの地点に泊まりたい
ユーザー fmhr
提出日時 2016-07-02 16:41:38
言語 Go
(1.23.4)
結果
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
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 12 WA * 15
権限があれば一括ダウンロードができます

ソースコード

diff # 

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
}
0