พิมพ์ Binary Tree แบบ 2 มิติใน C++

โจทย์

พิมพ์ Binary Tree บนหน้าจอแบบ 2 มิติ (หมุนต้นไม้ 90 องศา ให้รากอยู่ซ้ายสุด ใบขวาสุดอยู่บนสุด)

แนวคิด

• Root อยู่บรรทัดที่ n (ระดับความลึก)
• ระดับใหม่ของต้นไม้อยู่บรรทัด n+i และ n-i
• ห่างจากต้นทาง i tab spaces
• ใบขวาสุดพิมพ์บรรทัดแรก ใบซ้ายสุดพิมพ์บรรทัดสุดท้าย

ตัวอย่างโค้ด C++

#include <bits/stdc++.h>
#include <iostream>
using namespace std;

#define COUNT 10

class Node {
public:
    int data;
    Node* left, *right;
    Node(int data) {
        this->data = data;
        this->left = NULL;
        this->right = NULL;
    }
};

void printTree(Node *root, int space) {
    if (root == NULL) return;
    space += COUNT;
    printTree(root->right, space);
    for (int i = COUNT; i < space; i++) cout << "\t";
    cout << root->data << "\n";
    printTree(root->left, space);
}

int main() {
    Node *root = new Node(43);
    root->left = new Node(25);
    root->right = new Node(67);
    root->left->left = new Node(14);
    root->left->right = new Node(51);
    root->right->left = new Node(26);
    root->right->right = new Node(97);
    root->left->left->left = new Node(81);
    root->left->left->right = new Node(49);
    root->left->right->left = new Node(07);
    root->left->right->right = new Node(31);
    root->right->left->left = new Node(29);
    root->right->left->right = new Node(13);
    root->right->right->left = new Node(59);
    root->right->right->right = new Node(16);
    printTree(root, 0);
    return 0;
}

ผลลัพธ์

        16         97          59      67          13      26          29 43          31      51          7      25          49      14          81