・4 min read
Rust✖️競プロ 覚え書きまとめ
はじめに
Rust と AtCoder を勉強するうちに「これよく使うな」と思った処理をまとめていきます！
随時更新していきます！
同じ要素の数を数える（Counter）
use std::collections::HashMap;
let mut counter: HashMap<_type_, usize> = HashMap::new();
for _value_ in _target_ {
let count = counter.entry(_value_).or_insert(0);
*count += 1;
}Entry が空の場合は default: V を挿入します。その後 Entry の値への参照を返します。
// 使用例
// 文字列に含まれるアルファベットがそれぞれいくつあるか数える
use std::collections::HashMap;
let target = "sasakisaku".chars();
let mut counter: HashMap<char, usize> = HashMap::new();
for value in target {
let count = counter.entry(value).or_insert(0);
*count += 1;
}
println!("{:?}", counter);{'k': 2, 'u': 1, 's': 3, 'a': 3, 'i': 1}
アルファベットを列挙する
AtCoder の環境では、
'a'..'c' のようにアルファベットの Range を作ることが出来ません。error[E0277]: the trait bound `char: std::iter::Step` is not satisfied
--> src/main.rs:2:12
|
2 | for c in 'a'..='d' {
| ^^^^^^^^^ the trait `std::iter::Step` is not implemented for `char`
|
= note: required because of the requirements on the impl of `std::iter::Iterator` for `std::ops::RangeInclusive<char>`そのため、以下のようにすると良いです。
// a ~ d を列挙
for c in (b'a'..=b'd').map(char::from) {
println!("{}", c);
};a
b
c
d
同じ値で埋めた Vec を作る
let x = vec![0; 5];
println!("{:?}", x);[0, 0, 0, 0, 0]
型の最大値・最小値で初期化する
std::T::MIN, std::T::MAX のようにします。let min_isize = std::isize::MIN;
let max_isize = std::isize::MAX;
println!("isize: {} ~ {}", min_isize, max_isize);isize: -9223372036854775808 ~ 9223372036854775807
符号なし整数の引き算で桁溢れを回避する
符号なし整数の最小値は当然 なので、引き算の結果負の数になるとエラーが発生します。
1u8 - 5u8Errorthis arithmetic operation will overflowattempt to compute `1_u8 - 5_u8`, which would overflowthis arithmetic operation will overflowprintln!("{}", 1u8.saturating_sub(5u8));0
Vec<f64>, Vec<f32> をソートする
Vec<f64> や Vec<f32> をソートしようとすると、以下のようにエラーが発生します。let mut a: Vec<f64> = Vec::from([0.3, 0.1, 0.8]);
a.sort();Errorthe trait bound `f64: Ord` is not satisfieda.sort(); ^^^^ the trait `Ord` is not implemented for `f64`the trait bound `f64: Ord` is not satisfiedこうしましょう。
a.sort_by(|x, y| x.partial_cmp(y).unwrap());
println!("{:?}", a);[0.1, 0.3, 0.8]
お手軽素数判定
for n in 500..600 {
if !(2..n).take_while(|i| i * i <= n).any(|i| n % i == 0) {
println!("{}", n);
}
};503
509
521
523
541
547
557
563
569
571
577
587
593
599
