Go

こんにちは、ナナオです。 前回に引き続き競プロを実施していきたいと思います。 今回はAtCoderの問題を解いていきます。 問題は以下です。 004 - Cross Sum（★2） 実装 二次元の累積和ですね。 とりあえず愚直に解いてみます。 package main import "fmt" func main() { var h, w int fmt.Scan(&h, &w) g := make([][]int, h) for i := 0; i < h; i++ { g[i] = make([]int, w) for j := 0; j < w; j++ { fmt.Scan(&g[i][j]) } } hg := make([]int, h) for i := 0; i < h; i++ { s := 0 for j := 0; j < w; j++ { s += g[i][j] } hg[i] = s } wg := make([]int, w) for i := 0; i < w; i++ { s := 0 for j := 0; j < h; j++ { s += g[j][i] } wg[i] = s } ans := make([][]int, h) for i := 0; i < h; i++ { ans[i] = make([]int, w) for j := 0; j < w; j++ { ans[i][j] = hg[i] + wg[j] - g[i][j] } } for i := 0; i < h; i++ { for j := 0; j < w; j++ { if j < w - 1 { fmt.Printf("%d ", ans[i][j]) } else { fmt.Printf("%d", ans[i][j]) } } fmt.Println() } } ですがこれだとタイムオーバーしてしまいました。 ...

こんにちは、ナナオです。 前回に引き続き競プロを実施していきたいと思います。 今回の問題は以下です。 【マップの扱い 2】マップの書き換え・縦横 | レベルアップ問題集 | プログラミング学習サイト【paizaラーニング】 実装 以下のように実装しました。 （解答例のpythonコードをgoに書き換えただけですが…） package main import "fmt" func main(){ var h, w, d int fmt.Scan(&h, &w, &d) g := make([][]int, h) for i := 0; i < h; i++ { g[i] = make([]int, w) } min := func(a, b int) int { if a > b { return b } else { return a } } c := 1 switch d { case 1: for i := 0; i < h; i++ { for j := 0; j < min(w, i + 1); j++ { g[i - j][j] = c c++ } } for i := 1; i < w; i++ { for j := 0; j < min(h, w - i); j++ { g[h - 1 - j][i + j] = c c++ } } case 2: for i := 0; i < h; i++ { for j := 0; j < w; j++ { ty, tx := i, j g[ty][tx] = c c++ } } case 3: for i := 0; i < w; i++ { for j := 0; j < h; j++ { ty, tx := j, i g[ty][tx] = c c++ } } case 4: for i := 0; i < w; i++ { for j := 0; j < min(h, i + 1); j++ { g[j][i - j] = c c++ } } for i := 1; i < h; i++ { for j := 0; j < min(w, h - i); j++ { g[i + j][w - 1 - j] = c c++ } } } for i := 0; i < h; i++ { for j := 0; j < w; j++ { fmt.Printf("%d", g[i][j]) if j < w - 1 { fmt.Print(" ") } } fmt.Println() } } これに関してはなんか複雑で自分でよくわかっていません。。。よくないとはわかりつつも。。。 ...

こんにちは、ナナオです。 前回に引き続き競プロを実施していきたいと思います。 今回の問題は以下です。 【マップの扱い 3】マップの判定・縦横斜め | レベルアップ問題集 | プログラミング学習サイト【paizaラーニング】 実装 最初は以下のように実装しましたが、この実装だとダメでした。 縦横斜めのカラムを反対にするのではなく、すべてのカラムを反対にしているからです。 package main import "fmt" func main(){ var h, w, sy, sx int fmt.Scan(&h, &w) g := make([]string, h) for i := 0; i < h; i++ { fmt.Scan(&g[i]) } repl := func(sy, sx int, r rune) { runes := []rune(g[sy]) runes[sx] = r g[sy] = string(runes) } fmt.Scan(&sy, &sx) for dy := -20; dy <= 20; dy++ { for dx := -20; dx <= 20; dx++ { ty, tx := sy + dy, sx + dx if ty >= 0 && tx >= 0 && ty < h && tx < w { if g[ty][tx] == '.' { repl(ty, tx, '#') } else { repl(ty, tx, '.') } } } } for i := 0; i < h; i++ { fmt.Println(g[i]) } } ということで解答例を見つつ修正します。 ...

