算法：如何较为直观地打印二叉树

题目描述

二叉树可以用常规的三种遍历结果来描述其结构，但是不够直观，尤其是二叉树中有重复 值的时候，仅通过三种遍历的结果来构造二叉树的真实结构更是难上加难，有时则根本不可能。 给定一棵二叉树的头节点 head，已知二叉树节点值的类型为 32 位整型，请实现一个打印二叉 树的函数，可以直观地展示树的形状，也便于画出真实的结构。

题目解析

（1）第一步要解决的问题是用什么样的方式可以无歧义的打印二叉树，也就是设计打印的样式。

这里我们是这样做的：假设有二叉树如下

打印出的样式如下：

怎么看呢？

• 首先，将二叉树顺时针选择90°
• 接下来，需要清晰的确定任何一个节点的父节点：
  • 如果一个节点打印结果的前缀和后缀都有"H"（H1H），说明这个节点是头结点，不存放父节点
  • 如果一个节点的前缀和后缀都带有v，表示父节点在该节点所在列的前一列，在该节点所在行的下方，并且是离该节点最近的节点。
    • 上图中v3v、v6v、v7v，父节点分别为H1H、v3v、^4^
  • 如果一个节点的前缀和后缀都带有^，表示父节点在该节点所在列的前一列，在该节点所在行的上方，并且是离该节点最近的节点。
    • 上图中^5^、^2^、4^4，父节点分别为v3v、H1H、^2^

（2）第二个要解决的问题：规定节点打印时占用的统一长度。

• 我们必须规定一个节点在打印时到达占多长
  • 比如，“1”、"433261"等，如果不统一规定打印长度，那么打印一个长短值交替的二叉树必然会出现格式不齐的问题
  • 那么，需要多少呢？整形值(INT32_MIN)占用的长度是11；加上前缀和后缀(H、v、^)之后占用长度为13；另外为了在打印之后更好的区分，再在前面加上两个空格，后面加上两个空格；总共长度是13 + 4 = 17
  • 举个例子，比如v8v长度为3，一共17个字符，那么17-3=14，所以前面补7个空格、后面也补7个空格
  • 举个例子，比如v11v长度为4，一共17个字符，那么17-3=13，所以前面补6个空格、后面补7个空格

（3）怎么实现

• 打印的整体过程是：右子树–>根节点–>左子树

class Solution {
    static std::string getSpace(int num){
        std::string s(num, ' ');
        return s;
    }
    void process(TreeNode *head, int height, const std::string& to, int len){
        if(head == nullptr){
            return;
        }

        process(head->right, height + 1, "v", len);
        std::string val = to + std::to_string(head->val) + to;
        int lenM = val.length();
        int lenL = (len - lenM) / 2;
        int lenR = len - lenM - lenL;
        val = getSpace(lenL) + val + getSpace(lenR);
        std::cout  << getSpace(height * len) << val << "\n";  // getSpace(height * len)用来区分每一层
        process(head->left, height + 1, "^", len);
    }
public:
    void printTree(TreeNode *head){
        process(head, 0, "H", 17);
    }
};