scalaで競技プログラミングを入門する方法　～標準入力～

この記事では競技プログラミングでscala使いたいけど、標準入力の仕方がわからない方向けの記事となります。

すでにこの手の記事はありますが、いろいろな問題に適応できなかったため、執筆する運びになりました。

scalaの標準入力とJavaの標準入力

scalaでは通常、標準入力を

io.StdIn.readLine

などで行うことが出来ます。
しかし、これはあまり使いやすくなく Javaの標準入力のほうが競技プログラミングでは書きやすいです。
なのでJavaの標準入力を使うために

val sc = new java.util.Scanner(System.in)

でインスタンス化しておきます。

それでは、問題別に標準入力の方法を紹介していきます。

Int型

このような問題では、Int型の整数を一つ入力します。
やり方は

var r = sc.nextInt()

このようになります。

https://atcoder.jp/contests/abc164/tasks/abc164_a
こちらの問題では、Int型の整数を複数個入力します。
やり方は

var s , w = sc.nextInt()

とcomma(,)を挟むことによってスペースで区切られた整数を一つずつ宣言して入力できます。

String型

https://atcoder.jp/contests/abc166/tasks/abc166_a
この問題では、String型の文字列を一つ入力します。
やり方は

var s = sc.next()

このようになります。

問題は見つけられませんでしたが文字列複数個のときもIntのときと同様に

var s , w = sc.next()

とするとスペースで自動で区切られ２つの文字列を入力できます。

Array型

この問題では配列を使って複数個、整数を入力します
やり方は

var a = Array.fill(n)(sc.nextInt)

このように、fillを使って埋めます１つ目のカッコに要素数、２つ目に先程の標準入力を書きます

こちらは二次元配列を使う問題です
残念ながら二次元配列ではあまりスマートに書くことが出来ません

var a = Array.ofDim[Int](3, 3)
for (i <- 0 until 3){
  a(i) = Array.fill(3)(sc.nextInt())
}

このようにfor文を使います。
はじめにArray.ofDim[Int](3, 3)を使って要素数3,3の二次元配列を作ります。
その後、forを回し、先程の一次元配列の要領で入力させます。

scalaでのList

Listはあまり計算速度が早くありません。
例えばfillを使って埋めるときなど、配列の２倍ほどの速度がかかってしまいます。
なのでなるべくListではなくArrayを使っていきましょう。

終わりに

現在atcoderではscalaで書いている人があまり多くはありませんしかし、scalaはpythonなどと比べ早い言語であり、C++などに比べ標準関数が豊富です。
scalaの関数とJavaの関数が使えることで、できることの幅が広がり、よりスピーディーによりスタイリッシュにコーディングすることが出来ます。

みんなもscalaやろうぜ！！！