Q1:给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。注意:答案中不可以包含重复的三元组。
自己写的:暴力解决,时间复杂度太大
class Solution(object):
def threeSum(self, nums):
nums.sort()
result = []
temp = []
for i in range(len(nums)):
for j in range(i + 1, len(nums)):
for k in range(j + 1, len(nums)):
if nums[i] + nums[j] + nums[k] == 0:
result.append([nums[i], nums[j], nums[k]])
for i in range(len(result)):
if result[i] not in temp:
temp.append(result[i])
else:
continue
return temp
if __name__ == "__main__":
s = Solution()
result = s.threeSum([-1, 0, 1, 2, -1, -4, 3, -5, -2, -3])
print(result)
网上大神的解法:
class Solution:
def threeSum(self, nums):
# 存储结果列表
result = []
# 对nums列表进行排序,无返回值,排序直接改变nums顺序
nums.sort()
for i in range(len(nums)):
# 因为是升序排列,如果排序后第一个数都大于0,则跳出循环,不可能有为0的三数之和
if nums[i] > 0:
break
# 排序后相邻两数如果相等,则跳出当前循环继续下一次循环,相同的数只需要计算一次
if i > 0 and nums[i] == nums[i-1]:
continue
# 记录i的下一个位置
j = i + 1
# 最后一个元素的位置
k = len(nums) - 1
while j < k:
# 判断三数之和是否为0
if nums[j] + nums[k] == -nums[i]:
# 把结果加入数组中
result.append([nums[i], nums[j], nums[k]])
# 判断j相邻元素是否相等,有的话跳过这个
while j < k and nums[j] == nums[j+1]: j += 1
# 判断后面k的相邻元素是否相等,是的话跳过
while j < k and nums[k] == nums[k-1]: k -= 1
# 没有相等则j+1,k-1,缩小范围
j += 1
k -= 1
# 小于-nums[i]的话还能往后取
elif nums[j] + nums[k] < -nums[i]:
j += 1
else:
k -= 1
return result
Q2:见下图
A2:
class Solution:
def romanToInt(self, s: str) -> int:
d = {'M': 1000,'D': 500 ,'C': 100,'L': 50,'X': 10,'V': 5,'I': 1}
result = 0
s_len = len(s)
for i in range(s_len-1):
if d[s[i]] < d[s[i+1]]:
result -= d[s[i]]
else:
result += d[s[i]]
result += d[s[-1]]
return result