LeetCode 279 Perfect Squares 题解

题意简述：给定一个正整数，求出它最少可以用多少个平方数之和表示。例如13=4+9只需两个平方数。
输入：正整数n
输出：平方数的最少个数

---

题解：
求解最少个数的话应该采用bfs。采用队列存放中间得到的整数。对于当前队头n，减去一个平方数可能得到n-1、n-4、……$n-\lfloor\sqrt{n}\rfloor^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();
            }
        }
    }
};