こんにちは、ナナオです。 前回に引き続き競プロを実施していきたいと思います。 今回の問題は以下です。 【マップの扱い 2】マップの書き換え・縦横 | レベルアップ問題集 | プログラミング学習サイト【paizaラーニング】 実装 以下のように実装しました。 package main import "fmt" func main(){ var h, w, sy, sx int fmt.Scan(&h, &w) g := make([]string, h) for i := 0; i < h; i++ { fmt.Scan(&g[i]) } fmt.Scan(&sy, &sx) repl := func(sy, sx int, r rune) { runes := []rune(g[sy]) runes[sx] = r g[sy] = string(runes) } dy := []int{1, -1, 0, 0, 0} dx := []int{0, 0, 1, -1, 0} for i := 0; i < 5; i++ { ty, tx := sy + dy[i], sx + dx[i] if g[ty][tx] == '.' { repl(ty, tx, '#') } else { repl(ty, tx, '.') } } for i := 0; i < h; i++ { fmt.Println(g[i]) } } ですが、ランタイムエラーが頻発してしまいました。。 ということで修正します。 package main import "fmt" func main(){ var h, w, sy, sx int fmt.Scan(&h, &w) g := make([]string, h) for i := 0; i < h; i++ { fmt.Scan(&g[i]) } fmt.Scan(&sy, &sx) repl := func(sy, sx int, r rune) { runes := []rune(g[sy]) runes[sx] = r g[sy] = string(runes) } dy := []int{1, -1, 0, 0, 0} dx := []int{0, 0, 1, -1, 0} for i := 0; i < 5; i++ { ty, tx := sy + dy[i], sx + dx[i] // 以下のif文を追加 if ty < 0 || tx < 0 || ty >= h || tx >= w { continue } if g[ty][tx] == '.' { repl(ty, tx, '#') } else { repl(ty, tx, '.') } } for i := 0; i < h; i++ { fmt.Println(g[i]) } } インデックスが表からはみ出していた場合の考慮が漏れていました。 ...

こんにちは、ナナオです。 前回に引き続き競プロを実施していきたいと思います。 今回の問題は以下です。 【マップの扱い 1】マップの書き換え・1 マス | レベルアップ問題集 | プログラミング学習サイト【paizaラーニング】 実装 以下のように実装しました。 package main import "fmt" func main(){ var h, w, x, y int fmt.Scan(&h, &w) g := make([]string, h) for i := 0; i < h; i++ { fmt.Scan(&g[i]) } fmt.Scan(&y, &x) if g[y][x] == '#' { runes := []rune(g[y]) runes[x] = '.' g[y] = string(runes) } else { runes := []rune(g[y]) runes[x] = '#' g[y] = string(runes) } for i := 0; i < h; i++ { fmt.Println(g[i]) } } 文字の置き換えがGoだと少し面倒ですが、実装できましたね。 では今日はこれで👍

こんにちは、ナナオです。 前回に引き続き競プロを実施していきたいと思います。 今回の問題は以下です。 りんご拾い（情報を持ちながらの移動） | レベルアップ問題集 | プログラミング学習サイト【paizaラーニング】 実装 以下のように実装しました。 package main import "fmt" func main(){ var n int fmt.Scan(&n) g := make([][]int, n) for i := 0; i < n; i++ { g[i] = make([]int, n) } for i := 1; i < n; i++ { var a, b int fmt.Scan(&a, &b) g[a - 1][b - 1] = 1 g[b - 1][a - 1] = 1 } for i := 0; i < n; i++ { var k int fmt.Scan(&k) for j := 0; j < n; j++ { if g[i][j] == 1 { g[i][j] = k } } } index := 0 sum := 0 for i := 0; i < n; i++ { for j := 0; j < n; j++ { if g[index][j] > 0 { sum += g[index][j] if i < n - 2 { g[j][index] = 0 } index = j break } } fmt.Println(sum) } } 前回のコードを少し弄れば解決しましたね。 ...

こんにちは、ナナオです。 前回に引き続き競プロを実施していきたいと思います。 今回の問題は以下です。 一方通行（グラフ上の移動） | レベルアップ問題集 | プログラミング学習サイト【paizaラーニング】 実装 まずは以下のように実装しました。 package main import "fmt" func main(){ var n int fmt.Scan(&n) g := make([][]int, n) for i := 0; i < n; i++ { g[i] = make([]int, n) } for i := 1; i < n; i++ { var a, b int fmt.Scan(&a, &b) g[a - 1][b - 1] = i g[b - 1][a - 1] = i } fmt.Println(1) for i := 0; i < n; i++ { for j := 0; j < n; j++ { if g[i][j] == i + 1 { fmt.Println(j + 1) break } } } } ですがこれだと入力例１は解けてもそれ以外が解けません。 ということで少し改修しました。 package main import "fmt" func main(){ var n int fmt.Scan(&n) g := make([][]int, n) for i := 0; i < n; i++ { g[i] = make([]int, n) } for i := 1; i < n; i++ { var a, b int fmt.Scan(&a, &b) g[a - 1][b - 1] = i g[b - 1][a - 1] = i } fmt.Println(1) i := 0 cnt := 1 for { for j := 0; j < n; j++ { if g[i][j] == cnt { fmt.Println(j + 1) i = j cnt++ break } } if cnt == n { break } } } 問題は解けるようになったはずですが、タイムオーバーになってしまいました。 ...

こんにちは、ナナオです。 前回に引き続き競プロを実施していきたいと思います。 今回の問題は以下です。 重みあり有向グラフの隣接行列と隣接リスト | レベルアップ問題集 | プログラミング学習サイト【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() } } これで実装できました。 重み付き有向グラフみたいな難しそうな言葉に惑わされそうになりますが、理解してしまえば簡単ですね。 では今回はこれで👍

