mirror of
https://github.com/krahets/hello-algo.git
synced 2026-11-08 19:38:08 +08:00
7257641e0e
* Retranslate the complete Japanese online edition from Chinese * Fit Japanese homepage labels and localize heading permalinks * Point Japanese contribution instructions to the Japanese preview * Replace locale conditionals with multilingual UI tables
43 lines
1.2 KiB
Python
43 lines
1.2 KiB
Python
"""
|
||
File: two_sum.py
|
||
Created Time: 2022-11-25
|
||
Author: krahets (krahets@163.com)
|
||
"""
|
||
|
||
|
||
def two_sum_brute_force(nums: list[int], target: int) -> list[int]:
|
||
"""方法 1：総当たり列挙"""
|
||
# 二重ループのため、時間計算量は 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 []
|
||
|
||
|
||
def two_sum_hash_table(nums: list[int], target: int) -> list[int]:
|
||
"""方法 2：補助ハッシュテーブル"""
|
||
# 補助ハッシュテーブルの空間計算量は 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 []
|
||
|
||
|
||
"""実行用コード"""
|
||
if __name__ == "__main__":
|
||
# ======= テストケース =======
|
||
nums = [2, 7, 11, 15]
|
||
target = 13
|
||
|
||
# ====== 実行用コード ======
|
||
# 方法 1
|
||
res: list[int] = two_sum_brute_force(nums, target)
|
||
print("方法 1 res =", res)
|
||
# 方法 2
|
||
res: list[int] = two_sum_hash_table(nums, target)
|
||
print("方法 2 res =", res)
|