競技プログラミング

こんにちは、ナナオです。 前回に引き続き競プロを実施していきたいと思います。 今回の問題は以下です。 幅のある移動 | レベルアップ問題集 | プログラミング学習サイト【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の言語仕様上しょうがない。。

こんにちは、ナナオです。 前回に引き続き競プロを実施していきたいと思います。 今回の問題は以下です。 1マスの陣取り2 | レベルアップ問題集 | プログラミング学習サイト【paizaラーニング】 実装 以下のように実装しました。 package main import "fmt" func main(){ var h, w int fmt.Scan(&h, &w) s := make([]string, h) var sy, sx int for i := 0; i < h; i++ { fmt.Scan(&s[i]) for j, v := range s[i] { if v == '*' { sy = i sx = j } } } 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] != '#' { runes := []rune(s[ly]) runes[lx] = '*' s[ly] = string(runes) } } for i := 0; i < h; i++ { fmt.Println(s[i]) } } やっぱりこの形で実装するのが一番効率的な気がするんだよなぁ…

こんにちは、ナナオです。 前回に引き続き競プロを実施していきたいと思います。 今回の問題は以下です。 1 マスの陣取り | レベルアップ問題集 | プログラミング学習サイト【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 = i sx = j } } } 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 < h && lx < w && ly >= 0 && lx >= 0 { runes := []rune(s[ly]) runes[lx] = '*' s[ly] = string(runes) } } for i := 0; i < h; i++ { fmt.Println(s[i]) } }gf 正攻法ではないかもしれませんが、ループ回数はかなり抑えられているのでいいんじゃないでしょうか。 ...

