C++ 11 手写二叉树中序遍历迭代器 [非递归]
·
本例实现迭代器模式,使用C++11手写二叉树的中序遍历迭代器。
程序结构如下,

test/CMakeLists.txt
cmake_minimum_required(VERSION 2.6)
if(APPLE)
message(STATUS "This is Apple, do nothing.")
set(CMAKE_MACOSX_RPATH 1)
set(CMAKE_PREFIX_PATH /Users/aabjfzhu/software/vcpkg/ports/cppwork/vcpkg_installed/x64-osx/share )
elseif(UNIX)
message(STATUS "This is linux, set CMAKE_PREFIX_PATH.")
set(CMAKE_PREFIX_PATH /vcpkg/ports/cppwork/vcpkg_installed/x64-linux/share)
endif(APPLE)
project(binary_tree_it)
set(CMAKE_CXX_STANDARD 20)
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -Wno-narrowing")
add_definitions(-g)
find_package(ZLIB)
find_package(OpenCV REQUIRED )
find_package(Arrow CONFIG REQUIRED)
find_package(unofficial-brotli REQUIRED)
find_package(unofficial-utf8proc CONFIG REQUIRED)
find_package(Thrift CONFIG REQUIRED)
find_package(glog REQUIRED)
find_package(OpenSSL REQUIRED)
find_package(Boost REQUIRED COMPONENTS
system
filesystem
serialization
program_options
thread
)
find_package(DataFrame REQUIRED)
if(APPLE)
MESSAGE(STATUS "This is APPLE, set INCLUDE_DIRS")
set(INCLUDE_DIRS ${Boost_INCLUDE_DIRS} /usr/local/include /usr/local/iODBC/include /opt/snowflake/snowflakeodbc/include/ ${CMAKE_CURRENT_SOURCE_DIR}/../include/ ${CMAKE_CURRENT_SOURCE_DIR}/../../../include)
elseif(UNIX)
MESSAGE(STATUS "This is linux, set INCLUDE_DIRS")
set(INCLUDE_DIRS ${Boost_INCLUDE_DIRS} /usr/local/include ${CMAKE_CURRENT_SOURCE_DIR}/../include/ ${CMAKE_CURRENT_SOURCE_DIR}/../../../include/)
endif(APPLE)
if(APPLE)
MESSAGE(STATUS "This is APPLE, set LINK_DIRS")
set(LINK_DIRS /usr/local/lib /usr/local/iODBC/lib /opt/snowflake/snowflakeodbc/lib/universal)
elseif(UNIX)
MESSAGE(STATUS "This is linux, set LINK_DIRS")
set(LINK_DIRS ${Boost_INCLUDE_DIRS} /usr/local/lib /vcpkg/ports/cppwork/vcpkg_installed/x64-linux/lib)
endif(APPLE)
if(APPLE)
MESSAGE(STATUS "This is APPLE, set ODBC_LIBS")
set(ODBC_LIBS iodbc iodbcinst)
elseif(UNIX)
MESSAGE(STATUS "This is linux, set LINK_DIRS")
set(ODBC_LIBS odbc odbcinst ltdl)
endif(APPLE)
include_directories(${INCLUDE_DIRS})
LINK_DIRECTORIES(${LINK_DIRS})
file( GLOB test_file_list ${CMAKE_CURRENT_SOURCE_DIR}/*.cpp)
file( GLOB APP_SOURCES ${CMAKE_CURRENT_SOURCE_DIR}/../impl/*.cpp ${CMAKE_CURRENT_SOURCE_DIR}/../include/*.h ${CMAKE_CURRENT_SOURCE_DIR}/../include/*.cpp ${CMAKE_CURRENT_SOURCE_DIR}/../../../include/arr_/impl/*.cpp ${CMAKE_CURRENT_SOURCE_DIR}/../../../include/http/impl/*.cpp ${CMAKE_CURRENT_SOURCE_DIR}/../../../include/yaml/impl/*.cpp ${CMAKE_CURRENT_SOURCE_DIR}/../../../include/df/impl/*.cpp ${CMAKE_CURRENT_SOURCE_DIR}/../../../include/death_handler/impl/*.cpp)
add_library(${PROJECT_NAME}_lib SHARED ${APP_SOURCES} ${test_file})
target_link_libraries(${PROJECT_NAME}_lib ${Boost_LIBRARIES} ZLIB::ZLIB glog::glog DataFrame::DataFrame ${OpenCV_LIBS})
target_link_libraries(${PROJECT_NAME}_lib OpenSSL::SSL OpenSSL::Crypto libgtest.a pystring libyaml-cpp.a libgmock.a ${ODBC_LIBS} libnanodbc.a pthread dl backtrace libzstd.a libbz2.a libsnappy.a re2::re2 parquet lz4 unofficial::brotli::brotlidec-static unofficial::brotli::brotlienc-static unofficial::brotli::brotlicommon-static utf8proc thrift::thrift arrow arrow_dataset)
foreach( test_file ${test_file_list} )
file(RELATIVE_PATH filename ${CMAKE_CURRENT_SOURCE_DIR} ${test_file})
string(REPLACE ".cpp" "" file ${filename})
add_executable(${file} ${test_file})
target_link_libraries(${file} ${PROJECT_NAME}_lib)
endforeach( test_file ${test_file_list})
test/binary_tree_iterator_test.cpp
#include "binary_tree_iterator.hpp"
#include <glog/logging.h>
#include <gtest/gtest.h>
#include <fstream>
#include <memory>
#include "death_handler/death_handler.h"
int main(int argc, char** argv) {
FLAGS_log_dir = "./";
FLAGS_alsologtostderr = true;
// 日志级别 INFO, WARNING, ERROR, FATAL 的值分别为0、1、2、3
FLAGS_minloglevel = 0;
Debug::DeathHandler dh;
google::InitGoogleLogging("./logs.log");
testing::InitGoogleTest(&argc, argv);
int ret = RUN_ALL_TESTS();
return ret;
}
// 中根遍历迭代器
GTEST_TEST(BinaryTreeIteratorTests, BinaryTreeIt) {
BinaryTree<std::string> family (
new Node<std::string>("me",
new Node<std::string>("mother",
new Node<std::string>("Mother's mother"),
new Node<std::string>("Mother's father")
),
new Node<std::string>("father"))
);
for(auto it = family.begin(); it!=family.end(); ++it) {
std::cout << (*it).value << "\n";
}
}
include/binary_tree_iterator.hpp
#ifndef _FREDRIC_BINARY_TREE_ITERATOR_HPP_
#define _FREDRIC_BINARY_TREE_ITERATOR_HPP_
#include <boost/config/warning_disable.hpp>
#include <boost/foreach.hpp>
#include <boost/fusion/include/adapt_struct.hpp>
#include <boost/spirit/include/phoenix_core.hpp>
#include <boost/spirit/include/phoenix_fusion.hpp>
#include <boost/spirit/include/phoenix_object.hpp>
#include <boost/spirit/include/phoenix_operator.hpp>
#include <boost/spirit/include/phoenix_stl.hpp>
#include <boost/spirit/include/qi.hpp>
#include <boost/variant/recursive_variant.hpp>
#include <iostream>
#include <memory>
#include <sstream>
#include <string>
#include <vector>
template <typename T>
struct BinaryTree;
template <typename T>
struct Node {
T value{};
Node<T>* left{nullptr}, *right{nullptr}, *parent{nullptr};
BinaryTree<T>* tree{nullptr};
Node(T value_) : value{value_} {}
Node(T value_, Node<T>* left_, Node<T>* right_)
: value(value_), left(left_), right{right_} {
// 初始化左子节点和右子节点的树为同一颗树
this->left->tree = this->right->tree = tree;
// 初始化左子节点和右子节点的父节点为当前节点
this->left->parent = this->right->parent = this;
}
void set_tree(BinaryTree<T>* t) {
tree = t;
if (left) left->set_tree(t);
if (right) right->set_tree(t);
}
~Node() {
if(left) delete left;
if(right) delete right;
}
};
template <typename T>
struct BinaryTree {
Node<T>* root{nullptr};
BinaryTree(Node<T>* root_) : root(root_) {
root->set_tree(this);
}
~BinaryTree() {
if(root) {
delete root;
}
}
// 中序遍历迭代器
template <typename U>
struct MidOrderIterator {
Node<U>* current;
MidOrderIterator(Node<U>* current_)
: current{current_} {}
bool operator!=(MidOrderIterator<U> const& other) {
return current != other.current;
}
// ++ 操作符重载
MidOrderIterator<U>& operator++() {
// 如果有右子树,就先转到右子树,然后一直寻找他的左子树
if (current->right) {
current = current->right;
while (current->left) {
current = current->left;
}
} else {
// 如果往上走,发现当前节点是右子树,
// 则一直往上走,直到发现当前节点是p的左子树或根节点为止
// 最终把p赋值给current,说明当前节点被赋值为根节点
auto p = current->parent;
while (p && current == p->right) {
current = p;
p = p->parent;
}
current = p;
}
return *this;
}
Node<U>& operator*() { return *current; }
};
using iterator = MidOrderIterator<T>;
iterator begin() {
Node<T>* n = root;
if(n) {
while(n->left) {
n = n->left;
}
}
return iterator{n};
}
iterator end() {
return iterator{nullptr};
}
};
#endif
程序输出如下,

更多推荐


所有评论(0)