Rust 插入和删除
slice并没有提供插入和删除功能,接下来我们就实现它们。
插入需要把目标位置后的所有元素都向右移动1。这里我们需要用到ptr::copy
,它就是C中的memmove
的Rust版。它把一块内存从一个地方拷贝到另一个地方,而且可以正确处理源和目标内存区域有重叠的情况(也正是我们这里遇到的情况)。
如果我们在i
的位置插入,我们需要把[i .. len]移动到[i+1 .. len+1],len指的是插入前的值。
pub fn insert(&mut self, index: usize, elem: T) { // 注意:<=是因为我们可以把值插到所有元素的后面 // 这种情况等同于push assert!(index <= self.len, "index out of bounds"); if self.cap == self.len { self.grow(); } unsafe { if index < self.len { // ptr::copy(src, dest, len): "从src拷贝len个元素到dest" ptr::copy(self.ptr.offset(index as isize), self.ptr.offset(index as isize + 1), self.len - index); } ptr::write(self.ptr.offset(index as isize), elem); self.len += 1; } }
删除则是完全相反的行为。我们要把元素[i+1 .. len + 1]移动到[i .. len],len是删除后的值。
pub fn remove(&mut self, index: usize) -> T { // 注意:<是因为我们不能删除所有元素之后的位置 assert!(index < self.len, "index out of bounds"); unsafe { self.len -= 1; let result = ptr::read(self.ptr.offset(index as isize)); ptr::copy(self.ptr.offset(index as isize + 1), self.ptr.offset(index as isize), self.len - index); result } }
我们继续编写迭代器。iter和iter_mut其实已经写过了,感谢神奇的DeRef。但是还有两个有意思的迭代器是Vec提供的而slice没有的:into_iter和drain。IntoIter以值而不是引用的形式 ...