ch6_2.cc

Go to the documentation of this file.

 
#include "binaryTree.hh"
#include <ctime>
#include <fstream>
#include <string>
 
static bool GenTreeData(const char *_file_path = "./test0.txt", const char *empty_flag = "@");
 
template <typename T, typename Comparator>
static void print_test_result(const Tree::binaryTree<T, Comparator> &binary_tree, const typename Tree::binaryTree<T, Comparator>::value_type &flag, std::ostream &out = std::cout);
 
int main(int argc, char const *argv[])
{
    // time
    clock_t t = clock();
    time_t rawtime;
    struct tm *timeinfo;
 
    time(&rawtime);                 // Get the current calendar time
    timeinfo = localtime(&rawtime); // Convert time_t to tm as local time
 
    // 获取文件路径，参考：http://www.cplusplus.com/reference/string/string/find_last_of/
    const std::string full_path_exec{argv[0]};
    std::string::size_type found = full_path_exec.find_last_of("/\\", std::string::npos);
    const std::string exec_path = full_path_exec.substr(0, found + 1);
    const std::string exec_filename = full_path_exec.substr(found + 1, std::string::npos);
    const std::string data_file_name("ch6_2.result");
 
    // 生成数据文件，然后从文件中读取数据，依此建立一棵二叉树
    std::string data_file_full_path{exec_path + data_file_name}; // 数据文件的绝对地址
 
    // 生成数据文件
    if (!GenTreeData(data_file_full_path.c_str(), "@"))
    {
        std::cout << "文件：" << data_file_full_path << "生成失败！\n";
        return 1;
    }
 
    // 读数据文件
    std::ifstream fin(data_file_full_path.c_str(), std::ios_base::in);
    if (fin.fail())
    {
        std::cout << "文件：" << data_file_full_path << "读取失败！\n";
        return 1;
    }
 
    // 建立二叉树
    Tree::binaryTree<std::string> binary_tree{};
    binary_tree.createTree("@", fin);
    fin.close();
 
    // 准备将结果写入数据文件
    std::ofstream fout(data_file_full_path.c_str(), std::ios_base::app);
    if (fout.fail())
    {
        std::cerr << "文件：" << data_file_full_path << "写入失败！\n";
        return 1;
    }
 
    // printf("Current local time and date: %s\n", asctime(timeinfo)); // Convert tm structure to string
    std::cout << "\nCurrent local time and date: " << asctime(timeinfo) << '\n';
    fout << "\nCurrent local time and date: " << asctime(timeinfo) << '\n';
 
    print_test_result(binary_tree, "@", std::cout);
    print_test_result(binary_tree, "@", fout);
 
    // ch6_1
    binary_tree.swaplr();
    std::cout << "\n交换左右结点后：\n";
    fout << "\n交换左右结点后：\n";
    print_test_result(binary_tree, "@", std::cout);
    print_test_result(binary_tree, "@", fout);
 
    t = clock() - t;
    // printf("\nIt took me %ld clicks (%f seconds).\n", t, ((float)t) / CLOCKS_PER_SEC);
    std::cout << "\nIt took me " << t << " clicks (" << ((float)t) / CLOCKS_PER_SEC << " seconds).\n";
    fout << "\nIt took me " << t << " clicks (" << ((float)t) / CLOCKS_PER_SEC << " seconds).\n";
 
    fout.close();
    system("pause");
    return 0;
}
 
bool GenTreeData(const char *_file_path, const char *empty_flag)
{
    std::string file_path(_file_path);
 
    // 调用DOS命令，询问是否删除文件
    std::string cmd = std::string("DEL /P \"") + file_path + '"';
    system("@echo on");
    system("echo We are trying to delete some files, which will be created later.");
    system("pause");
    system(cmd.c_str());
 
    // 检查文件是否存在且可读
    std::ifstream fin(file_path.c_str(), std::ios_base::in);
    if (fin.good())
    {
        std::cerr << "文件已存在！\n";
        fin.close();
 
        system("pause");
        return true;
    }
 
    // 文件不存在或不可读，尝试新建
    std::ofstream fout(file_path.c_str(), std::ios_base::out);
    if (fout.fail())
    {
        std::cerr << "无写权限，文件生成失败！\n";
 
        system("pause");
        return false;
    }
    fout << "A\n"
         << "L\tC\n"
         << "B\tE\t" << empty_flag << "\tD\n"
         << empty_flag << '\t' << empty_flag << '\t' << empty_flag << '\t' << empty_flag << '\t' << "W\t" << empty_flag << '\n'
         << empty_flag << "\tX\n"
         << empty_flag << '\t' << empty_flag << std::endl;
 
    // 检查是否生成成功
    if (fout.good())
    {
        system("echo File created successfully!");
        fout.close();
        system("pause");
        return true;
    }
 
    std::cerr << "无法生成文件！\n";
    system("pause");
    return false;
}
 
template <class T, typename Comparator>
void print_test_result(const Tree::binaryTree<T, Comparator> &binary_tree, const typename Tree::binaryTree<T, Comparator>::value_type &flag, std::ostream &out)
{
    out << "二叉树的规模(递归 非递归)：\n"
        << binary_tree.size() << ' ' << binary_tree.size_loop();
    out << "\n二叉树的高(深)度(递归 非递归)，从0起：\n"
        << binary_tree.height() << ' ' << binary_tree.height_loop();
    // ch6_2
    out << "\n度为2的结点的个数是：\n"
        << binary_tree.CountDegreeTwo();
    out << "\n前序遍历(递归)：\n";
    binary_tree.preOrder(out);
    out << "\n前序遍历(非递归)：\n";
    binary_tree.preOrder_loop(out);
    out << "\n中序遍历(递归)：\n";
    binary_tree.inOrder(out);
    out << "\n中序遍历(非递归)：\n";
    binary_tree.inOrder_loop(out);
    out << "\n后序遍历(递归)：\n";
    binary_tree.postOrder(out);
    out << "\n后序遍历(非递归)：\n";
    binary_tree.postOrder_loop(out);
    out << "\n层次遍历：\n";
    binary_tree.levelOrder(out);
    out << "\n层次打印(调用lchild(), rchild(), root()等API)：\n";
    Tree::printBinaryTree(binary_tree, flag, out);
}