97 lines
2.7 KiB
C++
97 lines
2.7 KiB
C++
//
|
||
// Created by 423A35C7 on 2023-11-12.
|
||
//
|
||
// 11:47
|
||
// 14:55
|
||
|
||
#include <algorithm>
|
||
#include <iostream>
|
||
#include <string>
|
||
#include "bitree.cpp"
|
||
|
||
#define move_and_output(x, y, str) printf("\033[s\033[%d;%dH%s\033u", x, y, str)
|
||
// 移动并输出到指定位置
|
||
#define moveto(x, y) printf("\033[s\033[%d;%dH", x, y)
|
||
// 移动到指定位置
|
||
#define movedown(x) printf("\033[%dB", (x))
|
||
// 向下移动指定行
|
||
#define save_cursor() printf("\033[s")
|
||
// 保存光标位置
|
||
#define restore_cursor() printf("\033[u")
|
||
// 恢复光标位置
|
||
#define clear_char(num) \
|
||
for (int i = 0; i < num; i++) \
|
||
printf(" ")
|
||
// 清空指定位置的字符
|
||
#define clear_screen() printf("\033[2J")
|
||
|
||
// class ViewBox {
|
||
// Node *left_node;
|
||
// int x, y;
|
||
// int width;
|
||
// int height;
|
||
// struct Size {
|
||
// int width;
|
||
// int height;
|
||
// };
|
||
// Size calculate_size(Node *left_node) {
|
||
// Size size {left_node->text.length(), 1};
|
||
// if (left_node->left) {
|
||
// Size increment;
|
||
// increment = calculate_size(left_node->left);
|
||
//
|
||
// }
|
||
// }
|
||
// };
|
||
|
||
enum LINE {NONE, HORIZONTAL, VERTICAL};
|
||
|
||
int get_length(Node *node) {
|
||
// 2是_和空格的长度
|
||
return std::to_string(node->num).length() + 2 + node->text.length();
|
||
}
|
||
|
||
void print(Node *node, int x, int y, LINE line_type) {
|
||
// 移动到指定位置
|
||
moveto(2 * x, y);
|
||
// 保存当前光标位置
|
||
save_cursor();
|
||
// 如果line_type为VERTICAL，则输出“|”
|
||
if (line_type == VERTICAL) {
|
||
std::cout << "|";
|
||
}
|
||
// 恢复光标位置
|
||
restore_cursor();
|
||
// 向下移动1个单位
|
||
movedown(1);
|
||
// 如果line_type为HORIZONTAL，则输出“——”
|
||
if (line_type == HORIZONTAL) {
|
||
std::cout << "——";
|
||
}
|
||
// 输出node的num和text
|
||
std::cout << node->num << "_" << node->text << " ";
|
||
}
|
||
|
||
int output(Node *left_node, int x, int y) {
|
||
int width = 0;
|
||
if (left_node == nullptr) return width;
|
||
// 打印左节点
|
||
print(left_node, x, y, VERTICAL);
|
||
// 计算宽度
|
||
width += std::max(get_length(left_node), output(left_node->left, x + 1, y + width));
|
||
// 遍历右节点
|
||
for (Node *temp = left_node; temp->right && !temp->if_right_chain; temp = temp->right) {
|
||
// 打印右节点
|
||
print(temp->right, x, y + width, HORIZONTAL);
|
||
// 计算宽度
|
||
// 2是——的长度
|
||
width += std::max(2 + get_length(temp->right), output(temp->right->left, x + 1, y + width));
|
||
// width += temp->right->text.length();
|
||
}
|
||
return width;
|
||
// if (left_node->left == nullptr) {
|
||
// return width;
|
||
// }
|
||
// depth += 1;
|
||
// return std::max(width, output(left_node->left, depth));
|
||
} |