N 叉树的层序遍历

#define MAX 10000

int** levelOrder(struct Node* root, int* returnSize, int** returnColumnSizes) {
    if (root == NULL) {
        *returnColumnSizes = NULL;
        *returnSize = 0;
        return NULL;
    }

    struct Node* queue[MAX] = {0};
    memset(queue, 0, sizeof(struct Node*) * MAX);
    int head = 0;
    int tail = 0;

    int** ret = (int**)malloc(sizeof(int*) * MAX);
    memset(ret, 0, sizeof(int*) * MAX);
    *returnSize = 0;
    *returnColumnSizes = (int*)malloc(sizeof(int*) * MAX);

    queue[tail++] = root;
    while (head != tail) {
        int size = tail - head;
        (*returnColumnSizes)[*returnSize] = size;
        ret[*returnSize] = (int*)malloc(sizeof(int) * size);
        for (int i = 0; i < size; i++) {
            ret[*returnSize][i] = queue[head]->val;
            for (int j = 0; j < queue[head]->numChildren; j++) {
                if (queue[head]->children[j] != NULL) {
                    queue[tail++] = queue[head]->children[j];
                }
            }
            head++;
        }
        (*returnSize)++;
    }

    return ret;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值