---
comments: true
---
# 10.4 雜湊最佳化策略
在演算法題中,**我們常透過將線性查詢替換為雜湊查詢來降低演算法的時間複雜度**。我們藉助一個演算法題來加深理解。
!!! question
給定一個整數陣列 `nums` 和一個目標元素 `target` ,請在陣列中搜索“和”為 `target` 的兩個元素,並返回它們的陣列索引。返回任意一個解即可。
## 10.4.1 線性查詢:以時間換空間
考慮直接走訪所有可能的組合。如圖 10-9 所示,我們開啟一個兩層迴圈,在每輪中判斷兩個整數的和是否為 `target` ,若是,則返回它們的索引。
![線性查詢求解兩數之和](replace_linear_by_hashing.assets/two_sum_brute_force.png){ class="animation-figure" }
圖 10-9 線性查詢求解兩數之和
程式碼如下所示:
=== "Python"
```python title="two_sum.py"
def two_sum_brute_force(nums: list[int], target: int) -> list[int]:
"""方法一:暴力列舉"""
# 兩層迴圈,時間複雜度為 O(n^2)
for i in range(len(nums) - 1):
for j in range(i + 1, len(nums)):
if nums[i] + nums[j] == target:
return [i, j]
return []
```
=== "C++"
```cpp title="two_sum.cpp"
/* 方法一:暴力列舉 */
vector twoSumBruteForce(vector &nums, int target) {
int size = nums.size();
// 兩層迴圈,時間複雜度為 O(n^2)
for (int i = 0; i < size - 1; i++) {
for (int j = i + 1; j < size; j++) {
if (nums[i] + nums[j] == target)
return {i, j};
}
}
return {};
}
```
=== "Java"
```java title="two_sum.java"
/* 方法一:暴力列舉 */
int[] twoSumBruteForce(int[] nums, int target) {
int size = nums.length;
// 兩層迴圈,時間複雜度為 O(n^2)
for (int i = 0; i < size - 1; i++) {
for (int j = i + 1; j < size; j++) {
if (nums[i] + nums[j] == target)
return new int[] { i, j };
}
}
return new int[0];
}
```
=== "C#"
```csharp title="two_sum.cs"
/* 方法一:暴力列舉 */
int[] TwoSumBruteForce(int[] nums, int target) {
int size = nums.Length;
// 兩層迴圈,時間複雜度為 O(n^2)
for (int i = 0; i < size - 1; i++) {
for (int j = i + 1; j < size; j++) {
if (nums[i] + nums[j] == target)
return [i, j];
}
}
return [];
}
```
=== "Go"
```go title="two_sum.go"
/* 方法一:暴力列舉 */
func twoSumBruteForce(nums []int, target int) []int {
size := len(nums)
// 兩層迴圈,時間複雜度為 O(n^2)
for i := 0; i < size-1; i++ {
for j := i + 1; j < size; j++ {
if nums[i]+nums[j] == target {
return []int{i, j}
}
}
}
return nil
}
```
=== "Swift"
```swift title="two_sum.swift"
/* 方法一:暴力列舉 */
func twoSumBruteForce(nums: [Int], target: Int) -> [Int] {
// 兩層迴圈,時間複雜度為 O(n^2)
for i in nums.indices.dropLast() {
for j in nums.indices.dropFirst(i + 1) {
if nums[i] + nums[j] == target {
return [i, j]
}
}
}
return [0]
}
```
=== "JS"
```javascript title="two_sum.js"
/* 方法一:暴力列舉 */
function twoSumBruteForce(nums, target) {
const n = nums.length;
// 兩層迴圈,時間複雜度為 O(n^2)
for (let i = 0; i < n; i++) {
for (let j = i + 1; j < n; j++) {
if (nums[i] + nums[j] === target) {
return [i, j];
}
}
}
return [];
}
```
=== "TS"
```typescript title="two_sum.ts"
/* 方法一:暴力列舉 */
function twoSumBruteForce(nums: number[], target: number): number[] {
const n = nums.length;
// 兩層迴圈,時間複雜度為 O(n^2)
for (let i = 0; i < n; i++) {
for (let j = i + 1; j < n; j++) {
if (nums[i] + nums[j] === target) {
return [i, j];
}
}
}
return [];
}
```
=== "Dart"
```dart title="two_sum.dart"
/* 方法一: 暴力列舉 */
List twoSumBruteForce(List nums, int target) {
int size = nums.length;
// 兩層迴圈,時間複雜度為 O(n^2)
for (var i = 0; i < size - 1; i++) {
for (var j = i + 1; j < size; j++) {
if (nums[i] + nums[j] == target) return [i, j];
}
}
return [0];
}
```
=== "Rust"
```rust title="two_sum.rs"
/* 方法一:暴力列舉 */
pub fn two_sum_brute_force(nums: &Vec, target: i32) -> Option> {
let size = nums.len();
// 兩層迴圈,時間複雜度為 O(n^2)
for i in 0..size - 1 {
for j in i + 1..size {
if nums[i] + nums[j] == target {
return Some(vec![i as i32, j as i32]);
}
}
}
None
}
```
=== "C"
```c title="two_sum.c"
/* 方法一:暴力列舉 */
int *twoSumBruteForce(int *nums, int numsSize, int target, int *returnSize) {
for (int i = 0; i < numsSize; ++i) {
for (int j = i + 1; j < numsSize; ++j) {
if (nums[i] + nums[j] == target) {
int *res = malloc(sizeof(int) * 2);
res[0] = i, res[1] = j;
*returnSize = 2;
return res;
}
}
}
*returnSize = 0;
return NULL;
}
```
=== "Kotlin"
```kotlin title="two_sum.kt"
/* 方法一:暴力列舉 */
fun twoSumBruteForce(nums: IntArray, target: Int): IntArray {
val size = nums.size
// 兩層迴圈,時間複雜度為 O(n^2)
for (i in 0..
此方法的時間複雜度為 $O(n^2)$ ,空間複雜度為 $O(1)$ ,在大資料量下非常耗時。
## 10.4.2 雜湊查詢:以空間換時間
考慮藉助一個雜湊表,鍵值對分別為陣列元素和元素索引。迴圈走訪陣列,每輪執行圖 10-10 所示的步驟。
1. 判斷數字 `target - nums[i]` 是否在雜湊表中,若是,則直接返回這兩個元素的索引。
2. 將鍵值對 `nums[i]` 和索引 `i` 新增進雜湊表。
=== "<1>"
![輔助雜湊表求解兩數之和](replace_linear_by_hashing.assets/two_sum_hashtable_step1.png){ class="animation-figure" }
=== "<2>"
![two_sum_hashtable_step2](replace_linear_by_hashing.assets/two_sum_hashtable_step2.png){ class="animation-figure" }
=== "<3>"
![two_sum_hashtable_step3](replace_linear_by_hashing.assets/two_sum_hashtable_step3.png){ class="animation-figure" }
圖 10-10 輔助雜湊表求解兩數之和
實現程式碼如下所示,僅需單層迴圈即可:
=== "Python"
```python title="two_sum.py"
def two_sum_hash_table(nums: list[int], target: int) -> list[int]:
"""方法二:輔助雜湊表"""
# 輔助雜湊表,空間複雜度為 O(n)
dic = {}
# 單層迴圈,時間複雜度為 O(n)
for i in range(len(nums)):
if target - nums[i] in dic:
return [dic[target - nums[i]], i]
dic[nums[i]] = i
return []
```
=== "C++"
```cpp title="two_sum.cpp"
/* 方法二:輔助雜湊表 */
vector twoSumHashTable(vector &nums, int target) {
int size = nums.size();
// 輔助雜湊表,空間複雜度為 O(n)
unordered_map dic;
// 單層迴圈,時間複雜度為 O(n)
for (int i = 0; i < size; i++) {
if (dic.find(target - nums[i]) != dic.end()) {
return {dic[target - nums[i]], i};
}
dic.emplace(nums[i], i);
}
return {};
}
```
=== "Java"
```java title="two_sum.java"
/* 方法二:輔助雜湊表 */
int[] twoSumHashTable(int[] nums, int target) {
int size = nums.length;
// 輔助雜湊表,空間複雜度為 O(n)
Map dic = new HashMap<>();
// 單層迴圈,時間複雜度為 O(n)
for (int i = 0; i < size; i++) {
if (dic.containsKey(target - nums[i])) {
return new int[] { dic.get(target - nums[i]), i };
}
dic.put(nums[i], i);
}
return new int[0];
}
```
=== "C#"
```csharp title="two_sum.cs"
/* 方法二:輔助雜湊表 */
int[] TwoSumHashTable(int[] nums, int target) {
int size = nums.Length;
// 輔助雜湊表,空間複雜度為 O(n)
Dictionary dic = [];
// 單層迴圈,時間複雜度為 O(n)
for (int i = 0; i < size; i++) {
if (dic.ContainsKey(target - nums[i])) {
return [dic[target - nums[i]], i];
}
dic.Add(nums[i], i);
}
return [];
}
```
=== "Go"
```go title="two_sum.go"
/* 方法二:輔助雜湊表 */
func twoSumHashTable(nums []int, target int) []int {
// 輔助雜湊表,空間複雜度為 O(n)
hashTable := map[int]int{}
// 單層迴圈,時間複雜度為 O(n)
for idx, val := range nums {
if preIdx, ok := hashTable[target-val]; ok {
return []int{preIdx, idx}
}
hashTable[val] = idx
}
return nil
}
```
=== "Swift"
```swift title="two_sum.swift"
/* 方法二:輔助雜湊表 */
func twoSumHashTable(nums: [Int], target: Int) -> [Int] {
// 輔助雜湊表,空間複雜度為 O(n)
var dic: [Int: Int] = [:]
// 單層迴圈,時間複雜度為 O(n)
for i in nums.indices {
if let j = dic[target - nums[i]] {
return [j, i]
}
dic[nums[i]] = i
}
return [0]
}
```
=== "JS"
```javascript title="two_sum.js"
/* 方法二:輔助雜湊表 */
function twoSumHashTable(nums, target) {
// 輔助雜湊表,空間複雜度為 O(n)
let m = {};
// 單層迴圈,時間複雜度為 O(n)
for (let i = 0; i < nums.length; i++) {
if (m[target - nums[i]] !== undefined) {
return [m[target - nums[i]], i];
} else {
m[nums[i]] = i;
}
}
return [];
}
```
=== "TS"
```typescript title="two_sum.ts"
/* 方法二:輔助雜湊表 */
function twoSumHashTable(nums: number[], target: number): number[] {
// 輔助雜湊表,空間複雜度為 O(n)
let m: Map = new Map();
// 單層迴圈,時間複雜度為 O(n)
for (let i = 0; i < nums.length; i++) {
let index = m.get(target - nums[i]);
if (index !== undefined) {
return [index, i];
} else {
m.set(nums[i], i);
}
}
return [];
}
```
=== "Dart"
```dart title="two_sum.dart"
/* 方法二: 輔助雜湊表 */
List twoSumHashTable(List nums, int target) {
int size = nums.length;
// 輔助雜湊表,空間複雜度為 O(n)
Map dic = HashMap();
// 單層迴圈,時間複雜度為 O(n)
for (var i = 0; i < size; i++) {
if (dic.containsKey(target - nums[i])) {
return [dic[target - nums[i]]!, i];
}
dic.putIfAbsent(nums[i], () => i);
}
return [0];
}
```
=== "Rust"
```rust title="two_sum.rs"
/* 方法二:輔助雜湊表 */
pub fn two_sum_hash_table(nums: &Vec, target: i32) -> Option> {
// 輔助雜湊表,空間複雜度為 O(n)
let mut dic = HashMap::new();
// 單層迴圈,時間複雜度為 O(n)
for (i, num) in nums.iter().enumerate() {
match dic.get(&(target - num)) {
Some(v) => return Some(vec![*v as i32, i as i32]),
None => dic.insert(num, i as i32),
};
}
None
}
```
=== "C"
```c title="two_sum.c"
/* 雜湊表 */
typedef struct {
int key;
int val;
UT_hash_handle hh; // 基於 uthash.h 實現
} HashTable;
/* 雜湊表查詢 */
HashTable *find(HashTable *h, int key) {
HashTable *tmp;
HASH_FIND_INT(h, &key, tmp);
return tmp;
}
/* 雜湊表元素插入 */
void insert(HashTable *h, int key, int val) {
HashTable *t = find(h, key);
if (t == NULL) {
HashTable *tmp = malloc(sizeof(HashTable));
tmp->key = key, tmp->val = val;
HASH_ADD_INT(h, key, tmp);
} else {
t->val = val;
}
}
/* 方法二:輔助雜湊表 */
int *twoSumHashTable(int *nums, int numsSize, int target, int *returnSize) {
HashTable *hashtable = NULL;
for (int i = 0; i < numsSize; i++) {
HashTable *t = find(hashtable, target - nums[i]);
if (t != NULL) {
int *res = malloc(sizeof(int) * 2);
res[0] = t->val, res[1] = i;
*returnSize = 2;
return res;
}
insert(hashtable, nums[i], i);
}
*returnSize = 0;
return NULL;
}
```
=== "Kotlin"
```kotlin title="two_sum.kt"
/* 方法二:輔助雜湊表 */
fun twoSumHashTable(nums: IntArray, target: Int): IntArray {
val size = nums.size
// 輔助雜湊表,空間複雜度為 O(n)
val dic = HashMap()
// 單層迴圈,時間複雜度為 O(n)
for (i in 0..
此方法透過雜湊查詢將時間複雜度從 $O(n^2)$ 降至 $O(n)$ ,大幅提升執行效率。
由於需要維護一個額外的雜湊表,因此空間複雜度為 $O(n)$ 。**儘管如此,該方法的整體時空效率更為均衡,因此它是本題的最優解法**。