原文链接:https://doc.rust-lang.org/nomicon/vec-insert-remove.html

插入和删除

slice并没有提供插入和删除功能,接下来我们就实现它们。

插入需要把目标位置后的所有元素都向右移动1。这里我们需要用到ptr::copy,它就是C中的memmove的Rust版。它把一块内存从一个地方拷贝到另一个地方,而且可以正确处理源和目标内存区域有重叠的情况(也正是我们这里遇到的情况)。

如果我们在i的位置插入,我们需要把[i .. len]移动到[i+1 .. len+1],len指的是插入前的值。

  1. pub fn insert(&mut self, index: usize, elem: T) {
  2. // 注意:<=是因为我们可以把值插到所有元素的后面
  3. // 这种情况等同于push
  4. assert!(index <= self.len, "index out of bounds");
  5. if self.cap == self.len { self.grow(); }
  6. unsafe {
  7. if index < self.len {
  8. // ptr::copy(src, dest, len): "从src拷贝len个元素到dest"
  9. ptr::copy(self.ptr.offset(index as isize),
  10. self.ptr.offset(index as isize + 1),
  11. self.len - index);
  12. }
  13. ptr::write(self.ptr.offset(index as isize), elem);
  14. self.len += 1;
  15. }
  16. }

删除则是完全相反的行为。我们要把元素[i+1 .. len + 1]移动到[i .. len],len是删除后的值。

  1. pub fn remove(&mut self, index: usize) -> T {
  2. // 注意:<是因为我们不能删除所有元素之后的位置
  3. assert!(index < self.len, "index out of bounds");
  4. unsafe {
  5. self.len -= 1;
  6. let result = ptr::read(self.ptr.offset(index as isize));
  7. ptr::copy(self.ptr.offset(index as isize + 1),
  8. self.ptr.offset(index as isize),
  9. self.len - index);
  10. result
  11. }
  12. }