steven/TinySTL
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

解决了迭代器向前操作的bug

Browse Source
This commit is contained in:
邹晓航
2015-01-14 14:36:55 +08:00
parent 5f49b81ab6
commit e3619bdbd5
1 changed files with 77 additions and 57 deletions
Download Patch File Download Diff File Expand all files Collapse all files

134
TinySTL/BinarySearchTree.h
View File

@@ -54,14 +54,14 @@ namespace TinySTL{
bool empty()const{ return root_ == 0; } bool empty()const{ return root_ == 0; }
size_t size()const{ return size_; } size_t size()const{ return size_; }
size_t height()const{ return height_aux(root_); } size_t height()const{ return height_aux(root_); }
const_iterator root(){ return const_iterator(root_, this); } const_iterator root()const{ return const_iterator(root_, this); }
const_iterator cbegin(){ return find_min(); } const_iterator cbegin()const{ return find_min(); }
const_iterator cend(){ return const_iterator(0, this); } const_iterator cend()const{ return const_iterator(0, this); }
const_iterator find_min(); const_iterator find_min()const;
const_iterator find_max(); const_iterator find_max()const;
const_iterator find(const T& val); const_iterator find(const T& val)const;
void print_preorder(const string& delim = " ", std::ostream& os = std::cout)const; void print_preorder(const string& delim = " ", std::ostream& os = std::cout)const;
void print_inorder(const string& delim = " ", std::ostream& os = std::cout)const; void print_inorder(const string& delim = " ", std::ostream& os = std::cout)const;
@@ -79,9 +79,9 @@ namespace TinySTL{
size_t height_aux(node *p)const; size_t height_aux(node *p)const;
void erase_elem(const T& val, node *&ptr); void erase_elem(const T& val, node *&ptr);
void insert_elem(const T& val, node *&ptr); void insert_elem(const T& val, node *&ptr);
const_iterator find_min_aux(const node *ptr); const_iterator find_min_aux(const node *ptr)const;
const_iterator find_max_aux(const node *ptr); const_iterator find_max_aux(const node *ptr)const;
const_iterator find_aux(const T& val, const node *ptr); const_iterator find_aux(const T& val, const node *ptr)const;
void print_preorder_aux(const string& delim, std::ostream& os, const node *ptr)const; void print_preorder_aux(const string& delim, std::ostream& os, const node *ptr)const;
void print_inorder_aux(const string& delim, std::ostream& os, const node *ptr)const; void print_inorder_aux(const string& delim, std::ostream& os, const node *ptr)const;
void print_postorder_aux(const string& delim, std::ostream& os, const node *ptr)const; void print_postorder_aux(const string& delim, std::ostream& os, const node *ptr)const;
@@ -132,6 +132,7 @@ namespace TinySTL{
ptr = ptr->right_; ptr = ptr->right_;
else else
ptr = ptr->left_; ptr = ptr->left_;
nodeAllocator::destroy(temp);
nodeAllocator::deallocate(temp); nodeAllocator::deallocate(temp);
--size_; --size_;
} }
@@ -145,9 +146,9 @@ namespace TinySTL{
void binary_search_tree<T>::insert_elem(const T& val, node *&ptr){//<2F>ظ<EFBFBD><D8B8><EFBFBD>Ԫ<EFBFBD>ز<EFBFBD><D8B2><EFBFBD><EFBFBD><EFBFBD> void binary_search_tree<T>::insert_elem(const T& val, node *&ptr){//<2F>ظ<EFBFBD><D8B8><EFBFBD>Ԫ<EFBFBD>ز<EFBFBD><D8B2><EFBFBD><EFBFBD><EFBFBD>
if (ptr == 0){ if (ptr == 0){
ptr = nodeAllocator::allocate(); ptr = nodeAllocator::allocate();
construct(&(ptr->data_, val)); TinySTL::construct(&(ptr->data_), val);
/*memset(ptr, 0, sizeof(node)); //memset(ptr, 0, sizeof(node));
ptr->data_ = val;*/ //ptr->data_ = val;
ptr->left_ = ptr->right_ = 0; ptr->left_ = ptr->right_ = 0;
++size_; ++size_;
} }
@@ -222,29 +223,29 @@ namespace TinySTL{
print_postorder_aux(delim, os, root_); print_postorder_aux(delim, os, root_);
} }
template<class T> template<class T>
typename binary_search_tree<T>::const_iterator binary_search_tree<T>::find_min_aux(const node *ptr){ typename binary_search_tree<T>::const_iterator binary_search_tree<T>::find_min_aux(const node *ptr)const{
while (ptr && ptr->left_ != 0){ while (ptr && ptr->left_ != 0){
ptr = ptr->left_; ptr = ptr->left_;
} }
return const_iterator(ptr, this); return const_iterator(ptr, this);
} }
template<class T> template<class T>
typename binary_search_tree<T>::const_iterator binary_search_tree<T>::find_min(){ typename binary_search_tree<T>::const_iterator binary_search_tree<T>::find_min()const{
return find_min_aux(root_); return find_min_aux(root_);
} }
template<class T> template<class T>
typename binary_search_tree<T>::const_iterator binary_search_tree<T>::find_max_aux(const node *ptr){ typename binary_search_tree<T>::const_iterator binary_search_tree<T>::find_max_aux(const node *ptr)const{
while (ptr && ptr->right_ != 0){ while (ptr && ptr->right_ != 0){
ptr = ptr->right_; ptr = ptr->right_;
} }
return const_iterator(ptr, this); return const_iterator(ptr, this);
} }
template<class T> template<class T>
typename binary_search_tree<T>::const_iterator binary_search_tree<T>::find_max(){ typename binary_search_tree<T>::const_iterator binary_search_tree<T>::find_max()const{
return find_max_aux(root_); return find_max_aux(root_);
} }
template<class T> template<class T>
typename binary_search_tree<T>::const_iterator binary_search_tree<T>::find_aux(const T& val, const node *ptr){ typename binary_search_tree<T>::const_iterator binary_search_tree<T>::find_aux(const T& val, const node *ptr)const{
while (ptr){ while (ptr){
if (val == ptr->data_) if (val == ptr->data_)
break; break;
@@ -256,7 +257,7 @@ namespace TinySTL{
return const_iterator(ptr, this); return const_iterator(ptr, this);
} }
template<class T> template<class T>
typename binary_search_tree<T>::const_iterator binary_search_tree<T>::find(const T& val){ typename binary_search_tree<T>::const_iterator binary_search_tree<T>::find(const T& val)const{
return find_aux(val, root_); return find_aux(val, root_);
} }
@@ -271,36 +272,40 @@ namespace TinySTL{
typedef typename binary_search_tree<typename T::value_type>::value_type value_type; typedef typename binary_search_tree<typename T::value_type>::value_type value_type;
typedef typename binary_search_tree<typename T::value_type>::const_reference const_reference; typedef typename binary_search_tree<typename T::value_type>::const_reference const_reference;
typedef typename const T::value_type *const_pointer; typedef typename const T::value_type *const_pointer;
typedef binary_search_tree<typename T::value_type> * cntrPtr; typedef const binary_search_tree<typename T::value_type> * cntrPtr;
private: private:
const T *ptr_; const T *ptr_;
cntrPtr container_; cntrPtr container_;
stack<const T *> parent_;//<2F><><EFBFBD><EFBFBD><EFBFBD><EFBFBD>root<6F><74>ptr_<72>ĸ<EFBFBD><C4B8>ڵ<EFBFBD><DAB5><EFBFBD>·<EFBFBD><C2B7> stack<const T *> parent_;//<2F><><EFBFBD><EFBFBD><EFBFBD><EFBFBD>root<6F><74>ptr_<72>ĸ<EFBFBD><C4B8>ڵ<EFBFBD><DAB5><EFBFBD>·<EFBFBD><C2B7>
std::set<const T *> visited_; std::set<const T *> rvisited_;//<2F><>ǰ<EFBFBD>ڵ<EFBFBD><DAB5><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ƿ񱻷<C7B7><F1B1BBB7>ʹ<EFBFBD>
std::set<const T *> visited_;//<2F><>ǰ<EFBFBD>ڵ<EFBFBD><DAB5>Ƿ񱻷<C7B7><F1B1BBB7>ʹ<EFBFBD><CAB9><EFBFBD><EFBFBD><EFBFBD>node<64><65><EFBFBD><EFBFBD><EFBFBD><EFBFBD>˵<EFBFBD><CBB5><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ѱ<EFBFBD><D1B1><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ˣ<EFBFBD>
public: public:
bst_iter(const T *ptr, cntrPtr container) bst_iter(const T *ptr, cntrPtr container)
:ptr_(ptr), container_(container){ :ptr_(ptr), container_(container){
if (!ptr_)
return;
auto temp = container_->root_; auto temp = container_->root_;
while (temp &&ptr_ && temp != ptr_ && temp->data_ != ptr_->data_){ while (temp && temp != ptr_ && temp->data_ != ptr_->data_){
parent_.push(temp); parent_.push(temp);
if (temp->data_ < ptr_->data_) if (temp->data_ < ptr_->data_){
temp = temp->right_; temp = temp->right_;
else if (temp->data_ > ptr_->data_) }else if (temp->data_ > ptr_->data_){
temp = temp->left_; temp = temp->left_;
} }
} }
}
operator const T*(){ return ptr_; } operator const T*(){ return ptr_; }
const_reference operator*(){ return ptr_->data_; } const_reference operator*()const{ return ptr_->data_; }
const_pointer operator ->(){ return &(operator*()); } const_pointer operator ->()const{ return &(operator*()); }
bst_iter left()const{ return bst_iter(ptr_->left_, container_); } bst_iter left()const{ return bst_iter(ptr_->left_, container_); }
bst_iter right()const{ return bst_iter(ptr_->right_, container_); } bst_iter right()const{ return bst_iter(ptr_->right_, container_); }
bst_iter& operator ++(); bst_iter& operator ++();
bst_iter operator ++(int); bst_iter operator ++(int);
bst_iter& operator --(); //bst_iter& operator --();
bst_iter operator --(int); //bst_iter operator --(int);
public: public:
template<class T> template<class T>
friend bool operator ==(const bst_iter<T>& it1, const bst_iter<T>& it2); friend bool operator ==(const bst_iter<T>& it1, const bst_iter<T>& it2);
@@ -309,19 +314,34 @@ namespace TinySTL{
};//end of bst_iter };//end of bst_iter
template<class T> template<class T>
bst_iter<T>& bst_iter<T>::operator ++(){ bst_iter<T>& bst_iter<T>::operator ++(){
visited_.insert(ptr_);//<2F><>Ϊ<EFBFBD>ѷ<EFBFBD><EFBFBD><EFBFBD> visited_.insert(ptr_);//<2F><>node<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
if (ptr_->right_ != 0){ if (ptr_->right_){//<2F><>node<64><65><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
rvisited_.insert(ptr_);
parent_.push(ptr_); parent_.push(ptr_);
ptr_ = ptr_->right_; ptr_ = ptr_->right_;
while (ptr_ != 0 && ptr_->left_ != 0){ while (ptr_ && ptr_->left_){
parent_.push(ptr_); parent_.push(ptr_);
ptr_ = ptr_->left_; ptr_ = ptr_->left_;
} }
}else{ }else{//node<64><65><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ֻ<EFBFBD><D6BB><EFBFBD>򸸽ڵ<F2B8B8BD>·<EFBFBD><C2B7><EFBFBD>ƶ<EFBFBD>
if (!parent_.empty() && visited_.count(parent_.top()) == 0){//<2F><><EFBFBD>ڵ<EFBFBD>δ<EFBFBD><CEB4><EFBFBD><EFBFBD>
while (!parent_.empty()){
ptr_ = parent_.top(); ptr_ = parent_.top();
parent_.pop(); parent_.pop();
}else{ if (visited_.count(ptr_) == 0){//<2F><><EFBFBD>ڵ<EFBFBD><DAB5><EFBFBD>δ<EFBFBD><CEB4><EFBFBD><EFBFBD>
visited_.insert(ptr_);
break;
}else if (rvisited_.count(ptr_) == 0){//<2F><><EFBFBD>ڵ<EFBFBD><DAB5><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>δ<EFBFBD><CEB4><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
rvisited_.insert(ptr_);
if (ptr_->right_){
parent_.push(ptr_);
ptr_ = ptr_->right_;
while (ptr_ && ptr_->left_){
parent_.push(ptr_);
ptr_ = ptr_->left_;
}
}
}
ptr_ = 0; ptr_ = 0;
} }
} }
@@ -333,30 +353,30 @@ namespace TinySTL{
++*this; ++*this;
return res; return res;
} }
template<class T> //template<class T>
bst_iter<T>& bst_iter<T>::operator --(){ //bst_iter<T>& bst_iter<T>::operator --(){
visited_.erase(ptr_);//<2F><>Ϊδ<CEAA><CEB4><EFBFBD><EFBFBD> // visited_.erase(ptr_);//<2F><>Ϊδ<CEAA><CEB4><EFBFBD><EFBFBD>
if (ptr_->left_ != 0){ // if (ptr_->left_ != 0){
parent_.push(ptr_); // parent_.push(ptr_);
ptr_ = ptr_->left_; // ptr_ = ptr_->left_;
while (ptr_ && ptr_->right_){ // while (ptr_ && ptr_->right_){
parent_.push(ptr_); // parent_.push(ptr_);
ptr_ = ptr_->right_; // ptr_ = ptr_->right_;
} // }
}else{ // }else{
if (!parent_.empty() && visited_.count(parent_.top()) == 1){//<2F><><EFBFBD>ڵ<EFBFBD><DAB5>Ѿ<EFBFBD><D1BE><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> // if (!parent_.empty() && visited_.count(parent_.top()) == 1){//<2F><><EFBFBD>ڵ<EFBFBD><DAB5>Ѿ<EFBFBD><D1BE><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
ptr_ = parent_.top(); // ptr_ = parent_.top();
parent_.pop(); // parent_.pop();
} // }
} // }
return *this; // return *this;
} //}
template<class T> //template<class T>
bst_iter<T> bst_iter<T>::operator --(int){ //bst_iter<T> bst_iter<T>::operator --(int){
auto res = *this; // auto res = *this;
--*this; // --*this;
return res; // return res;
} //}
template<class T> template<class T>
bool operator ==(const bst_iter<T>& it1, const bst_iter<T>& it2){ bool operator ==(const bst_iter<T>& it1, const bst_iter<T>& it2){
return it1.ptr_ == it2.ptr_; return it1.ptr_ == it2.ptr_;