競技プログラミング

こんにちは、ナナオです。 前回に引き続き競プロを実施していきたいと思います。 今回の問題は以下です。 規則的な数列の和 | レベルアップ問題集 | プログラミング学習サイト【paizaラーニング】 実装 最初は以下のように実装しました。 package main import "fmt" func main(){ var n, k int fmt.Scan(&n, &k) m := []int{1, 0, -1} c := n - 1 sum := 0 for c < k { sum += m[c % 3] c++ } fmt.Println(sum) } しかし、うまくいきませんでした。。 実行時間エラーになってしまったため、高速化が必要です。 解法としては、法則性を考えれば以下のようにN(1)で実装できます。 package main import "fmt" func main(){ var n, k int fmt.Scan(&n, &k) result := 0 switch n % 3 { case 1: if k % 3 >= 1 { result = 1 } case 2: if k % 3 == 0 { result = -1 } case 0: if k % 3 == 0 { result = -1 } } fmt.Println(result) } これですっきり実装できました👍

こんにちは、ナナオです。 前回に引き続き競プロを実施していきたいと思います。 今回の問題は以下です。 最長の区間 | レベルアップ問題集 | プログラミング学習サイト【paizaラーニング】 実装 以下のように実装しました。 package main import "fmt" func main(){ var n, k int fmt.Scan(&n, &k) a := make([]int, n) for i := 0; i < n; i++ { fmt.Scan(&a[i]) } x := 1 c := 1 for i := 0; i < n; i++ { x *= a[i] if x == 0 { c = 1 x = 1 } else if x >= k { fmt.Println(c) return } c++ } } ですがこれだと攻略はできませんでした。。 回答を見ながら修正しました。 package main import "fmt" func main(){ var n, k int fmt.Scan(&n, &k) a := make([]int, n) for i := 0; i < n; i++ { fmt.Scan(&a[i]) } start := 0 end := 0 count := 1000000 mul := func(data []int) int { result := 1 for _, v := range data { result *= v } return result } for end < n && start <= end { if mul(a[start: end + 1]) >= k { if end - start + 1 < count { count = end - start + 1 } start += 1 } else if a[end] == 0 { start = end + 1 end += 1 } else { end += 1 } } fmt.Println(count) } ですがいまいち理解できていません。クソー ...

こんにちは、ナナオです。 前回に引き続き競プロを実施していきたいと思います。 今回の問題は以下です。 最長の区間 | レベルアップ問題集 | プログラミング学習サイト【paizaラーニング】 実装 まずは以下のように実装しました。 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]) } s := make([]int, n) for i := 0; i < m; i++ { var mn, mx, t int fmt.Scan(&mn, &mx, &t) for k := mn - 1; k < mx; k++ { s[k] += } } for i := 0; i < n; i++ { fmt.Println(a[i] + s[i]) } この回答でもGoの場合解けるのですが、まぁ実行時間制限ギリギリなので、一応模範解答のように改修しておきましょう。 この実装、よく考えると累積和で解けるんですよね。 つまり、mnとmxは累積和の区間として考えることができるわけです。 実装としては以下の通りです。 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]) } s := make([]int, n) for i := 0; i < m; i++ { var mn, mx, t int fmt.Scan(&mn, &mx, &t) s[mn - 1] += t if mx < len(s) { s[mx] -= t } } x := 0 for i := 0; i < n; i++ { x += s[i] fmt.Println(a[i] + x) } } あーすっきりした。 ...

こんにちは、ナナオです。 前回に引き続き競プロを実施していきたいと思います。 今回の問題は以下です。 最長の区間 | レベルアップ問題集 | プログラミング学習サイト【paizaラーニング】 実装 以下のように実装しました。 package main import "fmt" func main(){ var n, m int fmt.Scan(&n, &m) a := make([]int, n) x := 0 s := make([]int, n) for i := 0; i < n; i++ { fmt.Scan(&a[i]) x += a[i] s[i] = x } sum := 0 d := 0 l := 0 for r := 0; r < n; r++ { sum += a[r] for sum > m { sum -= a[l] l++ } c := r - l + 1 if d < c { d = c } } fmt.Println(d) } なんででしょう、しゃくとり法がなぜか直感的にちゃんと理解できていない… もっとAIと壁打ちして、ちゃんと理解できるようにしないと…！！ ということで今回はこれで👍

こんにちは、ナナオです。 前回に引き続き競プロを実施していきたいと思います。 今回の問題は以下です。 累積和の計算 | レベルアップ問題集 | プログラミング学習サイト【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])) } } ネストはさらに深くなりましたが、書きやすかったです。 ...