AtCoder Beginners Selectionの実施記録(Rust)

概要

表題のプログラミング言語でAtCoder Beginners Selectionを実施記録した際の記録です
基本文法の学習が目的です

https://atcoder.jp/contests/abs
AtCoder Beginners Selectionとは？
このコンテストは、「AtCoderに登録したけど何をしていいか分からない・・・！」という人に向けて作られた、初心者向け問題集です。
問題は、dr.kenさんの、「AtCoder に登録したら次にやること ～ これだけ解けば十分闘える！過去問精選 10 問 ～」から選出しています。

基本的には以下ができれば大体の問題を解けるようになっている。

• 標準入力
• 標準出力
• 四則演算
• 条件分岐
• 文字列操作
• ループ
• 配列の操作
• 全通りの探索
• 組み込み関数等の利用
• ソート
• 集合
• 二次元配列の操作

PracticeA - Welcome to AtCoder

PracticeA - Welcome to AtCoder

問題文
高橋君はデータの加工が行いたいです。
整数 a,b,cと、文字列 s が与えられます。 a+b+c の計算結果と、文字列 s を並べて表示しなさい。

制約
1≤a,b,c≤1,000
1≤∣s∣≤100

入力
入力は以下の形式で与えられる。
a
b c
s

出力
a+b+c と s を空白区切りで 1 行に出力せよ。

基本的な標準入力、標準出力、四則演算ができればOK

main.rust

use std::io;

fn main() {
    // 整数a、b、cを読み取る
    let mut a = String::new();
    io::stdin().read_line(&mut a).expect("failed to read a");
    let a: i32 = a.trim().parse().expect("failed to parse a");

    let mut b_c = String::new();
    io::stdin().read_line(&mut b_c).expect("failed to read b and c");
    let mut b_c = b_c.split_whitespace();
    let b: i32 = b_c.next().unwrap().parse().expect("failed to parse b");
    let c: i32 = b_c.next().unwrap().parse().expect("failed to parse c");

    // 文字列sを読み取る
    let mut s = String::new();
    io::stdin().read_line(&mut s).expect("failed to read s");
    let s = s.trim();

    // a+b+cとsを空白で区切って1行に出力する
    println!("{} {}", a + b + c, s);
}

ABC086A - Product

ABC086A - Product

問題文
シカのAtCoDeerくんは二つの正整数 a,b を見つけました。 a と b の積が偶数か奇数か判定してください。

制約
1 ≤ a,b ≤ 10000
a,b は整数

入力
入力は以下の形式で標準入力から与えられる。
a b

出力
積が奇数なら Odd と、 偶数なら Even と出力せよ。

基本的な標準入力、標準出力、四則演算、条件分岐ができればOK

main.rust

fn main() {
    let mut input = String::new();
    std::io::stdin().read_line(&mut input).unwrap();
    let mut iter = input.split_whitespace();
    let a: i32 = iter.next().unwrap().parse().unwrap();
    let b: i32 = iter.next().unwrap().parse().unwrap();

    if (a * b) % 2 == 0 {
        println!("Even");
    } else {
        println!("Odd");
    }
}

ABC081A - Placing Marbles

ABC081A - Placing Marbles

問題文
すぬけ君は 1,2,3 の番号がついた 3 つのマスからなるマス目を持っています。各マスには 0 か 1 が書かれており、マス i には si が書かれています。
すぬけ君は 1 が書かれたマスにビー玉を置きます。 ビー玉が置かれるマスがいくつあるか求めてください。

制約
s1 ,s2 ,s3 は 1 あるいは 0

入力
入力は以下の形式で標準入力から与えられる。
s1s2s3

出力
答えを出力せよ。

入力例 1
101

出力例 1
2

基本的な標準入力、標準出力、文字列操作ができればOK

main.rust

fn main() {
    let s = read_line();
    let mut count = 0;

    for c in s.chars() {
        if c == '1' {
            count += 1;
        }
    }

    println!("{}", count);
}

fn read_line() -> String {
    let mut s = String::new();
    std::io::stdin().read_line(&mut s).unwrap();
    s.trim_end().to_string()
}

ABC081B - Shift only

ABC081B - Shift only

問題文
黒板に N 個の正の整数 A1,...,AN が書かれています．
すぬけ君は，黒板に書かれている整数がすべて偶数であるとき，次の操作を行うことができます．
黒板に書かれている整数すべてを，2 で割ったものに置き換える．
すぬけ君は最大で何回操作を行うことができるかを求めてください．

制約
1≤N≤200
1≤Ai≤10^9

入力
入力は以下の形式で標準入力から与えられる。
N
A1 A2 ... AN

出力
すぬけ君は最大で何回操作を行うことができるかを出力せよ．

入力例 1
3
8 12 40

出力例 1
2

