競技プログラミング

こんにちは、ナナオです。 前回に引き続き競プロを実施していきたいと思います。 今回の問題は以下です。 累積和の計算 | レベルアップ問題集 | プログラミング学習サイト【paizaラーニング】 実装 以下のように実装しました。 package main import "fmt" func main(){ var n, m int fmt.Scan(&n, &m) x := 0 a := make([]int, n) s := make([]int, n) for i := 0; i < n; i++ { fmt.Scan(&a[i]) x += a[i] s[i] = x } if x < m { fmt.Println(-1) return } z := 100000 for l := 0; l < n; l++ { for r := n - 1; r >= 0; r-- { sum := s[r] - s[l] + a[l] if sum >= m && z > r - l + 1 { z = r - l + 1 } } } fmt.Println(z) } ですが、これだと計算量が多く実行時間エラーになってしまいました。 ということで、AIと壁打ちして「しゃくとり法」で実装しました。 package main import "fmt" func main(){ var n, m int fmt.Scan(&n, &m) a := make([]int, n) for i := 0; i < n; i++ { fmt.Scan(&a[i]) } z := n + 1 sum := 0 l := 0 for r := 0; r < n; r++ { sum += a[r] for sum >= m { c := r - l + 1 if z > c { z = c } sum -= a[l] l++ } } if z > n { fmt.Println(-1) } else { fmt.Println(z) } } これで計算量もぐっと減ってきれいに実装できました。 ...

こんにちは、ナナオです。 前回に引き続き競プロを実施していきたいと思います。 今回の問題は以下です。 累積和の計算 | レベルアップ問題集 | プログラミング学習サイト【paizaラーニング】 実装 以下のように実装しました。 package main import "fmt" func main(){ var n, m int fmt.Scan(&n) s := make([]int, n) for i := 0; i < n; i++ { var a int fmt.Scan(&a) if i > 0 { s[i] = s[i - 1] + a } else { s[i] = a } } fmt.Scan(&m) for i := 0; i < m; i++ { var l, r int fmt.Scan(&l, &r) if l > 1 { fmt.Println(s[r] - s[l - 1]) } else { fmt.Println(s[r]) } } } ですが、これだとテストがすべて動作しませんでした。 解説コードを見て修正します。 package main import "fmt" func main(){ var n, m int fmt.Scan(&n) a := make([]int, n) s := make([]int, n) x := 0 for i := 0; i < n; i++ { fmt.Scan(&a[i]) x += a[i] s[i] = x } fmt.Scan(&m) for i := 0; i < m; i++ { var l, r int fmt.Scan(&l, &r) fmt.Println(s[r] - s[l] + a[l]) } } これで動くようになりました👍

こんにちは、ナナオです。 前回に引き続き競プロを実施していきたいと思います。 今回の問題は以下です。 累積和の計算 | レベルアップ問題集 | プログラミング学習サイト【paizaラーニング】 実装 以下のように実装しました。 package main import "fmt" func main(){ var n int fmt.Scan(&n) s := 0 for i := 0; i < n; i++ { var num int fmt.Scan(&num) s += num fmt.Println(s) } } これは簡単に実装できましたね👍 ではまた。

こんにちは、ナナオです。 前回に引き続き競プロを実施していきたいと思います。 今回の問題は以下です。 いびつなリバーシ対戦 | レベルアップ問題集 | プログラミング学習サイト【paizaラーニング】 実装 以下のように実装しました。 n変数使ってない… package main import ( "fmt" "strconv" ) func main(){ var h, w, n, m int fmt.Scan(&h, &w, &n, &m) g := make([][]rune, h) for i := 0; i < h; i++ { var s string fmt.Scan(&s) g[i] = []rune(s) } for i := 0; i < m; i++ { var p, sy, sx int fmt.Scan(&p, &sy, &sx) pr := []rune(strconv.Itoa(p))[0] g[sy][sx] = pr for dy := -1; dy <= 1; dy++ { for dx := -1; dx <= 1; dx++ { for j := 1;; j++ { ly, lx := sy + dy * j, sx + dx * j if ly < 0 || lx < 0 || ly >= h || lx >= w || g[ly][lx] == '#' { break } if g[ly][lx] == pr { for k := j - 1; k > 0; k-- { ny, nx := sy + dy * k, sx + dx * k g[ny][nx] = pr } break } } } } } for i := 0; i < h; i++ { fmt.Println(string(g[i])) } } n、何のために使うんだ…？と思って解説コード見たら、そっちでも使ってなかったです（使わないんかい） ...

こんにちは、ナナオです。 前回に引き続き競プロを実施していきたいと思います。 今回の問題は以下です。 いびつなリバーシ対戦（２人） | レベルアップ問題集 | プログラミング学習サイト【paizaラーニング】 実装 以下のように実装しました。 実装するたびにどんどんネストが深くなっている… package main import "fmt" func main(){ var h, w, n int fmt.Scan(&h, &w, &n) g := make([][]rune, h) for i := 0; i < h; i++ { var s string fmt.Scan(&s) g[i] = []rune(s) } for i := 0; i < n; i++ { for _, v := range []rune{'A', 'B'} { var sy, sx int fmt.Scan(&sy, &sx) g[sy][sx] = v for dy := -1; dy <= 1; dy++ { for dx := -1; dx <= 1; dx++ { for j := 1;; j++ { ly, lx := sy + dy * j, sx + dx * j if ly < 0 || lx < 0 || ly >= h || lx >= w || g[ly][lx] == '#' { break } if g[ly][lx] == v { for k := j - 1; k > 0; k-- { ny, nx := sy + dy * k, sx + dx * k g[ny][nx] = v } break } } } } } } for i := 0; i < h; i++ { fmt.Println(string(g[i])) } } もちろん関数化すれば可読性が増す実装はできるでしょう。 ...

こんにちは、ナナオです。 前回に引き続き競プロを実施していきたいと思います。 今回の問題は以下です。 いびつなひとりリバーシ | レベルアップ問題集 | プログラミング学習サイト【paizaラーニング】 実装 以下のように実装しました。 package main import "fmt" func main(){ var h, w, n int fmt.Scan(&h, &w, &n) g := make([][]rune, h) for i := 0; i < h; i++ { var s string fmt.Scan(&s) g[i] = []rune(s) } for i := 0; i < n; i++ { var sy, sx int fmt.Scan(&sy, &sx) g[sy][sx] = '*' for dy := -1; dy <= 1; dy++ { for dx := -1; dx <= 1; dx++ { for j := 1;; j++ { ly, lx := sy + dy * j, sx + dx * j if ly < 0 || lx < 0 || ly >= h || lx >= w || g[ly][lx] == '#' { break } if g[ly][lx] == '*' { for k := j - 1; k > 0; k-- { ny, nx := sy + dy * k, sx + dx * k g[ny][nx] = '*' } 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) 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])) } } 割とすっきり書けたと思います👍 ...