题意简述:给定一个正整数,求出它最少可以用多少个平方数之和表示。例如13=4+9只需两个平方数。
输入:正整数n
输出:平方数的最少个数
题解:
求解最少个数的话应该采用bfs。采用队列存放中间得到的整数。对于当前队头n,减去一个平方数可能得到n-1、n-4、……
n−⌊n√⌋2
,将这些数当中之前在队列中未出现的数全部放进队列里,平方数个数+1。每一次循环只处理队列同一层的,继续处理可得到减去两个平方数的、三个……直到某个数减去平方数后等于0,搜索结束,返回当前的平方数个数。
class Solution {
public:
int numSquares(int n) {
unordered_map<int,int> mymap;
queue<int> myqueue;
int cur = 0;
mymap[n] = cur;
myqueue.push(n);
while(1) {
cur++;
int last = myqueue.size();
while(last-- > 0) {
int top = myqueue.front();
for(int i = 1;i*i <= top;i++) {
int cand = top - i*i;
if(cand == 0) return cur;
if(mymap.find(cand) == mymap.end()) {
mymap[cand] = cur;
myqueue.push(cand);
}
}
myqueue.pop();
}
}
}
};