基本的な標準入力、標準出力、四則演算、ループ、条件分岐、配列の操作ができればOK

main.rust

fn main() {
    // 標準入力から配列を読み込む
    let mut s = String::new();
    std::io::stdin().read_line(&mut s).unwrap();
    let n: usize = s.trim().parse().unwrap();
    let mut a: Vec<i32> = Vec::new();
    s.clear();
    std::io::stdin().read_line(&mut s).unwrap();
    for x in s.split_whitespace() {
        a.push(x.parse().unwrap());
    }

    // 配列の各要素を 2 で割る操作を繰り返す
    let mut count = 0;
    loop {
        let mut is_even = true;
        for i in 0..n {
            if a[i] % 2 == 1 {
                is_even = false;
                break;
            }
            a[i] /= 2;
        }
        if !is_even {
            break;
        }
        count += 1;
    }

    // 結果を出力する
    println!("{}", count);
}

ABC087B - Coins

ABC087B - Coins

問題文
あなたは、500 円玉を A 枚、100 円玉を B 枚、50 円玉を C 枚持っています。 これらの硬貨の中から何枚かを選び、合計金額をちょうど X 円にする方法は何通りありますか。
同じ種類の硬貨どうしは区別できません。2 通りの硬貨の選び方は、ある種類の硬貨についてその硬貨を選ぶ枚数が異なるとき区別されます。

制約
0≤A,B,C≤50
A+B+C≥1
50≤X≤20,000
A,B,C は整数である
X は 50 の倍数である

入力
入力は以下の形式で標準入力から与えられる。
A
B
C
X

出力
硬貨を選ぶ方法の個数を出力せよ。

基本的な標準入力、標準出力、四則演算、ループ、条件分岐、配列の操作、全通りの探索ができればOK

main.rust

use std::io;

fn main() {
    // A, B, C, X を受け取る
    let mut input = String::new();
    io::stdin().read_line(&mut input).unwrap();
    let a: u32 = input.trim().parse().unwrap();

    let mut input = String::new();
    io::stdin().read_line(&mut input).unwrap();
    let b: u32 = input.trim().parse().unwrap();

    let mut input = String::new();
    io::stdin().read_line(&mut input).unwrap();
    let c: u32 = input.trim().parse().unwrap();

    let mut input = String::new();
    io::stdin().read_line(&mut input).unwrap();
    let x: u32 = input.trim().parse().unwrap();

    // 答えを保存する変数
    let mut ans = 0;

    // A, B, C の組み合わせを全て試す
    for i in 0..(a+1) {
        for j in 0..(b+1) {
            for k in 0..(c+1) {
                // 金額が X 円になったらカウントする
                if 500 * i + 100 * j + 50 * k == x {
                    ans += 1;
                }
            }
        }
    }

    // 答えを出力する
    println!("{}", ans);
}

ABC083B - Some Sums

ABC083B - Some Sums

問題文
1 以上 N 以下の整数のうち、10 進法での各桁の和が A 以上 B 以下であるものの総和を求めてください。

制約
1≤N≤10^4
1≤A≤B≤36
入力はすべて整数である

入力
入力は以下の形式で標準入力から与えられる。
N A B

出力
1 以上 N 以下の整数のうち、10 進法での各桁の和が A 以上 B 以下であるものの総和を出力せよ。

基本的な標準入力、標準出力、四則演算、ループ、条件分岐、配列の操作、組み込み関数の利用、ができればOK

main.rust

use std::io;

fn main() {
    let mut input = String::new();

    // 標準入力から文字列として入力を受け取る
    io::stdin().read_line(&mut input).unwrap();

    // スペースで区切って、各要素を数値に変換して取り出す
    let mut numbers = input.split_whitespace().map(|s| s.parse::<u32>().unwrap());
    let n = numbers.next().unwrap();
    let a = numbers.next().unwrap();
    let b = numbers.next().unwrap();

    // 1以上N以下の整数を列挙する
    let mut sum = 0;
    for i in 1..=n {
        // 10進法で各桁の和を求める
        let mut digit_sum = 0;
        let mut j = i;
        while j > 0 {
            digit_sum += j % 10;
            j /= 10;
        }

        // 各桁の和がA以上B以下であれば、sumに加算する
        if a <= digit_sum && digit_sum <= b {
            sum += i;
        }
    }

    // 結果を出力する
    println!("{}", sum);
}

ABC088B - Card Game for Two

ABC088B - Card Game for Two

問題文
N 枚のカードがあります. i 枚目のカードには, ai という数が書かれています.
Alice と Bob は, これらのカードを使ってゲームを行います. ゲームでは, Alice と Bob が交互に 1 枚ずつカードを取っていきます. Alice が先にカードを取ります.
2 人がすべてのカードを取ったときゲームは終了し, 取ったカードの数の合計がその人の得点になります. 2 人とも自分の得点を最大化するように最適な戦略を取った時, Alice は Bob より何点多く取るか求めてください.

