競技プログラミング

こんにちは、ナナオです。 前回に引き続き競プロを実施していきたいと思います。 今回の問題は以下です。 いびつなひとりリバーシ（１ターン） | レベルアップ問題集 | プログラミング学習サイト【paizaラーニング】 実装 以下のように実装しました。 package main import "fmt" func main(){ var h, w, sy, sx int fmt.Scan(&h, &w) g := make([][]rune, h) for i := 0; i < h; i++ { var s string fmt.Scan(&s) g[i] = []rune(s) for j, v := range s { if v == '!' { sy, sx = i, j } } } g[sy][sx] = '*' dy := []int{1, -1, 0, 0, 1, 1, -1, -1} dx := []int{0, 0, 1, -1, 1, -1, 1, -1} for i := 0; i < 8; i++ { for j := 1;; j++ { ly, lx := sy + dy[i] * j, sx + dx[i] * j if ly >= 0 && lx >= 0 && ly < h && lx < w { if g[ly][lx] == '*' { for k := j - 1; k > 0; k-- { ny, nx := sy + dy[i] * k, sx + dx[i] * k g[ny][nx] = '*' } break } else if g[ly][lx] == '#' { break } } else { break } } } for i := 0; i < h; i++ { fmt.Println(string(g[i])) } } 前回のコードとそこまで変わりがないですね。ifが一つ増えたくらい。 ...

こんにちは、ナナオです。 前回に引き続き競プロを実施していきたいと思います。 今回の問題は以下です。 リバーシの操作 | レベルアップ問題集 | プログラミング学習サイト【paizaラーニング】 実装 以下のように実装しました。 実装はほぼ変わらんですね。 package main import "fmt" func main(){ var h, w, sy, sx int fmt.Scan(&h, &w, &sy, &sx) g := make([][]rune, h) for i := 0; i < h; i++ { var s string fmt.Scan(&s) g[i] = []rune(s) } g[sy][sx] = '*' dy := []int{1, -1, 0, 0, 1, 1, -1, -1} dx := []int{0, 0, 1, -1, 1, -1, 1, -1} for i := 0; i < 8; i++ { for j := 1;; j++ { ly, lx := sy + dy[i] * j, sx + dx[i] * j if ly >= 0 && lx >= 0 && ly < h && lx < w { if g[ly][lx] == '*' { for k := j - 1; k > 0; k-- { ny, nx := sy + dy[i] * k, sx + dx[i] * k g[ny][nx] = '*' } break } } else { break } } } for i := 0; i < h; i++ { fmt.Println(string(g[i])) } } いい感じじゃないでしょうか👍

こんにちは、ナナオです。 前回に引き続き競プロを実施していきたいと思います。 今回の問題は以下です。 リバーシの操作（斜め） | レベルアップ問題集 | プログラミング学習サイト【paizaラーニング】 実装 以下のように実装しました。 （ネストが深い…！） package main import "fmt" func main(){ var h, w, sy, sx int fmt.Scan(&h, &w, &sy, &sx) g := make([][]rune, h) for i := 0; i < h; i++ { var s string fmt.Scan(&s) g[i] = []rune(s) } g[sy][sx] = '*' dy := []int{1, 1, -1, -1} dx := []int{1, -1, 1, -1} for i := 0; i < 4; i++ { for j := 1;; j++ { ly, lx := sy + dy[i] * j, sx + dx[i] * j if ly >= 0 && lx >= 0 && ly < h && lx < w { if g[ly][lx] == '*' { for k := j - 1; k > 0; k-- { ny, nx := sy + dy[i] * k, sx + dx[i] * k g[ny][nx] = '*' } break } } else { break } } } for i := 0; i < h; i++ { fmt.Println(string(g[i])) } } 前回はキューを使いましたが、まぁこちらの方がすっきりして見えますね。 ...

こんにちは、ナナオです。 前回に引き続き競プロを実施していきたいと思います。 今回の問題は以下です。 裏返せる可能性（斜め） | レベルアップ問題集 | プログラミング学習サイト【paizaラーニング】 実装 以下のように実装しました。 package main import "fmt" func main(){ var h, w, sy, sx int fmt.Scan(&h, &w, &sy, &sx) g := make([][]rune, h) for i := 0; i < h; i++ { row := make([]rune, w) for j := 0; j < w; j++ { if i == sy && j == sx { row[j] = '!' } else { row[j] = '.' } } g[i] = row } dy := []int{1, 1, -1, -1} dx := []int{1, -1, 1, -1} for i := 0; i < 4; i++ { for j := 1;; j++ { ly, lx := sy + dy[i] * j, sx + dx[i] * j if ly >= 0 && lx >= 0 && ly < h && lx < w { g[ly][lx] = '*' } else { break } } } for i := 0; i < h; i++ { fmt.Println(string(g[i])) } } 割とすっきり書けたと思います👍 ...

