hello-algo/ja/codes/rust/chapter_computational_complexity/iteration.rs

/*
 * File: iteration.rs
 * Created Time: 2023-09-02
 * Author: night-cruise ([email protected])
 */

/* for ループ */
fn for_loop(n: i32) -> i32 {
    let mut res = 0;
    // ループで 1, 2, ..., n-1, n の和を求める
    for i in 1..=n {
        res += i;
    }
    res
}

/* while ループ */
fn while_loop(n: i32) -> i32 {
    let mut res = 0;
    let mut i = 1; // 条件変数を初期化する

    // ループで 1, 2, ..., n-1, n の和を求める
    while i <= n {
        res += i;
        i += 1; // 条件変数を更新する
    }
    res
}

/* while ループ（2 回の更新） */
fn while_loop_ii(n: i32) -> i32 {
    let mut res = 0;
    let mut i = 1; // 条件変数を初期化する

    // ループで 1, 4, 10, ... の和を求める
    while i <= n {
        res += i;
        // 条件変数を更新する
        i += 1;
        i *= 2;
    }
    res
}

/* 二重 for ループ */
fn nested_for_loop(n: i32) -> String {
    let mut res = vec![];
    // i = 1, 2, ..., n-1, n のループ
    for i in 1..=n {
        // j = 1, 2, ..., n-1, n のループ
        for j in 1..=n {
            res.push(format!("({}, {}), ", i, j));
        }
    }
    res.join("")
}

/* 実行用コード */
fn main() {
    let n = 5;
    let mut res;

    res = for_loop(n);
    println!("\nfor ループで求めた和 res = {res}");

    res = while_loop(n);
    println!("\nwhile ループで求めた和 res = {res}");

    res = while_loop_ii(n);
    println!("\nwhile ループ（2 回の更新）で求めた和 res = {}", res);

    let res = nested_for_loop(n);
    println!("\n二重 for ループの走査結果 {res}");
}