You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
147 lines
3.1 KiB
147 lines
3.1 KiB
/**
|
|
* File: print_util.h
|
|
* Created Time: 2022-12-21
|
|
* Author: MolDum (moldum@163.com)、Reanon (793584285@qq.com)
|
|
*/
|
|
|
|
#ifndef PRINT_UTIL_H
|
|
#define PRINT_UTIL_H
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
|
|
#include "list_node.h"
|
|
#include "tree_node.h"
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
/* Print an Array */
|
|
static void printArray(int arr[], int size) {
|
|
printf("[");
|
|
if (arr != NULL && size != 0) {
|
|
for (int i = 0; i < size - 1; i++) {
|
|
if (arr[i] != INT_MAX) {
|
|
printf("%d, ", arr[i]);
|
|
} else {
|
|
printf("NULL, ");
|
|
}
|
|
}
|
|
if (arr[size - 1] != INT_MAX) {
|
|
printf("%d]\n", arr[size - 1]);
|
|
} else {
|
|
printf("NULL]\n");
|
|
}
|
|
} else {
|
|
printf("]");
|
|
}
|
|
}
|
|
|
|
/* Print an Array */
|
|
static void printArrayFloat(float arr[], int size) {
|
|
printf("[");
|
|
if (arr != NULL && size != 0) {
|
|
for (int i = 0; i < size - 1; i++) {
|
|
if (arr[i] != INT_MAX) {
|
|
printf("%.2f, ", arr[i]);
|
|
} else {
|
|
printf("NULL, ");
|
|
}
|
|
}
|
|
if (arr[size - 1] != INT_MAX) {
|
|
printf("%.2f]\n", arr[size - 1]);
|
|
} else {
|
|
printf("NULL]\n");
|
|
}
|
|
} else {
|
|
printf("]");
|
|
}
|
|
}
|
|
|
|
/* Print a linked list */
|
|
static void printLinkedList(ListNode *node) {
|
|
if (node == NULL) {
|
|
return;
|
|
}
|
|
while (node->next != NULL) {
|
|
printf("%d -> ", node->val);
|
|
node = node->next;
|
|
}
|
|
printf("%d\n", node->val);
|
|
}
|
|
|
|
struct Trunk {
|
|
struct Trunk *prev;
|
|
char *str;
|
|
};
|
|
|
|
typedef struct Trunk Trunk;
|
|
|
|
Trunk *newTrunk(Trunk *prev, char *str) {
|
|
Trunk *trunk = (Trunk *)malloc(sizeof(Trunk));
|
|
trunk->prev = prev;
|
|
trunk->str = (char *)malloc(sizeof(char) * 10);
|
|
strcpy(trunk->str, str);
|
|
return trunk;
|
|
}
|
|
|
|
/* Helper function to print branches of the binary tree */
|
|
void showTrunks(Trunk *trunk) {
|
|
if (trunk == NULL) {
|
|
return;
|
|
}
|
|
showTrunks(trunk->prev);
|
|
printf("%s", trunk->str);
|
|
}
|
|
|
|
/* Help to print a binary tree, hide more details */
|
|
static void printTreeHelper(TreeNode *node, Trunk *prev, bool isLeft) {
|
|
if (node == NULL) {
|
|
return;
|
|
}
|
|
char *prev_str = " ";
|
|
Trunk *trunk = newTrunk(prev, prev_str);
|
|
printTreeHelper(node->right, trunk, true);
|
|
if (prev == NULL) {
|
|
trunk->str = "———";
|
|
} else if (isLeft) {
|
|
trunk->str = "/———";
|
|
prev_str = " |";
|
|
} else {
|
|
trunk->str = "\\———";
|
|
prev->str = prev_str;
|
|
}
|
|
showTrunks(trunk);
|
|
printf("%d\n", node->val);
|
|
|
|
if (prev != NULL) {
|
|
prev->str = prev_str;
|
|
}
|
|
trunk->str = " |";
|
|
|
|
printTreeHelper(node->left, trunk, false);
|
|
}
|
|
|
|
/* Print a binary tree */
|
|
static void printTree(TreeNode *root) {
|
|
printTreeHelper(root, NULL, false);
|
|
}
|
|
|
|
/* Print a Heap */
|
|
static void printHeap(int arr[], int size) {
|
|
TreeNode *root;
|
|
printf("堆的数组表示：");
|
|
printArray(arr, size);
|
|
printf("堆的树状表示：\n");
|
|
root = arrToTree(arr, size);
|
|
printTree(root);
|
|
}
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|
|
|
|
#endif // PRINT_UTIL_H
|