Решение на упр.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>,
}
struct Iter<'a, T> {
head: Option<&'a Node<T>>
}
impl<T> List<T> {
// Създава празен списък
fn new() -> Self {
List { head: None }
}
// Връща първия елемент от списъка
fn head(&self) -> Option<&T> {
match self.head.as_ref() {
Some(node) => Some(&node.elem),
None => None
}
}
// Връща опашката на списъка - т.е. списък съдържащ
// всички елементи освен първия.
// Опашката на празен списък е празен списък.
fn tail(&self) -> List<T> {
match self.head.as_ref() {
Some(node) => List { head: node.next.clone() },
None => List { head: None },
}
}
// Добавя елемент към списъка.
// Връща нов списък с първи елемент е `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> {
match self.head.as_ref() {
Some(node) => Iter { head: Some(node) },
None => Iter { head: None },
}
}
}
impl<'a, T> Iterator for Iter<'a, T> {
type Item = &'a T;
fn next(&mut self) -> Option<Self::Item> {
let next: Option<Self::Item>;
match self.head {
Some(link) => {
match &link.next {
Some(node) => self.head = Some(&node),
None => self.head = None,
}
next = Some(&link.elem)
},
None => next = None,
};
next
}
}