017 Rustでリンクリストによる要素追加を実装する

導入

ビデオアドレス: https://www.bilibili.com/video/av78062009/
関連ソースコード: https://github.com/anonymousGiga/Rust-link-list

詳細

前のセクションでは二重リンク リストを定義しましたが、このセクションでは二重リンク リスト用の新しい関数とプッシュフロント関数を作成します。

新しい

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) => {
				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);
			}
		}
	}

完全なコード

追加後の完全なコードは次のとおりです。

use std::rc::Rc;
use std::cell::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) => {
				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);
			}
		}
	}
}