LeetCode系列之(1)两数之和
题目描述
给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
示例:
给定 nums = [2, 7, 11, 15], target = 9 因为 nums[0] + nums[1] = 2 + 7 = 9 所以返回 [0, 1]
我的解题思路
当遇见这个问题的时候,首先想到的就是遍历所有两两数之和,从组合角度共有
C
n
2
=
n
×
(
n
−
1
)
2
C^2_n=\frac{n\times(n-1)}{2}
Cn2=2n×(n−1)中可能,如何遍历成为了一个解决问题的关键。
那么如何遍历所有两两组合呢?下面围绕这个问题进行分析:
假设嵌套循环的长度都为n,过滤掉两次选的同样位置的元素,那么从排序角度,是没有重复的,但从组合角度会重复,但从循环的子语句中加以判断过滤掉重复组合,反而会增大存储空间,解决不了循环次数,也解决不了实质问题,由此只能从循环语句上进行设计,让遍历的次数正好等于组合的个数,并且不会重复组合。
是否存在遍历的次数正好等于组合的个数,并且不会重复组合?
答案是肯定的,比如索引
0
,
1
,
.
.
.
,
n
0,1,...,n
0,1,...,n的数组,在双层循环中,外层取索引为0时候,内层从
1
,
.
.
.
,
n
1,...,n
1,...,n,外层取1时候,内层取
2
,
.
.
.
,
n
2,...,n
2,...,n,以此类推,这样循环的次数等于组合数量,并且不会重复组合,满足要求。
具体python代码如下:
@python
class Solution(object):
def twoSum(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: List[int]
"""
target_list = []
for i in range(len(nums)):
for j in range(i+1,len(nums)):
if (nums[i]+nums[j] == target):
target_list = [i,j]
return target_list
官方解题思路
官方给出了3个思路
- 暴力破解
给出的暴力破解方法如我给出的方法,这个方法优点是容易想出来和实现,缺点就是复杂度 O ( n 2 ) O(n^2) O(n2),算法复杂度较高。有没有复杂度低一点的方法呢,请看下面。 - 两遍哈希表
两边哈希表,借助哈希表这个数据结构,以空间换时间去快速解决该问题,具体如何解决的呢?可以分为两步:
(1)循环一遍,获取列表中的每个值以及所对应的索引。
(2)循环一遍,计算 t e m p = t a r g e t − n u m s [ i ] temp=target-nums[i] temp=target−nums[i],查找temp是否存在列表中(这里需要排除本身),如果存在返回其对应的索引,程序结束。
@python
class Solution(object):
def twoSum(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: List[int]
"""
dict_nums = {}
for i,num in enumerate(nums):
dict_nums[num] = i
for i,num in enumerate(nums):
temp = target - num
get_v = dict_nums.get(temp,None)
if (get_v):
if get_v != i:
return [i,get_v]
从时间复杂度角度考虑,使用了 O ( n ) O(n) O(n),大大缩短了时间。
- 一遍哈希表
程序能否在两遍哈希表的基础上做一下简化呢?
答案是可以的,其实少做改进,两次循环完全可以做成一次。因为每个值都会被查到 1 − n 1-n 1−n次,最左边的元素第一次进入哈希表,以后没循环一次,就会第一个就会被询问1次,最右边的元素只有最后一次会被问到,所以从这个方面讲,一遍哈希表完全可以完成。具体实现如下代码:
class Solution(object):
def twoSum(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: List[int]
"""
dict_nums = {}
temp_list = []
for i,num in enumerate(nums):
temp = target - num
if dict_nums.get(num,None):
dict_nums[num].append(i)
else:
dict_nums[num] = [i]
get_v = dict_nums.get(temp,None)
temp_list.append(get_v)
if (get_v):
if get_v[0] !=i:
return [get_v[0],i]
反思
(1)解决问题的方法有很多中,如何选择一个更好的方法。
(2)同样的一种方法,是否有更进一步的优化空间。