shrink_to_fit bug fix

4ee7718e · zouxiaohang · b728798d · 4ee7718e
隐藏空白更改
内联并排

Showing with 6 addition and 1 deletion

TinySTL/Detail/Vector.impl.h TinySTL/Detail/Vector.impl.h +6 -1

未找到文件。
--- 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>