こんにちは、ナナオです。 前回に引き続き競プロを実施していきたいと思います。 今回の問題は以下です。 へび | レベルアップ問題集 | プログラミング学習サイト【paizaラーニング】 ついにボス…！ 実装 以下のように実装しました。 package main import "fmt" func main(){ var h, w, sy, sx, n int fmt.Scan(&h, &w, &sy, &sx, &n) s := make([]string, h) for i := 0; i < h; i++ { fmt.Scan(&s[i]) } dir := []string{"N", "E", "S", "W"} ct := 0 cd := 120 // 初期の座標に印をつける runes := []rune(s[sy]) runes[sx] = '*' s[sy] = string(runes) for i := 0; i < n; i++ { var t int var d string fmt.Scan(&t, &d) lr := 1 if d == "L" { lr = -1 } result := true l := t - ct for j := 0; j < l; j++ { switch dir[cd%4] { case "N": sy-- case "S": sy++ case "E": sx++ case "W": sx-- } if sy >= h || sx >= w || sy < 0 || sx < 0 || s[sy][sx] == '#' || s[sy][sx] == '*' { result = false break } else { runes := []rune(s[sy]) runes[sx] = '*' s[sy] = string(runes) } } ct = t cd += lr if !result { break } } l := 100 - ct for i := 0; i < l; i++ { switch dir[cd%4] { case "N": sy-- case "S": sy++ case "E": sx++ case "W": sx-- } if sy >= h || sx >= w || sy < 0 || sx < 0 || s[sy][sx] == '#' || s[sy][sx] == '*' { break } else { runes := []rune(s[sy]) runes[sx] = '*' s[sy] = string(runes) } } for i := 0; i < h; i++ { fmt.Println(s[i]) } } 実装はできましたが、ちょっと重複している部分があるので修正します。 ...

こんにちは、ナナオです。 前回に引き続き競プロを実施していきたいと思います。 今回の問題は以下です。 時刻に伴う移動 | レベルアップ問題集 | プログラミング学習サイト【paizaラーニング】 実装 結構躓きましたが、最終的にこのように実装しました。 package main import "fmt" func main(){ var h, w, sy, sx, n int fmt.Scan(&h, &w, &sy, &sx, &n) s := make([]string, h) for i := 0; i < h; i++ { fmt.Scan(&s[i]) } // 現在の時間 ct := 0 // 現在の方向 cd := 120 dir := []string{"N", "E", "S", "W"} for i := 0; i < n; i++ { var t int var d string fmt.Scan(&t, &d) lr := 1 if d == "L" { lr = -1 } l := t - ct // fmt.Println("l: ", l) for j := 0; j < l; j++ { switch dir[cd%4] { case "N": sy-- case "S": sy++ case "E": sx++ case "W": sx-- } if sy >= h || sx >= w || sy < 0 || sx < 0 || s[sy][sx] == '#' { fmt.Println("Stop") return } else { fmt.Println(sy, sx) } } ct = t cd += lr } } 十分すっきり実装できていますね👍 ...

こんにちは、ナナオです。 前回に引き続き競プロを実施していきたいと思います。 今回の問題は以下です。 幅のある移動 | レベルアップ問題集 | プログラミング学習サイト【paizaラーニング】 実装 前回の実装から少し変わりました。 文字列置き換えの処理はちょっとGoっぽいですね。 （Goの文字列はイミュータブル） package main import "fmt" func main(){ var h, w, sy, sx, n int fmt.Scan(&h, &w, &sy, &sx, &n) s := make([]string, h) for i := 0; i < h; i++ { fmt.Scan(&s[i]) } dir := []string{"N", "E", "S", "W"} c := 120 result := false // 初期の座標に印をつける runes := []rune(s[sy]) runes[sx] = '*' s[sy] = string(runes) for i := 0; i < n; i++ { var d string var l int fmt.Scan(&d, &l) lr := 1 if d == "L" { lr = -1 } for j := 0; j < l; j++ { switch dir[c%4] { case "N": sx += lr case "S": sx -= lr case "E": sy += lr case "W": sy -= lr } if sy >= h || sx >= w || sy < 0 || sx < 0 || s[sy][sx] == '#' { result = true break } else { runes := []rune(s[sy]) runes[sx] = '*' s[sy] = string(runes) } } c += lr if result { break } } for i := 0; i < h; i++ { fmt.Println(s[i]) } } 一応実装はできたんですが、なんかresultのあたりはもうちょっときれいにできそうな気がする。。 ...

こんにちは、ナナオです。 前回に引き続き競プロを実施していきたいと思います。 今回の問題は以下です。 移動が可能かの判定・幅のある移動 | レベルアップ問題集 | プログラミング学習サイト【paizaラーニング】 実装 コアとなるロジックはあまり変わらないのですが、途中で壁があったり壁を超えたりする場合はそれを超える手前の座標を出力したうえでStopと出力しなければいけないようです。 ちょっと大変でしたが実装しました。 package main import "fmt" func main(){ var h, w, sy, sx, n int fmt.Scan(&h, &w, &sy, &sx, &n) s := make([]string, h) for i := 0; i < h; i++ { fmt.Scan(&s[i]) } dir := []string{"N", "E", "S", "W"} c := 120 // 現在の向き for i := 0; i < n; i++ { var d string var l int fmt.Scan(&d, &l) lr := 1 if d == "L" { lr = -1 } move := l * lr // 移動量 // X軸上に壁があるかどうか判定 checkWallX := func(y, fromX, toX, lr int) bool { i := fromX for { i += lr if i == toX { fmt.Println(y, toX) return true } if i >= w || i < 0 || s[sy][i] == '#' { fmt.Println(y, i - lr) fmt.Println("Stop") return false } } } // Y軸上に壁があるかどうか判定 checkWallY := func(x, fromY, toY, lr int) bool { i := fromY for { i += lr if i == toY { fmt.Println(toY, x) return true } if i >= h || i < 0 || s[i][sx] == '#' { fmt.Println(i - lr, x) fmt.Println("Stop") return false } } } // チェック結果を格納 result := true switch dir[c%4] { case "N": result = checkWallX(sy, sx, sx + move, lr) sx += move case "S": result = checkWallX(sy, sx, sx - move, -lr) sx -= move case "E": result = checkWallY(sx, sy, sy + move, lr) sy += move case "W": result = checkWallY(sx, sy, sy - move, -lr) sy -= move } c += lr if !result { return } } } ポイントはcheckWall関数です。 ...

こんにちは、ナナオです。 前回に引き続き競プロを実施していきたいと思います。 今回の問題は以下です。 移動が可能かの判定・方向 | レベルアップ問題集 | プログラミング学習サイト【paizaラーニング】 実装 実装しました。 ポイントとしてはlr変数を使って向きと座標の管理をしている点くらいですかね。 package main import "fmt" func main(){ var h, w, sy, sx, n int fmt.Scan(&h, &w, &sy, &sx, &n) s := make([]string, h) for i := 0; i < h; i++ { fmt.Scan(&s[i]) } dir := []string{"N", "E", "S", "W"} c := 120 // 現在の向き for i := 0; i < n; i++ { var d string fmt.Scan(&d) lr := 1 if d == "L" { lr = -1 } switch dir[c%4] { case "N": sx += lr case "S": sx -= lr case "E": sy += lr case "W": sy -= lr } c += lr if sx >= w || sy >= h || 0 > sx || 0 > sy || s[sy][sx] == '#' { fmt.Println("Stop") break } else { fmt.Println(sy, sx) } } } 👍

こんにちは、ナナオです。 前回に引き続き競プロを実施していきたいと思います。 今回の問題は以下です。 移動が可能かの判定・方向 | レベルアップ問題集 | プログラミング学習サイト【paizaラーニング】 実装 以下のように実装しました！ package main import "fmt" func main(){ var h, w, sy, sx int var d, m string fmt.Scan(&h, &w, &sy, &sx, &d, &m) s := make([]string, h) for i := 0; i < h; i++ { fmt.Scan(&s[i]) } dir := []string{"N", "E", "S", "W"} var count int switch d { case "N": count = 4 case "E": count = 5 case "S": count = 6 case "W": count = 7 } if m == "R" { count++ } else { count-- } switch dir[count%4] { case "N": sy-- case "S": sy++ case "E": sx++ case "W": sx-- } if sy >= h || sx >= w || sy < 0 || sx < 0 || s[sy][sx] == '#' { fmt.Println("No") } else { fmt.Println("Yes") } } でもなんかスリムじゃないですよね。 ...