mirror of
https://github.com/krahets/hello-algo.git
synced 2026-06-15 22:57:48 +08:00
52 lines
1.7 KiB
Python
52 lines
1.7 KiB
Python
"""
|
||
File: hashing_search.py
|
||
Created Time: 2022-11-26
|
||
Author: timi (xisunyy@163.com)
|
||
"""
|
||
|
||
import sys
|
||
from pathlib import Path
|
||
|
||
sys.path.append(str(Path(__file__).parent.parent))
|
||
from modules import ListNode, list_to_linked_list
|
||
|
||
|
||
def hashing_search_array(hmap: dict[int, int], target: int) -> int:
|
||
"""ハッシュ探索（配列）"""
|
||
# ハッシュテーブルの key: 目標要素、value: インデックス
|
||
# ハッシュテーブルにこの key がなければ -1 を返す
|
||
return hmap.get(target, -1)
|
||
|
||
|
||
def hashing_search_linkedlist(
|
||
hmap: dict[int, ListNode], target: int
|
||
) -> ListNode | None:
|
||
"""ハッシュ探索（連結リスト）"""
|
||
# ハッシュテーブルの key: 対象要素、value: ノードオブジェクト
|
||
# ハッシュテーブルにこの key がなければ None を返す
|
||
return hmap.get(target, None)
|
||
|
||
|
||
"""Driver Code"""
|
||
if __name__ == "__main__":
|
||
target = 3
|
||
|
||
# ハッシュ探索（配列）
|
||
nums = [1, 5, 3, 2, 4, 7, 5, 9, 10, 8]
|
||
# ハッシュテーブルを初期化
|
||
map0 = dict[int, int]()
|
||
for i in range(len(nums)):
|
||
map0[nums[i]] = i # key: 要素、value: インデックス
|
||
index: int = hashing_search_array(map0, target)
|
||
print("対象要素 3 のインデックス =", index)
|
||
|
||
# ハッシュ探索（連結リスト）
|
||
head: ListNode = list_to_linked_list(nums)
|
||
# ハッシュテーブルを初期化
|
||
map1 = dict[int, ListNode]()
|
||
while head:
|
||
map1[head.val] = head # key: ノード値、value: ノード
|
||
head = head.next
|
||
node: ListNode = hashing_search_linkedlist(map1, target)
|
||
print("対象ノード値 3 に対応するノードオブジェクトは", node)
|