視訊地址:www.bilibili.com/video/av78062009/
相關原始碼:github.com/anonymousGiga/Rust-link...
本節我們實現雙連結串列的Peek。
根據我們在之前實現單連結串列的經驗,我們很容易想到我們的peek函式怎麼實現,如下:
pub fn peek_front(&self) -> Option<&T> {
self.head.as_ref().map(|node| {
&node.borrow().elem
})
}
編譯,發現會報錯。
解決辦法,我們可以檢視手冊上RefCell的borrow函式,發現定義如下:
pub fn borrow(&self) -> Ref<'_, T>
pub fn borrow_mut(&self) -> RefMut<'_, T>
那我們將程式碼改為如下:
pub fn peek_front(&self) -> Option<Ref<T>> {
self.head.as_ref().map(|node| {
node.borrow()
})
}
編譯,發現仍然報錯,找不到Ref,則在程式碼中新增如下:
use std::cell::Ref;
繼續報錯,報錯情況如下:
= note: expected enum `std::option::Option<std::cell::Ref<'_, T>>`
found enum `std::option::Option<std::cell::Ref<'_, Node<T>>>`
期望的是
std::option::Option<std::cell::Ref<'_, T>>
但是此處得到的是
std::option::Option<std::cell::Ref<'_, Node<T>>>
很容易發現,我們需要將
Node<T>
中的T拿出來。
此時該如何解決呢?
這裡我們就使用到
std::cell::ref
中的map,如下:
pub fn map<U, F>(orig: Ref<'b, T>, f: F) -> Ref<'b, U>
where
F: FnOnce(&T) -> &U,
U: ?Sized,
最終,我們的peek的程式碼如下:
pub fn peek_front(&self) -> Option<Ref<T>> {
self.head.as_ref().map(|node| {
Ref::map(node.borrow(), |node| &node.elem)
})
}
#[test]
fn peek() {
let mut list = List::new();
assert!(list.peek_front().is_none());
list.push_front(1); list.push_front(2); list.push_front(3);
assert_eq!(&*list.peek_front().unwrap(), &3);
}
use std::rc::Rc;
use std::cell::{Ref, RefCell};
pub struct List<T> {
head: Link<T>,
tail: Link<T>,
}
type Link<T> = Option<Rc<RefCell<Node<T>>>>;
struct Node<T> {
elem: T,
next: Link<T>,
prev: Link<T>,
}
impl<T> Node<T> {
fn new(elem: T) -> Rc<RefCell<Self>> {
Rc::new(RefCell::new(Node {
elem: elem,
prev: None,
next: None,
}))
}
}
impl<T> List<T> {
pub fn new() -> Self {
List { head: None, tail: None }
}
pub fn push_front(&mut self, elem: T) {
let node = Node::new(elem);
match self.head.take() {
Some(head) => {
//取可變引用使用borrow_mut
head.borrow_mut().prev = Some(node.clone());
node.borrow_mut().next = Some(head);
self.head = Some(node);
}
None => {
self.tail = Some(node.clone());
self.head = Some(node);
}
}
}
pub fn pop_front(&mut self) -> Option<T> {
self.head.take().map(|node| {
match node.borrow_mut().next.take() {
Some(next) => {
next.borrow_mut().prev.take();
self.head = Some(next);
}
None => {
self.tail.take();
}
}
//取值使用into_inner
Rc::try_unwrap(node).ok().unwrap().into_inner().elem
})
}
//pub fn peek_front(&self) -> Option<&T> {
pub fn peek_front(&self) -> Option<Ref<T>> {
self.head.as_ref().map(|node| {
//node.borrow()
Ref::map(node.borrow(), |node| &node.elem)
})
}
}
#[cfg(test)]
mod tests {
use super::List;
#[test]
fn basics() {
let mut list = List::new();
assert_eq!(list.pop_front(), None);
list.push_front(1);
list.push_front(2);
list.push_front(3);
assert_eq!(list.pop_front(), Some(3));
assert_eq!(list.pop_front(), Some(2));
list.push_front(4);
list.push_front(5);
assert_eq!(list.pop_front(), Some(5));
assert_eq!(list.pop_front(), Some(4));
assert_eq!(list.pop_front(), Some(1));
assert_eq!(list.pop_front(), None);
}
}
本作品採用《CC 協議》,轉載必須註明作者和本文連結