标签：02 node mut self let 二叉树 stack Rust result

环境

• Time 2022-04-21
• Rust 1.60.0

前言

说明

基于标准库来学习各种数据结构，并不是从头实现数据结构，未考虑实现性能。
实现了二叉树的前序、中序和后序遍历。

示例

引入模块

pub mod binary_tree;

结构定义

use super::{NodeRef, Tree};

#[derive(Default)]
pub struct BinaryTree<T> {
    root: NodeRef<T>,
}

前序遍历

fn pre_order(&self) -> Vec<&T> {
    let mut result = Vec::new();
    let mut stack = vec![&self.root];
    while let Some(node) = stack.pop() {
        if let Some(node) = node {
            stack.push(&node.right);
            stack.push(&node.left);
            result.push(&node.value);
        }
    }
    result
}

中序遍历

fn in_order(&self) -> Vec<&T> {
    let mut result = Vec::new();
    let mut stack = Vec::new();
    let mut current = &self.root;
    while current.is_some() || !stack.is_empty() {
        while let Some(node) = current {
            stack.push(current);
            current = &node.left;
        }
        current = stack.pop().unwrap();
        result.push(&current.as_ref().unwrap().value);
        current = &current.as_ref().unwrap().right;
    }
    result
}

后序遍历

fn post_order(&self) -> Vec<&T> {
    let mut stack = vec![&self.root];
    let mut result = vec![];
    while let Some(node) = stack.pop() {
        if let Some(node) = node {
            result.push(&node.value);
            stack.push(&node.left);
            stack.push(&node.right);
        }
    }
    result.reverse();
    result
}

其它方法

impl<T> BinaryTree<T> {
    pub(crate) fn root_mut(&mut self) -> &mut NodeRef<T> {
        &mut self.root
    }

    pub(crate) fn root(&self) -> &NodeRef<T> {
        &self.root
    }
}

其它未实现方法

fn insert(&mut self, _: T) {
    unimplemented!()
}

fn remove(&mut self, _: &T) -> Option<T> {
    unimplemented!()
}

fn contains(&mut self, _: &T) -> bool {
    unimplemented!()
}

总结

实现了二叉树的前序、中序和后序遍历方法，其它方法未实现。

附录

源码

use super::{NodeRef, Tree};

#[derive(Default)]
pub struct BinaryTree<T> {
    root: NodeRef<T>,
}

impl<T> Tree<T> for BinaryTree<T> {
    fn pre_order(&self) -> Vec<&T> {
        let mut result = Vec::new();
        let mut stack = vec![&self.root];
        while let Some(node) = stack.pop() {
            if let Some(node) = node {
                stack.push(&node.right);
                stack.push(&node.left);
                result.push(&node.value);
            }
        }
        result
    }

    fn in_order(&self) -> Vec<&T> {
        let mut result = Vec::new();
        let mut stack = Vec::new();
        let mut current = &self.root;
        while current.is_some() || !stack.is_empty() {
            while let Some(node) = current {
                stack.push(current);
                current = &node.left;
            }
            current = stack.pop().unwrap();
            result.push(&current.as_ref().unwrap().value);
            current = &current.as_ref().unwrap().right;
        }
        result
    }

    fn post_order(&self) -> Vec<&T> {
        let mut stack = vec![&self.root];
        let mut result = vec![];
        while let Some(node) = stack.pop() {
            if let Some(node) = node {
                result.push(&node.value);
                stack.push(&node.left);
                stack.push(&node.right);
            }
        }
        result.reverse();
        result
    }

    fn insert(&mut self, _: T) {
        unimplemented!()
    }

    fn remove(&mut self, _: &T) -> Option<T> {
        unimplemented!()
    }

    fn contains(&mut self, _: &T) -> bool {
        unimplemented!()
    }
}

impl<T> BinaryTree<T> {
    pub(crate) fn root_mut(&mut self) -> &mut NodeRef<T> {
        &mut self.root
    }

    pub(crate) fn root(&self) -> &NodeRef<T> {
        &self.root
    }
}

标签：02,node,mut,self,let,二叉树,stack,Rust,result
来源： https://www.cnblogs.com/jiangbo4444/p/16536130.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。