制約
N は 1 以上 100 以下の整数
ai(1≤i≤N) は 1 以上 100 以下の整数

入力
入力は以下の形式で標準入力から与えられる.
N
a1 a2 a3 ... aN

出力
両者が最適な戦略を取った時, Alice は Bob より何点多く取るかを出力してください.

基本的な標準入力、標準出力、四則演算、条件分岐、配列の操作、ループ、ソート、ができればOK

main.rust

use std::io;

fn main() {
    let mut input = String::new();
    io::stdin().read_line(&mut input).unwrap();
    let n: usize = input.trim().parse().unwrap();

    let mut a: Vec<i32> = Vec::new();
    input.clear();
    io::stdin().read_line(&mut input).unwrap();
    for x in input.split_whitespace() {
        let x: i32 = x.parse().unwrap();
        a.push(x);
    }

    a.sort_by(|a, b| b.cmp(a));

    let mut alice_score = 0;
    let mut bob_score = 0;
    for (i, x) in a.iter().enumerate() {
        if i % 2 == 0 {
            alice_score += x;
        } else {
            bob_score += x;
        }
    }

    println!("{}", alice_score - bob_score);
}

ABC085B - Kagami Mochi

ABC085B - Kagami Mochi

問題文
X 段重ねの鏡餅 (X≥1) とは、X 枚の円形の餅を縦に積み重ねたものであって、どの餅もその真下の餅より直径が小さい（一番下の餅を除く）もののことです。例えば、直径 10、8、6 センチメートルの餅をこの順に下から積み重ねると 3 段重ねの鏡餅になり、餅を一枚だけ置くと 1 段重ねの鏡餅になります。
ダックスフンドのルンルンは N 枚の円形の餅を持っていて、そのうち i 枚目の餅の直径は diセンチメートルです。これらの餅のうち一部または全部を使って鏡餅を作るとき、最大で何段重ねの鏡餅を作ることができるでしょうか。

制約
1≤N≤100
1≤di≤100
入力値はすべて整数である。

入力
入力は以下の形式で標準入力から与えられる。
N
d1
​:
dN

​出力
作ることのできる鏡餅の最大の段数を出力せよ。

入力例 1
4
10
8
8
6

出力例 1
3

基本的な標準入力、標準出力、四則演算、ループ、配列、集合、ができればOK

main.rust

use std::io;

fn main() {
    let mut n = String::new();
    io::stdin().read_line(&mut n).unwrap();
    let n: i32 = n.trim().parse().unwrap();

    let mut ds = Vec::new();
    for _ in 0..n {
        let mut d = String::new();
        io::stdin().read_line(&mut d).unwrap();
        let d: i32 = d.trim().parse().unwrap();
        ds.push(d);
    }

    // 餅の直径を昇順に並び替える
    ds.sort();

    // 答えを保存する変数
    let mut ans = 0;

    // 現在の最大直径を保存する変数
    let mut max_d = 0;

    // 餅の直径を順に見ていく
    for d in ds {
        // 最大直径よりも小さければ、餅を重ねることができる
        if d > max_d {
            max_d = d;
            ans += 1;
        }
    }

    println!("{}", ans);
}

ABC085C - Otoshidama

ABC085C - Otoshidama

問題文
日本でよく使われる紙幣は、10000 円札、5000 円札、1000 円札です。以下、「お札」とはこれらのみを指します。
青橋くんが言うには、彼が祖父から受け取ったお年玉袋にはお札が N 枚入っていて、合計で Y 円だったそうですが、嘘かもしれません。このような状況がありうるか判定し、ありうる場合はお年玉袋の中身の候補を一つ見つけてください。なお、彼の祖父は十分裕福であり、お年玉袋は十分大きかったものとします。

制約
1≤N≤2000
1000≤Y≤2×10^7
N は整数である。
Y は 1000 の倍数である。

入力
入力は以下の形式で標準入力から与えられる。
N Y

出力
N 枚のお札の合計金額が Y 円となることがありえない場合は、-1 -1 -1 と出力せよ。
N 枚のお札の合計金額が Y 円となることがありうる場合は、そのような N 枚のお札の組み合わせの一例を「10000 円札 x 枚、5000 円札 y 枚、1000 円札 z 枚」として、x、y、z を空白で区切って出力せよ。複数の可能性が考えられるときは、そのうちどれを出力してもよい。

入力例 1
9 45000

出力例 1
4 0 5

基本的な標準入力、標準出力、四則演算、ループができればOK

main.rust

