leetcode279. Perfect Squares

题目描述

Given a positive integer n, find the least number of perfect square numbers (for example, 1, 4, 9, 16, …) which sum to n.

Example 1:

Input: n = 12
Output: 3
Explanation: 12 = 4 + 4 + 4.

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/perfect-squares
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

代码

将问题转化为一个图论问题。
从n到0，每个数字表示一个节点；
如果两个数字x到y相差一个完全平方数，则连接一条边。
由此形成一个无权图。
原问题转化为了，求这个无权图中从n到0的最短路径。

class Solution { // 基本逻辑：广度优先遍历模板
public:
    int numSquares(int n) {
    
        assert( n>0 );
        
        queue<pair<int, int>> q; // 数据对，第一个int表示具体第几个数字，第二个int表示图中经历几段路径走到该数字
        q.push(make_pair(n, 0)); //初始化
        
        while(!q.empty()){
            int num = q.front().first; // 取出队首数据对
            int step = q.front().second;
            q.pop();
            
            if（ num ==0）// 若取出元素num为0则对应step为解
                 return step;

            for(int i = 1; num - i * i >= 0; i ++){ //否则若num还能承受一个完全平方数，i++
                q.push(make_pair( num-i*i, step + 1)); //向队列中推入新的数据对            
        }

        throw invalid_argument("No Solution."); // 正常情况下不会抛出该异常
    }
};

不足：
并非标准的广度优先的实现，存在性能问题。
因为对于每一个数字可以从多种渠道获得，故可能重复推入很多num-i*i进入队列中。
之所以存在这种现象是因为该结构是一个图而非一棵树

改进

处理冗余节点的推入

#include <iostream>
#include <vector>
#include <queue>
#include <stdexcept>

using namespace std;

/// BFS
/// Time Complexity: O(n)
/// Space Complexity: O(n)
class Solution {
public:
    int numSquares(int n) {

        if(n == 0)
            return 0;

        queue<pair<int, int>> q;
        q.push(make_pair(n, 0));

        vector<bool> visited(n + 1, false); //改进，用于表示0至n这n+1个数字是否被访问过
        visited[n] = true; //n已经在初始时被推入队列，更改为true

        while(!q.empty()){
            int num = q.front().first;
            int step = q.front().second;
            q.pop();

            for(int i = 1; num - i * i >= 0; i ++){
                int a = num - i * i; // 每次进入循环先求出a，减少计算量
                if(!visited[a]){ //入队前先判断该数字是否已被访问过
                    if(a == 0) return step + 1;
                    q.push(make_pair(a, step + 1));
                    visited[a] = true;
                }
            }
        }

        throw invalid_argument("No Solution.");
    }
};

int main() {

    cout << Solution().numSquares(12) << endl;
    cout << Solution().numSquares(13) << endl;

    return 0;
}