判断是否为平方数之和-LintCode

给一个整数 c, 你需要判断是否存在两个整数 a 和 b 使得 a^2 + b^2 = c.

样例：
给出 n = 5
返回 true // 1 * 1 + 2 * 2 = 5
给出 n = -5
返回 false

思路：
构建set包含0*0，1*1，…，$\sqrt {c}*\sqrt{c}$
利用指针it，遍历set，查找是否有元素等于c-*it，
若存在，返回true

#ifndef C697_H
#define C697_H
#include<iostream>
#include<set>
#include<math.h>
using namespace std;
class Solution {
public:
    /*
    * @param : the given number
    * @return: whether whether there're two integers
    */
    bool checkSumOfSquareNumbers(int num) {
        // write your code here
        if (num < 0)
            return false;
        int n = (int)sqrt(num);
        set<int> set;
        for (int i = 0; i <= n; ++i)
            set.insert(i*i);
        for (auto it = set.begin(); it != set.end(); ++it)
        {
            if (set.find(num - *it) != set.end())
            {
                return true;
            }
        }
        return false;
    }
};
#endif