完成初始化函数

This commit is contained in:
邹晓航
2015-01-19 11:33:04 +08:00
parent e3124e599d
commit 080d41814d
2 changed files with 99 additions and 0 deletions

View File

@@ -0,0 +1,56 @@
#ifndef _UNORDERED_SET_IMPL_H_
#define _UNORDERED_SET_IMPL_H_
namespace TinySTL{
template<class Key, class Hash, class KeyEqual, class Allocator>
size_t Unordered_set<Key, Hash, KeyEqual, Allocator>::prime_list_[PRIME_LIST_SIZE] = {
53, 97, 193, 389, 769, 1543, 3079, 6151, 12289, 24593, 49157, 98317, 196613, 393241,
786433, 1572869, 3145739, 6291469, 12582917, 25165843, 50331653, 100663319, 201326611,
402653189, 805306457, 1610612741, 3221225473, 4294967291,
};
template<class Key, class Hash, class KeyEqual, class Allocator>
bool Unordered_set<Key, Hash, KeyEqual, Allocator>::has_key(const key_type& key){
auto& list = buckets_[bucket_index(key)];
return TinySTL::find(list.begin(), list.end(), key) != list.end();
}
template<class Key, class Hash, class KeyEqual, class Allocator>
typename Unordered_set<Key, Hash, KeyEqual, Allocator>::size_type
Unordered_set<Key, Hash, KeyEqual, Allocator>::bucket_index(const key_type& key)const{
return haser()(key) % buckets_.size();
}
template<class Key, class Hash, class KeyEqual, class Allocator>
typename Unordered_set<Key, Hash, KeyEqual, Allocator>::size_type
Unordered_set<Key, Hash, KeyEqual, Allocator>::next_prime(size_type n)const{
auto i = 0;
for (; i != PRIME_LIST_SIZE; ++i){
if (n > prime_list_[i])
continue;
else
break;
}
i = (i == PRIME_LIST_SIZE ? PRIME_LIST_SIZE - 1 : i);
return prime_list_[i];
}
template<class Key, class Hash, class KeyEqual, class Allocator>
Unordered_set<Key, Hash, KeyEqual, Allocator>::Unordered_set(size_type bucket_count){
bucket_count = next_prime(bucket_count);
buckets_.resize(bucket_count);
size_ = 0;
}
template<class Key, class Hash, class KeyEqual, class Allocator>
template<class InputIterator>
Unordered_set<Key, Hash, KeyEqual, Allocator>::Unordered_set(InputIterator first, InputIterator last){
size_ = 0;
auto len = last - first;
buckets_.resize(len);
for (; first != last; ++first){
auto index = bucket_index(*first);
if (!has_key(*first)){
buckets_[index].push_front(*first);
++size_;
}
}
}
}
#endif

43
TinySTL/Unordered_set.h Normal file
View File

@@ -0,0 +1,43 @@
#ifndef _UNORDERED_SET_H_
#define _UNORDERED_SET_H_
#include "Allocator.h"
#include "Algorithm.h"
#include "Functional.h"
#include "List.h"
#include "Vector.h"
#include <list>
#include <vector>
namespace TinySTL{
template<class Key, class Hash = std::hash<Key>,
class KeyEqual = TinySTL::equal_to<Key>, class Allocator = TinySTL::allocator < Key >>
class Unordered_set{
public:
typedef Key key_type;
typedef Key value_type;
typedef size_t size_type;
typedef Hash haser;
typedef Allocator allocator_type;
typedef value_type& reference;
typedef const value_type& const_reference;
private:
TinySTL::vector<TinySTL::list<key_type>> buckets_;
size_type size_;
#define PRIME_LIST_SIZE 28
static size_t prime_list_[PRIME_LIST_SIZE];
public:
explicit Unordered_set(size_t bucket_count);
template<class InputIterator>
Unordered_set(InputIterator first, InputIterator last);
//Unordered_set(const Unordered_set& ust);
//Unordered_set& operator = (const Unordered_set& ust);
private:
size_type next_prime(size_type n)const;
size_type bucket_index(const key_type& key)const;
bool has_key(const key_type& key);
};
}//end of namespace TinySTL
#include "Detail\Unordered_set.impl.h"
#endif