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

/*
 * File: hashing_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::collections::HashMap;
use std::rc::Rc;

/* 雜湊查詢（陣列） */
fn hashing_search_array<'a>(map: &'a HashMap<i32, usize>, target: i32) -> Option<&'a usize> {
    // 雜湊表的 key: 目標元素，value: 索引
    // 若雜湊表中無此 key ，返回 None
    map.get(&target)
}

/* 雜湊查詢（鏈結串列） */
fn hashing_search_linked_list(
    map: &HashMap<i32, Rc<RefCell<ListNode<i32>>>>,
    target: i32,
) -> Option<&Rc<RefCell<ListNode<i32>>>> {
    // 雜湊表的 key: 目標節點值，value: 節點物件
    // 若雜湊表中無此 key ，返回 None
    map.get(&target)
}

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

    /* 雜湊查詢（陣列） */
    let nums = [1, 5, 3, 2, 4, 7, 5, 9, 10, 8];
    // 初始化雜湊表
    let mut map = HashMap::new();
    for (i, num) in nums.iter().enumerate() {
        map.insert(*num, i); // key: 元素，value: 索引
    }
    let index = hashing_search_array(&map, target);
    println!("目標元素 3 的索引 = {}", index.unwrap());

    /* 雜湊查詢（鏈結串列） */
    let head = ListNode::arr_to_linked_list(&nums);
    // 初始化雜湊表
    // let mut map1 = HashMap::new();
    let map1 = ListNode::linked_list_to_hashmap(head);
    let node = hashing_search_linked_list(&map1, target);
    println!("目標節點值 3 的對應節點物件為 {:?}", node);
}