Two Sum 的Python实现
本科学文,刷算法题确实有难度,不过万事开头难,想想开始学编程的那个时候也觉得难,后来还不是学会了吗?fuxk it 开始吧!
- 先挑简单的练练手
题目地址:Two Sum
后续还有第二版、第三版…
问题
Given an array of integers, return indices of the two numbers such that they add up to a specific target. You may assume that each input would have exactly one solution, and you may not use the same element twice.
粗暴的翻译:给你一个整数数组和一个目标值,数组内的任意两个不相同元素相加,若结果与目标值相同则返回这两个元素在数组中的索引值。
比如:
Given nums = [2, 7, 11, 15], target = 9,
Because nums[0] + nums[1] = 2 + 7 = 9,
return [0, 1].
思路:
用字典记录遍历过的列表元素(作为键key)及其索引值(作为值value),如果某一个元素与目标值的差刚好存在于字典中,返回这个键的值,即它的索引值。
补充知识点:
Python的字典(dict)是Hash表(也叫散列表,是通过键直接访问内存存储位置的数据结构),查询速度O(1)。
字典的键(key)是唯一的。
补充阅读:
散列函数(哈希)
散列表(哈希表)
Rabin–Karp algorithm(一种字符串搜索算法)
ps:wiki链接,打不开的话,有兴趣可自行搜索
实现:
class Solution(object):
def twoSum(self, nums, target):
d = {}
for i, n in enumerate(nums):
m = target - n
if m in d:
return [d[m], i]
else:
d[n] = i