こんにちは、ナナオです。

前回に引き続き競プロを実施していきたいと思います。

今回の問題は以下です。

重みあり有向グラフの隣接行列と隣接リスト | レベルアップ問題集 | プログラミング学習サイト【paizaラーニング】

実装

前回の実装を少し変えてあげれば実装できます。

具体的には今まで重みとして1を与えていた部分をkに変えてあげればいいです。

package main
import "fmt"
func main(){
    var n, m int
    fmt.Scan(&n, &m)
    g := make([][]int, n)
    for i := 0; i < n; i++ {
        g[i] = make([]int, n)
    }
    for i := 0; i < m; i++ {
        var a, b, k int
        fmt.Scan(&a, &b, &k)
        g[a - 1][b - 1] = k // 前回まで必ず1を入れていた箇所を、入力されたkにしてあげる
    }
    for i := 0; i < n; i++ {
        for j := 0; j < n; j++ {
            fmt.Printf("%d", g[i][j])
        }
        fmt.Println()
    }
    for i := 0; i < n; i++ {
        for j := 0; j < n; j++ {
            if g[i][j] > 0 {
                fmt.Printf("%d(%d)", j, g[i][j])
            }
        }
        fmt.Println()
    }
}

これで実装できました。

重み付き有向グラフみたいな難しそうな言葉に惑わされそうになりますが、理解してしまえば簡単ですね。

では今回はこれで👍