Решение на упр.08 задача 1 от Мариян Момчилов
Към профила на Мариян Момчилов
Код
use std::rc::Rc;
struct List<T> {
head: Link<T>,
}
type Link<T> = Option<Rc<Node<T>>>;
struct Node<T> {
elem: T,
next: Link<T>,
}
impl<T> List<T> {
// Създава празен списък
fn new() -> Self {
List { head: None }
}
// Връща първия елемент от списъка
fn head(&self) -> Option<&T> {
self.head.as_ref().map(|rc| &rc.elem)
}
// Връща опашката на списъка - т.е. списък съдържащ
// всички елементи освен първия.
// Опашката на празен списък е празен списък.
fn tail(&self) -> List<T> {
let head: Link<T>;
match &self.head {
Some(rc) => head = rc.next.clone(),
None => head = None,
}
List { head }
}
// Добавя елемент към списъка.
// Връща нов списък с първи елемент е `value` и
// опашка стария списък.
fn prepend(&self, value: T) -> List<T> {
List {
head: Some(Rc::new(Node {
elem: value,
next: self.head.clone(),
})),
}
}
// Създава итератор по елементите на списъка.
// Итератора връща елементи от тип `&T`.
fn iter(&self) -> Iter<'_, T> {
Iter { link: &self.head }
}
}
struct Iter<'a, T> {
link: &'a Link<T>,
}
impl<'a, T> Iterator for Iter<'a, T> {
type Item = &'a T;
fn next(&mut self) -> Option<Self::Item> {
match self.link {
None => None,
Some(l) => {
self.link = &l.next;
return Some(&l.elem);
}
}
}
}
impl<T> Drop for List<T> {
fn drop(&mut self) {
let mut link = self.head.take();
while let Some(rc) = link {
if let Ok(mut node) = Rc::try_unwrap(rc) {
link = node.next.take();
} else {
break;
}
}
}
}