こんにちは、ナナオです。 前回に引き続き競プロを実施していきたいと思います。 今回の問題は以下です。 有向グラフの隣接行列と隣接リスト | レベルアップ問題集 | プログラミング学習サイト【paizaラーニング】 実装 これも一瞬よくわからなかったので、AIと少し壁打ちしました。 これまでは双方に関係性があることが前提でしたが、これが片方のみになるということですね。 例として、以下のような方向の場合 1 → 2 1 → 3 4 → 2 4 → 5 5 → 3 以下のような表になります。 実装としては以下のようになります。 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 int fmt.Scan(&a, &b) g[a - 1][b - 1] = 1 // 前回からここの部分が変わっただけ } 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] == 1 { fmt.Printf("%d", j) } } fmt.Println() } } 前回から変わった部分はコメントの部分だけです。 ...

こんにちは、ナナオです。 前回に引き続き競プロを実施していきたいと思います。 今回の問題は以下です。 隣接リスト | レベルアップ問題集 | プログラミング学習サイト【paizaラーニング】 実装 問題文を読んでも、何を出力すればよいのかいまいちよく分かりませんでした。 そこでこれもAIと壁打ちしました。 問題の例の図の場合は以下のように出力すればよいです。 図1（五角形のような形）を「隣接リスト」にすると、こんな感じのメモ帳セットになります。 ※コンピュータの世界なので、番号を「-1」して 0番から数えるルールで書きますね。 0番（頂点1）のリスト: [1, 2] （2番と3番とつながっている） 1番（頂点2）のリスト: [0, 3] （1番と4番とつながっている） 2番（頂点3）のリスト: [0, 4] （1番と5番とつながっている） 3番（頂点4）のリスト: [1, 4] （2番と5番とつながっている） 4番（頂点5）のリスト: [2, 3] （3番と4番とつながっている） ということで以下のように実装しました。 前回とあまり変わらないですね。 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 int fmt.Scan(&a, &b) g[a - 1][b - 1] = 1 g[b - 1][a - 1] = 1 } for i := 0; i < n; i++ { for j := 0; j < n; j++ { if g[i][j] == 1 { fmt.Printf("%d", j) } } fmt.Println() } } ということで実装できました。 ...