判断是否为平方数之和-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，…，c√∗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