shrink_to_fit bug fix

2015-12-25 12:50:25 +08:00
parent b728798dde
commit 4ee7718ef2
1 changed files with 6 additions and 1 deletions
--- a/TinySTL/Detail/Vector.impl.h
+++ b/TinySTL/Detail/Vector.impl.h
@@ -226,7 +226,12 @@ namespace TinySTL{
 	}
 	template<class T, class Alloc>
 	void vector<T, Alloc>::shrink_to_fit(){
-		dataAllocator::deallocate(finish_, endOfStorage_ - finish_);
+		//dataAllocator::deallocate(finish_, endOfStorage_ - finish_);
+		//endOfStorage_ = finish_;
+		T* t = (T*)dataAllocator::allocate(size());
+		finish_ = TinySTL::uninitialized_copy(start_, finish_, t);
+		dataAllocator::deallocate(start_, capacity());
+		start_ = t;
 		endOfStorage_ = finish_;
 	}
 	template<class T, class Alloc>