hello-algo/zh-hant/codes/rust/chapter_searching/linear_search.rs

/*
 * File: linear_search.rs
 * Created Time: 2023-07-09
 * Author: codingonion (coderonion@gmail.com)
 */

include!("../include/include.rs");

use list_node::ListNode;
use std::cell::RefCell;
use std::rc::Rc;

/* 線性查詢（陣列） */
fn linear_search_array(nums: &[i32], target: i32) -> i32 {
    // 走訪陣列
    for (i, num) in nums.iter().enumerate() {
        // 找到目標元素，返回其索引
        if num == &target {
            return i as i32;
        }
    }
    // 未找到目標元素，返回 -1
    return -1;
}

/* 線性查詢（鏈結串列） */
fn linear_search_linked_list(
    head: Rc<RefCell<ListNode<i32>>>,
    target: i32,
) -> Option<Rc<RefCell<ListNode<i32>>>> {
    // 找到目標節點，返回之
    if head.borrow().val == target {
        return Some(head);
    };
    // 找到目標節點，返回之
    if let Some(node) = &head.borrow_mut().next {
        return linear_search_linked_list(node.clone(), target);
    }
    // 未找到目標節點，返回 None
    return None;
}

/* Driver Code */
pub fn main() {
    let target = 3;

    /* 在陣列中執行線性查詢 */
    let nums = [1, 5, 3, 2, 4, 7, 5, 9, 10, 8];
    let index = linear_search_array(&nums, target);
    println!("目標元素 3 的索引 = {}", index);

    /* 在鏈結串列中執行線性查詢 */
    let head = ListNode::arr_to_linked_list(&nums);
    let node = linear_search_linked_list(head.unwrap(), target);
    println!("目標節點值 3 的對應節點物件為 {:?}", node);
}