二叉树-BM27 按之字形顺序打印二叉树

最新推荐文章于 2024-10-16 22:24:44 发布

jiqian4953

最新推荐文章于 2024-10-16 22:24:44 发布

阅读量81

点赞数

分类专栏：刷题文章标签：数据结构算法链表

本文链接：https://blog.csdn.net/jiqian4953/article/details/129017932

版权

刷题专栏收录该内容

34 篇文章 0 订阅

订阅专栏

该算法使用队列和双指针实现二叉树的之字形层序遍历。首先初始化一个队列，将根节点入队，然后在循环中处理每一层的节点，按照之字形顺序收集节点值。每次遍历完一层，切换左右方向标志。返回的结果是一个二维数组，包含每一层的节点值。

摘要由CSDN通过智能技术生成

BM27 按之字形顺序打印二叉树

给定一个二叉树，返回该二叉树的之字形层序遍历，（第一层从左向右，下一层从右向左，一直这样交替）

解法(队列+双指针)

/**
 * struct TreeNode {
 *  int val;
 *  struct TreeNode *left;
 *  struct TreeNode *right;
 * };
 */
/**
 * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
 *
 *
 * @param root TreeNode类
 * @return int整型二维数组
 * @return int* returnSize 返回数组行数
 * @return int** returnColumnSizes 返回数组列数
 */
typedef struct TreeNode* pnode;

int** Print(struct TreeNode* root, int* returnSize,  int** returnColumnSizes ) {
    // write code here
    if (root == NULL) {
        return NULL;
    }
    bool right_di = false;
    pnode queue[1500];
    int front = 0;
    int rear = 0;
    queue[rear++] = root;

    int** ans = (int**)malloc(1500 * sizeof(int*));
    *returnSize = 0;
    *returnColumnSizes = (int*)malloc(1500 * sizeof(int));

    while (front != rear) {
        int prerear = rear;
        int dcreas_rear = rear;
        int k = 0;
        ans[*returnSize] = (int*)malloc((prerear - front) * sizeof(int));
        while (front < prerear) {
            pnode p = queue[front++];
            ans[*returnSize][k++] = right_di?queue[--dcreas_rear]->val:p->val;
            if (p->left != NULL) {
                queue[rear++] = p->left;
            }
            if (p->right != NULL) {
                queue[rear++] = p->right;
            }
        }
        right_di=!right_di;
        (*returnColumnSizes)[*returnSize] = k;
        (*returnSize)++;
    }
    return ans;
}