fn main() {
    let mut input = String::new();
    std::io::stdin().read_line(&mut input).unwrap();
    let mut iter = input.split_whitespace();
    let n: i32 = iter.next().unwrap().parse().unwrap();
    let y: i32 = iter.next().unwrap().parse().unwrap();

    for i in 0..n+1 {
        for j in 0..n-i+1 {
            let k = n - i - j;
            if i*10000 + j*5000 + k*1000 == y {
                println!("{} {} {}", i, j, k);
                return;
            }
        }
    }
    println!("-1 -1 -1");
}

ABC049C - 白昼夢

ABC049C - 白昼夢

問題文
英小文字からなる文字列 S が与えられます。 Tが空文字列である状態から始め、以下の操作を好きな回数繰り返すことで S=T とすることができるか判定してください。

制約
1≦∣S∣≦10^5
S は英小文字からなる。

入力
入力は以下の形式で標準入力から与えられる。
S

出力
S=T とすることができる場合 YES を、そうでない場合 NO を出力せよ。

入力例 1
erasedream

出力例 1
YES

基本的な標準入力、標準出力、文字列操作ができればOK

main.rust

fn main() {
    // 文字列Sを受け取る
    let mut s = String::new();
    std::io::stdin().read_line(&mut s).unwrap();
    s.pop(); // 改行文字を削除

    // "dream"、"dreamer"、"erase"、"eraser"を順に消していく
    while !s.is_empty() {
        if s.ends_with("eraser") {
            s.truncate(s.len() - 6);
        } else if s.ends_with("erase") {
            s.truncate(s.len() - 5);
        } else if s.ends_with("dreamer") {
            s.truncate(s.len() - 7);
        } else if s.ends_with("dream") {
            s.truncate(s.len() - 5);
        } else {
            // 消すことができなかった場合はNOを出力して終了
            println!("NO");
            return;
        }
    }

    // 全て消せた場合はYESを出力
    println!("YES");
}

ABC086C - Traveling

ABC086C - Traveling

問題文
シカのAtCoDeerくんは二次元平面上で旅行をしようとしています。 AtCoDeerくんの旅行プランでは、時刻 0 に 点 (0,0) を出発し、 1 以上 N 以下の各 i に対し、時刻tiに 点 (xi,yi) を訪れる予定です。
AtCoDeerくんが時刻 t に 点 (x,y) にいる時、 時刻 t+1 には 点 (x+1,y), (x−1,y), (x,y+1), (x,y−1) のうちいずれかに存在することができます。 その場にとどまることは出来ないことに注意してください。 AtCoDeerくんの旅行プランが実行可能かどうか判定してください。

制約
1 ≤ N ≤ 10^5
0 ≤ xi ≤ 10^5
0 ≤ yi ≤ 10^5
1 ≤ ti ≤ 10^5
ti < ti+1 (1 ≤ i ≤ N−1)
入力は全て整数

入力
入力は以下の形式で標準入力から与えられる。
N
t1 x1 y1
​t2 x2 y2
:
tN xN yN

出力
旅行プランが実行可能ならYesを、不可能ならNoを出力してください。

入力例 1
2
3 1 2
6 1 1

出力例 1
Yes

入力例 2
1
2 100 100

出力例 2
No
(0,0) にいる状態から 2 秒後に (100,100) にいるのは不可能です。

入力例 3
2
5 1 1
100 1 1

出力例 3
No

基本的な標準入力、標準出力、四則演算、ループ、二次元配列の操作、ができればOK

main.rust

use std::io;

fn main() {
    // Nを読み込む
    let n: i32 = read_line().parse().unwrap();

    // (t, x, y)の組を読み込む
    let mut points: Vec<(i32, i32, i32)> = Vec::new();
    for _ in 0..n {
        let line = read_line();
        let mut iter = line.split_whitespace();
        let t: i32 = iter.next().unwrap().parse().unwrap();
        let x: i32 = iter.next().unwrap().parse().unwrap();
        let y: i32 = iter.next().unwrap().parse().unwrap();
        points.push((t, x, y));
    }

    // 最初の点は(0, 0)
    let mut prev_point = (0, 0, 0);

    // (t, x, y)の組を1つずつ処理する
    for point in points {
        let (t, x, y) = point;

        // 前の時刻t-1にいた点から、次の時刻tにいる点への最短移動距離を求める
        let dist = (x - prev_point.1).abs() + (y - prev_point.2).abs();
        let time_diff = (t - prev_point.0).abs();
      
        // 最短移動距離が1でない場合は、旅行プランが実行不可能
        if  dist > time_diff || (time_diff - dist)%2 != 0 {
            println!("No");
            return;
        }

        // 前の点を更新する
        prev_point = point;
    }

    // 旅行プランが実行可能
    println!("Yes");
}

// 1行を読み込む
fn read_line() -> String {
    let mut input = String::new();
    io::stdin().read_line(&mut input).unwrap();
    input.trim().to_string()
}

参考