完全平方数

先算小于N的完全平方数,再用这些数做广度优先:

class Solution:
    def numSquares(self, n: int) -> int:
        
        if n == 0:
            return 0
        
        tempNum = []
        
        x = 1
        i = 1
        #计算小于n的完全平方数
        while x <= n:
            tempNum.append(x)
            i += 1
            x = pow(i, 2)
        
        from collections import deque
        stack = deque([[n, 0]])
        
        while stack:     #使用广度优先算法遍历每层
            num, level = stack.popleft()
            for i in tempNum:
                temp = num - i
                if temp > 0:
                    stack.append([temp, level + 1])
                elif temp == 0:
                    return level + 1

大神解法:

class Solution:
    # isSquare 为新增
    def isSquare(self, n: int) -> bool:
        sq = int(n ** 0.5)
        return sq * sq == n  

    def numSquares(self, n: int) -> int:
        #######  参考题解 用数学方法  拉格朗日四平方和定理######
        ################# python 动态规划超时  ##############
        if self.isSquare(n):
            return 1
        while n & 3 == 0:
            n >>= 2
        if n & 7 == 7:
            return 4    # 满足 4^a (8b+7)   返回 4
        for i in range(1, int(n ** 0.5)+1):
            if self.isSquare(n - i*i):
                return 2
        return 3

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
完全平方数是对某个整数的平方结果,例如4(2^2)、9(3^2)等。在数学上,对于任意正整数n,如果存在另一个整数m使得m * m = n,则称n为完全平方数。 在 C++ 中处理完全平方数数对(假设这里指的是寻找两个整数,它们各自的平方相加等于目标值),我们可以创建一个函数,该函数接收一个整数作为输入,并尝试找到所有满足条件的整数组合。下面是具体的实现: ```cpp #include <iostream> #include <vector> using namespace std; void findSquarePairs(int targetSum) { for (int i = 0; i <= sqrt(targetSum); ++i) { // 避免不必要的循环迭代 int squareRoot = i * i; if (squareRoot > targetSum) break; // 如果当前平方根已经大于targetSum,则跳出循环 int otherPart = targetSum - squareRoot; int squareRootOfOtherPart = sqrt(otherPart); // 检查otherPart是否是一个完全平方数 if(squareRootOfOtherPart == floor(sqrt(otherPart))) { cout << "一对完全平方数:" << i << "^2 + " << squareRootOfOtherPart << "^2 = " << targetSum << endl; } } } // 使用示例 int main() { int target = 25; findSquarePairs(target); return 0; } ``` 在这个例子中,`findSquarePairs` 函数首先遍历小于等于 `sqrt(targetSum)` 的数字,然后计算出剩余部分是否也是一个完全平方数。如果是的话,就找到了一组符合条件的数对。 --- 相关问题 --- 1. 这种方法的时间复杂度是多少?如何优化? 2. 如果目标值很大,那么如何减少内存消耗? 3. 对于负数或零的情况,这个算法应该如何修改?
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值