为了提升自己的编码能力,所以开始刷起来了LeetCode。以后争取每天刷一题。
第一题TwoSum
问题描述:
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.
Example:
Given nums = [2, 7, 11, 15], target = 9,
Because nums[0] + nums[1] = 2 + 7 = 9,
return [0, 1].
问题分析
穷举法
问题所要求的是给定一串一维数组和一个定值,问这个数组中哪两个数相加可以得到给定的定值,并且返回这两个数在数组中的下标。
为此设置两个for循环,遍历数组,并用if语句判断哪两个数相加才能得到定值。
详细代码
class Solution(object):
def twoSum(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: List[int]
"""
for i in range(len(nums)):
for j in range(i+1,len(nums)):
if nums[i] + nums[j] == target:
return [i,j]
时间复杂度为O(N^2)
空间复杂度为O(1)
这个算法应该是消耗内存和时间都比较大的笨方法,以后会看一些大牛的算法提高自己,并看到厉害有意思的代码会再从新添加。
字典(hash表)
只需一个for循环,降低了时间复杂度,但是提高了空间复杂度,以空间换时间的例子。
class Solution(object):
def twoSum(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: List[int]
"""
dict = {}
for i in range(len(nums)):
if nums[i] in dict:
return [dict[nums[i]],i]
else:
dict[target - nums[i]] = i
时间复杂度为o(n)
空间复杂度为O(n)