こんにちは、ナナオです。 前回に引き続き競プロを実施していきたいと思います。 今回の問題は以下です。 ABC081B - Shift only 実装 以下のように実装しました。 use proconio::input; fn main() { input! { n: u8, mut a: [u32; n], } let mut cnt = 0; loop { if !a.iter().find(|&&a| a % 2 != 0).is_some() { a = a.into_iter().map(|a| a / 2).collect(); } else { break; } cnt += 1; } println!("{cnt}"); } まぁまぁすっきり実装できたんじゃないでしょうか。 では今回はこれで👍
こんにちは、ナナオです。 前回に引き続き競プロを実施していきたいと思います。 今回の問題は以下です。 ABC081A - Placing Marbles 実装 以下のようにすっきり実装できました。 use proconio::{input, marker::Chars}; fn main() { input! { s: Chars, } let cnt = s.into_iter().filter(|c| c == &'1').count(); println!("{cnt}"); } proconio、マジで最高すぎる。。。 では今回はこれで👍
こんにちは、ナナオです。 前回に引き続き競プロを実施していきたいと思います。 今回の問題は以下です。 ABC086A - Product 実装 前回は入力に手間取りましたが、今回はproconioというライブラリを使ってかなりすっきり書くことができました。 use proconio::input; fn main() { input! { a: i32, b: i32, } if a * b % 2 == 0 { println!("Even"); } else { println!("Odd"); } } いやぁ、標準入力がすっきり書けるだけでここまですっきり書けるようになるとは… 驚きです。 ということで今回はこれで👍
こんにちは、ナナオです。 前回に引き続き競プロを実施していきたいと思います。 今回の問題は以下です。 PracticeA - Welcome to AtCoder 実装 以下のように実装しました。 今回はRustを使います。 （前回の記事であまりにもGoが遅かったので。。。） fn main() { let a = read().trim().parse::<i32>().unwrap(); let t = read(); let mut iter = t.split_whitespace().map(|i| i.parse::<i32>().unwrap()); let b = iter.next().unwrap(); let c = iter.next().unwrap(); let s = read(); println!("{} {s}", a + b + c) } fn read() -> String { let mut s = String::new(); std::io::stdin().read_line(&mut s).unwrap(); s.clone() } 文字列受け取って数値に変換するだけで手間かかりすぎでしたが、Rustの強力な型安全性やエラーハンドリングを考えると納得…ではあるんですが、やっぱり競技プログラミングには向いていない気がしますね。 というわけで今回はこれで👍
こんにちは、ナナオです。 前回に引き続き競プロを実施していきたいと思います。 今回は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) } } 前回のコードを少し弄れば解決しましたね。 ...