【Lintcode】checkSumOfSquareNumbers巧用无序集合unordered_set

#include <bits/stdc++.h>
#include <unordered_set>
using namespace std;
bool judgeSquareSum(int c) {
unordered_set<int> s;
for (int i = 0; i <= sqrt(c); ++i) {
s.insert(i * i);
if (s.count(c - i * i)) return true;
}
return false;
}
int main(){
printf("%d", judgeSquareSum(5));
return 0;
} 

由于编译器版本问题，可能会无法使用unordered_set，这时可用以下方式修改宏定义。

#if(__cplusplus == 201103L)
#include <unordered_map>
#include <unordered_set>
#else
#include <tr1/unordered_map>
#include <tr1/unordered_set>
namespace std
{
using std::tr1::unordered_map;
using std::tr1::unordered_set;
}
#endif

还可以打印出无序集合unordered_set

unordered_set<int>::iterator it;
for ( it = s.begin(); it != s.end(); it++ )
cout << *it << " ";
cout << endl;

checkSumOfSquareNumbers的其他算法

C++中unordered_map的版本兼容性问题

Lintcode-697

最后

以上就是冷静薯片最近收集整理的关于【Lintcode】checkSumOfSquareNumbers巧用无序集合unordered_set的全部内容，更多相关【Lintcode】checkSumOfSquareNumbers巧用无序集合unordered_set内容请搜索靠谱客的其他文章。