2018/08/30

競プロで標準入力のパースがめんどうだったのでマクロ作りました。
AtCoderのRust(1.15.1)でテスト済みです。

ソースコード

https://github.com/kgtkr/procon-lib-rs/blob/master/src/parser.rs

使い方

事前準備

標準入力を全て読み込む

let text = {
    let mut s = String::new();
    io::stdin().read_to_string(&mut s).unwrap();
    s
};

例

入力値

5
10 20
1 a
2 b
3 c
4 d
5 e
1 2 3 4 5

input!(text=>
    (n:usize)
    (a:i64 b:i64)
    {n;list:(i64,#)}
    (arr:[i64])
);
assert_eq!(n, 5);
assert_eq!(a, 10);
assert_eq!(b, 20);
assert_eq!(list, vec![(1,"a"),(2,"b"),(3,"c"),(4,"d"),(5,"e")]);
assert_eq!(arr, vec![1,2,3,4,5]);

解説

input!(入力値=>パーサー...);

1行の入力

(変数名:型...)

複数行の入力

{行数;変数名:型}

型

構文	解説	例	例入力
型	単一の型とマッチ(parse::<型>()でパースされる)	i64	10
#	文字列	#	abc
@	1から始まるインデックスを0からに変換するやつ。つまりusizeでパースして-1してるだけ	@	1
[型]	配列	[i64]	1 2 3 4 5
(型,...)	タプル	(i64,#)	10 abc