【Lintcode】checkSumOfSquareNumbers巧用无序集合unordered_set

yes小宇

于 2018-04-30 20:53:01 发布

阅读量194

点赞数

CC 4.0 BY-SA版权

分类专栏：数据结构和算法文章标签： Lintcose unordered_set checkSumOfSquareNumbers 无序集合兼容性

本文链接：https://blog.youkuaiyun.com/qq_19297719/article/details/80151839

数据结构和算法专栏收录该内容

2 篇文章

订阅专栏

本文介绍了一种使用C++实现的算法，该算法用于判断一个给定的整数是否可以表示为两个整数平方和的形式。通过使用无序集合unordered_set进行优化，提高了查找效率。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#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