Goで競技プログラミング 1

はじめに

Go言語を知るきっかけとするために、競技プログラミングをGo言語で記載しています。
今回は、Goで書くときに使用しているコードのテンプレートをGistに置いているので紹介します。

環境

Go 1.6

補足

paizaではGoのバージョンが少し高いので、sort.StableSort が使用できるのですが、
Atcoderでは1.6のためsort.StableSort は使用できなかったです。
なので、コメントアウトしています。

コード

https://gist.github.com/katsuobushiFPGA/8a5ceac12bb7db5d1c38cb267a5dffc2

コードについての補足

readLine

標準入力から1行読み取りをする。
Revisions を見るとバレてしまうのですが、初めの方は bufio.NewScanner を使用していました。
bufio.NewScanner を使用していたときは、1行を読み取るサイズがデフォルトのままだったので、
下記に記載の通り、 MaxScanTokenSize までしか読み取れていませんでした。
それに気づかずに、 Atcoder Beginner Contest 148 で試行錯誤してた際にWAしてました。。

bufio.NewScanner を使用するときには、 Buffer を呼んでバッファサイズを変更すれば良さそうです。
https://golang.org/pkg/bufio/

Gistのコードでは、下記のような感じでグローバルに reader と writer に分けて宣言し、最大バッファのサイズを指定してます。

const (
    initialBufSize = 10000
    maxBufSize = 1000000
)
var reader = bufio.NewReaderSize(os.Stdin, maxBufSize)
var writer = bufio.NewWriter(os.Stdout)

2020/01/13 追記 上記の書き方だと、1行のサイズがおもすぎて配列に展開する際にメモリ消費量がやばかったので、やめました。
Atcoder Beginner Contest 151 にて TLEしたので（本当は map 使いまくったせいなのですが)、ついでに直しました。
今は、readInt, readString, readRunes みたいなヘルパー関数用意して、word ごとに読み取ってます。
メモリ効率良さげなので

s2i s2f i2s

strconv.Atoi を書くのが面倒なので短い名前の関数にラップしているだけです。
s2i: string -> int
s2f: string -> float64
i2s: int -> string

strAry2intAry intAry2strAry

こちらも同じ系統です。
よく使う用途は、

r = readLine()
is := strAry2intAry(split(r))

のような感じで、1行読み込みのときに、全部 string で返ってくるのを
[]stringに変換してこれを []int に変換します。

ary2str

期待の出力する際に、 []string -> string するときがあるのでそのため用

reverse sum

あまり使わないけどあったらたまに使う程度
名前の通りの動作をしてくれます。

stableSortExample

安定ソートがほしいときに参考にして使います。
構造体の配列をソートする際に、第1ソート, 第2ソートみたいな感じでソートできるので良いです。

おわりに

paiza の方は長らく放置していたので 2019年11月頃からまたはじめました。 (Bランク完答を目指している途中です)
Atcoderの方は登録したままでコンテストに一切出ていなかったので、 ABC 148から出ております。
2020年内に Atcoderの方はレート1000は行きたいです。