633.平方数之和:找出是否存在a,b,使得a^2 + b^2=c,c是给定的
这个题是1.两数之和的升级版,找出nums数组中是否有两个数之和等于给定的目标值—target
脑子一热,觉得可以拿两数之和的代码魔改一下,提交了3次才过,要么超时,要么解答错了,贴代码:
class Solution:
def judgeSquareSum(self, c: int) -> bool:
nums = range(0, ceil(sqrt(c))+1) #一个小优化,小于根号C的作为要寻找的对象
hashtable = dict()
for index, value in enumerate(nums):
if pow(value, 2) > c: #没加这个之前一直报错,因为负数不能开方
continue
temp = sqrt(c-pow(value,2))
if temp in hashtable: #找到了直接返回True
return True
hashtable[nums[index]] = index
if sqrt(c/2) in hashtable: #这里发现c=2过不去,因为哈希表只存了一个1,所以这里单独考虑a==b的情况
return True
return False
其实,介个解法的时间空间效果都很差,只是想该改一改以前的代码,我觉得能解决这个问题。。。