LeetCode_279.完全平方数

279. 完全平方数

给定正整数 n，找到若干个完全平方数（比如 1, 4, 9, 16, ...）使得它们的和等于 n。你需要让组成和的完全平方数的个数最少。

示例 1:

输入: n = 12

输出: 3 
解释: 12 = 4 + 4 + 4.

示例 2:

输入: n = 13

输出: 2
解释: 13 = 4 + 9.

---

解题思路：

 对问题建模，将整个问题转化为一个图论问题。从n到0，每个数字代表一个节点；如果两个数字x和y相差一个平方数，则连接一条边。我们得到了一个无权图。

原问题转化成，求这个无权图中从n到0的最短路径。

图的广度优先遍历要用到队列queue

解题代码：

1.图的广度优先遍历（和树的层序遍历方法一样）

class Solution {
public:
    int numSquares(int n) {
        
        assert(n>0);
        
        queue< pair<int,int> > q;
        q.push(make_pair(n,0));
        
        vector<bool> visited(n+1,false);
        visited[n]=true;
        while(!q.empty())
        {
            int num =q.front().first;
            int step=q.front().second;
            q.pop();
            
            if(num==0)
                return step;
            for(int i=1;num-i*i>=0;i++)
            {
                if(!visited[num-i*i])
                {
                    q.push(make_pair(num-i*i,step+1));
                    visited[num-i*i]=true;
                }
            }
            
        }
        return -1;
    }
}; 

pair<int,int> 第一个int代表第几个数字，第二个int代表图中经历了几段路径 走到数字。

2.优化的版本

class Solution {
public:
    int numSquares(int n) {
        
        assert(n>0);
        
        queue< pair<int,int> > q;
        q.push(make_pair(n,0));
        
        vector<bool> visited(n+1,false);
        visited[n]=true;
        while(!q.empty())
        {
            int num =q.front().first;
            int step=q.front().second;
            q.pop();
            
            for(int i=1; ;i++)
            {
                int a=num-i*i;
                if(a<0)
                    break;
                if(a==0)
                    return step+1;
                if(!visited[a])
                {
                    q.push(make_pair(a,step+1));
                    visited[a]=true;
                }
            }
            
        }
        return -1;
    }
};