mirror of
https://github.com/krahets/hello-algo.git
synced 2024-12-26 12:26:29 +08:00
111 lines
2.7 KiB
Swift
111 lines
2.7 KiB
Swift
|
/**
|
|||
|
* File: array_hash_map.swift
|
|||
|
* Created Time: 2023-01-16
|
|||
|
* Author: nuomi1 (nuomi1@qq.com)
|
|||
|
*/
|
|||
|
|
|||
|
import utils
|
|||
|
|
|||
|
/* 基於陣列實現的雜湊表 */
|
|||
|
class ArrayHashMap {
|
|||
|
private var buckets: [Pair?]
|
|||
|
|
|||
|
init() {
|
|||
|
// 初始化陣列,包含 100 個桶
|
|||
|
buckets = Array(repeating: nil, count: 100)
|
|||
|
}
|
|||
|
|
|||
|
/* 雜湊函式 */
|
|||
|
private func hashFunc(key: Int) -> Int {
|
|||
|
let index = key % 100
|
|||
|
return index
|
|||
|
}
|
|||
|
|
|||
|
/* 查詢操作 */
|
|||
|
func get(key: Int) -> String? {
|
|||
|
let index = hashFunc(key: key)
|
|||
|
let pair = buckets[index]
|
|||
|
return pair?.val
|
|||
|
}
|
|||
|
|
|||
|
/* 新增操作 */
|
|||
|
func put(key: Int, val: String) {
|
|||
|
let pair = Pair(key: key, val: val)
|
|||
|
let index = hashFunc(key: key)
|
|||
|
buckets[index] = pair
|
|||
|
}
|
|||
|
|
|||
|
/* 刪除操作 */
|
|||
|
func remove(key: Int) {
|
|||
|
let index = hashFunc(key: key)
|
|||
|
// 置為 nil ,代表刪除
|
|||
|
buckets[index] = nil
|
|||
|
}
|
|||
|
|
|||
|
/* 獲取所有鍵值對 */
|
|||
|
func pairSet() -> [Pair] {
|
|||
|
buckets.compactMap { $0 }
|
|||
|
}
|
|||
|
|
|||
|
/* 獲取所有鍵 */
|
|||
|
func keySet() -> [Int] {
|
|||
|
buckets.compactMap { $0?.key }
|
|||
|
}
|
|||
|
|
|||
|
/* 獲取所有值 */
|
|||
|
func valueSet() -> [String] {
|
|||
|
buckets.compactMap { $0?.val }
|
|||
|
}
|
|||
|
|
|||
|
/* 列印雜湊表 */
|
|||
|
func print() {
|
|||
|
for pair in pairSet() {
|
|||
|
Swift.print("\(pair.key) -> \(pair.val)")
|
|||
|
}
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
@main
|
|||
|
enum _ArrayHashMap {
|
|||
|
/* Driver Code */
|
|||
|
static func main() {
|
|||
|
/* 初始化雜湊表 */
|
|||
|
let map = ArrayHashMap()
|
|||
|
|
|||
|
/* 新增操作 */
|
|||
|
// 在雜湊表中新增鍵值對 (key, value)
|
|||
|
map.put(key: 12836, val: "小哈")
|
|||
|
map.put(key: 15937, val: "小囉")
|
|||
|
map.put(key: 16750, val: "小算")
|
|||
|
map.put(key: 13276, val: "小法")
|
|||
|
map.put(key: 10583, val: "小鴨")
|
|||
|
print("\n新增完成後,雜湊表為\nKey -> Value")
|
|||
|
map.print()
|
|||
|
|
|||
|
/* 查詢操作 */
|
|||
|
// 向雜湊表中輸入鍵 key ,得到值 value
|
|||
|
let name = map.get(key: 15937)!
|
|||
|
print("\n輸入學號 15937 ,查詢到姓名 \(name)")
|
|||
|
|
|||
|
/* 刪除操作 */
|
|||
|
// 在雜湊表中刪除鍵值對 (key, value)
|
|||
|
map.remove(key: 10583)
|
|||
|
print("\n刪除 10583 後,雜湊表為\nKey -> Value")
|
|||
|
map.print()
|
|||
|
|
|||
|
/* 走訪雜湊表 */
|
|||
|
print("\n走訪鍵值對 Key->Value")
|
|||
|
for pair in map.pairSet() {
|
|||
|
print("\(pair.key) -> \(pair.val)")
|
|||
|
}
|
|||
|
print("\n單獨走訪鍵 Key")
|
|||
|
for key in map.keySet() {
|
|||
|
print(key)
|
|||
|
}
|
|||
|
print("\n單獨走訪值 Value")
|
|||
|
for val in map.valueSet() {
|
|||
|
print(val)
|
|||
|
}
|
|||
|
}
|
|||
|
}
|