Rustで標準入力受け取りたいとき

こんにちは。ナナオです。

AtCoderでRustが使えるので使いたいなーと思ったんですが、標準入力どう受け取ればいいんだ？ってなったので備忘録。

Rustで標準入力受け取るコード

こんな感じにしました。

use std::io::{self, Read};

fn main() {
    let input = read_stdin();
    print!("{}", input);
}

/// 標準入力を受け取る関数
fn read_stdin() -> String {
    let mut buf = String::new();
    io::stdin().read_to_string(&mut buf).expect("Failed to read stdin.");
    buf.to_string()
}

これで大抵の競プロは乗り切れるはず。。