こんにちは、ナナオです。 前回に引き続き競プロを実施していきたいと思います。 今回の問題は以下です。 リバーシの操作（縦横） | レベルアップ問題集 | プログラミング学習サイト【paizaラーニング】 実装 結構悩んだりAIと壁打ちしたりした結果、以下のような実装になりました。 package main import "fmt" func main(){ var h, w, y, x int fmt.Scan(&h, &w, &y, &x) g := make([][]rune, h) for i := 0; i < h; i++ { var s string fmt.Scan(&s) g[i] = []rune(s) } // 4方向でループする dy := []int{1, -1, 0, 0} dx := []int{0, 0, 1, -1} // 置いた石のキュー qy := []int{y} qx := []int{x} for i := 0; i < 4; i++ { var ty, tx []int // 壁に当たるまでループする for j := 1;; j++ { sy, sx := y + dy[i] * j, x + dx[i] * j if sy >= 0 && sx >= 0 && sy < h && sx < w { if g[sy][sx] == '*' { qy = append(qy, ty...) qx = append(qx, tx...) break } else { ty = append(ty, sy) tx = append(tx, sx) } } else { break } } } // キューを消化 for len(qy) > 0 { sy, sx := qy[0], qx[0] qy, qx = qy[1:], qx[1:] g[sy][sx] = '*' } for i := 0; i < h; i++ { fmt.Println(string(g[i])) } } 前回からキューを活用した実装になりがちですね。 ...

こんにちは、ナナオです。 前回に引き続き競プロを実施していきたいと思います。 今回の問題は以下です。 裏返せる可能性（縦横） | レベルアップ問題集 | プログラミング学習サイト【paizaラーニング】 実装 以下のように実装しました。 （xとyの入力順序を最初間違えていて頭抱えたのは内緒） package main import "fmt" func main(){ var h, w, y, x int fmt.Scan(&h, &w, &y, &x) for i := 0; i < h; i++ { for j := 0; j < w; j++ { if i == y && j == x { fmt.Printf("%s", "!") } else if i == y || j == x { fmt.Printf("%s", "*") } else { fmt.Printf("%s", ".") } } fmt.Println() } } 👍

こんにちは、ナナオです。 前回に引き続き競プロを実施していきたいと思います。 今回の問題は以下です。 陣取りゲーム | レベルアップ問題集 | プログラミング学習サイト【paizaラーニング】 実装 DOTやWALLは定数定義するようにしました。 前回とロジックはそこまで変わっていません。 package main import "fmt" const DOT = -1 const WALL = -2 const A = 0 const B = 500000 func main(){ var h, w int var n string fmt.Scan(&h, &w, &n) var qy, qx []int var ay, ax, by, bx int g := make([][]int, h) for i := 0; i < h; i++ { var s string fmt.Scan(&s) row := make([]int, w) for j, v := range s { switch v { case 'A': row[j] = A ay, ax = i, j case 'B': row[j] = B by, bx = i, j case '#': row[j] = WALL default: row[j] = DOT } } g[i] = row } // 順番を決める if n == "A" { qy = append(qy, ay, by) qx = append(qx, ax, bx) } else { qy = append(qy, by, ay) qx = append(qx, bx, ax) } dy := []int{1, -1, 0, 0} dx := []int{0, 0, 1, -1} for len(qy) > 0 { sy, sx := qy[0], qx[0] qy, qx = qy[1:], qx[1:] c := g[sy][sx] for i := 0; i < 4; i++ { ly, lx := sy + dy[i], sx + dx[i] if ly >= 0 && lx >= 0 && ly < h && lx < w && g[ly][lx] == DOT { g[ly][lx] = c + 1 qy = append(qy, ly) qx = append(qx, lx) } } } var a, b int for i := 0; i < h; i++ { for j := 0; j < w; j++ { if g[i][j] >= B { b++ } else if g[i][j] >= A { a++ } } } fmt.Println(a, b) if a > b { fmt.Println("A") } else { fmt.Println("B") } } 完成しましたが、なんか結構長くなってしまいました。 ...

こんにちは、ナナオです。 前回に引き続き競プロを実施していきたいと思います。 今回の問題は以下です。 陣取りのターン数 | レベルアップ問題集 | プログラミング学習サイト【paizaラーニング】 実装 試行錯誤しながら以下のように実装しました。 1時間くらいかかった。。 ポイントとしては一度g変数でマップを作っている部分ですね。 package main import ( "fmt" ) func main(){ var h, w, n int fmt.Scan(&h, &w, &n) g := make([][]int, h) var qy, qx []int for i := 0; i < h; i++ { var s string fmt.Scan(&s) row := make([]int, w) for j, v := range s { if v == '*' { row[j] = 0 qy = append(qy, i) qx = append(qx, j) } else if v == '#' { row[j] = -2 } else { row[j] = -1 } } g[i] = row } m := map[int]struct{}{} for i := 0; i < n; i++ { var l int fmt.Scan(&l) m[l] = struct{}{} } dy := []int{1, -1, 0, 0} dx := []int{0, 0, 1, -1} for len(qy) > 0 { sy, sx := qy[0], qx[0] qy, qx = qy[1:], qx[1:] c := g[sy][sx] for i := 0; i < 4; i++ { ly, lx := sy + dy[i], sx + dx[i] if ly >= 0 && lx >= 0 && ly < h && lx < w && g[ly][lx] == -1 && g[ly][lx] != -2 { g[ly][lx] = c + 1 qy = append(qy, ly) qx = append(qx, lx) } } } for i := 0; i < h; i++ { for j := 0; j < w; j++ { if g[i][j] == -2 { fmt.Printf("%s", "#") } else if _, ok := m[g[i][j]]; ok { fmt.Printf("%s", "?") } else if g[i][j] >= 0 { fmt.Printf("%s", "*") } else { fmt.Printf("%s", ".") } } fmt.Println() } } 割とすっきりかけている気がするけど、これでもまだ長い気はする…

こんにちは、ナナオです。 前回に引き続き競プロを実施していきたいと思います。 今回の問題は以下です。 幅のある移動 | レベルアップ問題集 | プログラミング学習サイト【paizaラーニング】 実装 ちょっと悩んでAIとも壁打ちした結果、queueという仕組みを使うのが一番効率的なんだと分かりました。 package main import ( "fmt" "strconv" ) func main(){ var h, w int fmt.Scan(&h, &w) s := make([]string, h) replace := func(sy, sx, r int) { runes := []rune(s[sy]) runes[sx] = rune(48 + r) s[sy] = string(runes) } var queueY, queueX []int for i := 0; i < h; i++ { fmt.Scan(&s[i]) for j, v := range s[i] { if v == '*' { replace(i, j, 0) queueY = append(queueY, i) queueX = append(queueX, j) } } } // 移動方向（上下左右） dy := []int{1, -1, 0, 0} dx := []int{0, 0, 1, -1} for len(queueY) > 0 { sy, sx := queueY[0], queueX[0] queueY, queueX = queueY[1:], queueX[1:] r, _ := strconv.Atoi(string(s[sy][sx])) var ly, lx int for i := 0; i < 4; i++ { ly, lx = sy + dy[i], sx + dx[i] if ly >= 0 && lx >= 0 && ly < h && lx < w && s[ly][lx] == '.' { replace(ly, lx, r + 1) queueY = append(queueY, ly) queueX = append(queueX, lx) } } } for i := 0; i < h; i++ { fmt.Println(s[i]) } } queue周りの実装はもう少しいい感じにできた気がしなくもないですが、とりあえずまぁこれで…

こんにちは、ナナオです。 前回に引き続き競プロを実施していきたいと思います。 今回の問題は以下です。 陣取りの結末 | レベルアップ問題集 | プログラミング学習サイト【paizaラーニング】 実装 再帰関数を定義してあげることで自然に実装できました。 package main import "fmt" func main(){ var h, w int fmt.Scan(&h, &w) s := make([]string, h) var sx, sy int for i := 0; i < h; i++ { fmt.Scan(&s[i]) for j, v := range s[i] { if v == '*' { sy, sx = i, j } } } var capture func(sy, sx int) capture = func(sy, sx int) { for i := 0; i < 4; i++ { var ly, lx int switch i { case 0: ly, lx = sy + 1, sx case 1: ly, lx = sy - 1, sx case 2: ly, lx = sy, sx + 1 case 3: ly, lx = sy, sx - 1 } if ly >= 0 && lx >= 0 && ly < h && lx < w && s[ly][lx] != '*' && s[ly][lx] != '#' { runes := []rune(s[ly]) runes[lx] = '*' s[ly] = string(runes) capture(ly, lx) } } } capture(sy, sx) for i := 0; i < h; i++ { fmt.Println(s[i]) } } 再帰関数を定義するためにcapture関数の定義をわざわざ二つ書くのがちょっと面倒でした。とはいえGoの言語仕様上しょうがない。。