Я определил шаблон класса Vec, который содержит последовательность вектора, подобного T:
template <typename T> class Vec {
public:
size_t size() const { return first_free - elements; }
size_t capacity() const { return cap - elements; }
T *begin() const { return elements; }
T *end() const { return first_free; }
void resize(const size_t); // something wrong in this function
...
private:
allocator<T> alloc; //allocates the elements
T *elements; // pointer to the first element in the array
T *first_free; // pointer to the first free element in the array
T *cap; // pointer to one past the end of the array
...
};
когда мой код вызывает эту функцию, программа падает,
template <typename T> void Vec<T>::resize(const size_t newcap)
{
// this part goes well
if(newcap > capacity()) {
auto newdata = alloc.allocate(newcap);
auto dest = newdata;
auto orig = elements;
for(size_t i = 0; i != size(); ++i)
alloc.construct(dest++, std::move(*orig++));
elements = newdata;
first_free = dest;
cap = elements + newcap;
}
// crash happens in the following two parts
else if(size() <= newcap && newcap < capacity()) { // deallocate the redundant memory where no element exists
alloc.deallocate(elements + newcap, capacity() - newcap); // this line causes crash
cap = elements + newcap;
}
else if(size() < newcap) { // destroy the elements and deallocate the memory
for(auto p = first_free; p != elements + newcap; /*empty*/)
alloc.destroy(--p);
alloc.deallocate(elements + newcap, capacity() - newcap); // this line causes crash
first_free = cap = elements + newcap;
}
}
код вызова:
Vec<string> v{"aaa", "bbb", "ccc"}; // size() == 3, capacity == 3
v.resize(2); // to resize v to size() == 2, capacity == 2
Я думаю, что я сделал правильный вызов освобождения и правильную арифметику указателя. Спасибо.
alloc.deallocate(elements + newcap, capacity() - newcap);
Вы не можете сделать это. Распределитель — это все или ничего: вы либо освобождаете весь выделенный блок, либо оставляете его в покое.
Чтобы уменьшить выделенную память, выполните тот же танец выделения-и-копирования, который вы делаете, когда новый размер больше текущей емкости, но выделяйте новый меньший размер.
Кстати, часть помечена //this part goes well
имеет большую проблему: он никогда не освобождает старый блок памяти.
Других решений пока нет …