pta 二叉树叶结点值和最大层 c语言链表
已知一棵非空二叉树结点的数据域为不等于0的整数,请编写程序找出该二叉树中叶结点数据值之和最大的层。
输入格式:
输入为一组用空格间隔的整数,个数不超过100个,表示带空指针信息的二叉树先根序列。
输出格式:
输出为一个整数,表示叶结点数据值之和最大的层,如果存在多个满足条件的层,则输出最下面一层。
输入样例1:
1 2 0 0 3 0 0
输出样例1
1
输入样例2:
1 -2 0 0 3 -1 0 0 -1 0 0
输出样例2:
2
完整代码:
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
typedef struct TreeNode* PrtToNode;
typedef struct TreeNode* BinTree;
typedef struct TreeNode {
int data;
int level;
PrtToNode left;
PrtToNode right;
}TreeNode;
int sum[105];//sum[i]是第i层叶子结点的和
int num[105];//num[i]是第i层叶子结点的数量
int maxlevel=0;//二叉树最大层数
BinTree createTree(int level) {
int n;
scanf("%d", &n);
if (n == 0)
return NULL;
BinTree BT = (BinTree)malloc(sizeof(TreeNode));
BT->data = n;
BT->level = level;
if (BT->level > maxlevel)
maxlevel = BT->level;
BT->left = createTree(level+1);
BT->right = createTree(level+1);
return BT;
}//建立二叉树
void InorderTraversal(BinTree BT) {
if (BT == NULL)
return;
if (BT->left == NULL && BT->right == NULL) {
sum[BT->level] += BT->data;
num[BT->level]++;
}
InorderTraversal(BT->left);
InorderTraversal(BT->right);
}
int main() {
BinTree BT = createTree(0);
InorderTraversal(BT);
int maxnum=-1000,level=-1;
for (int i = 0; i <= maxlevel; i++) {
if (sum[i] >= maxnum && num[i]!=0) {
maxnum = sum[i];
level = i;
}
}
printf("%d\n", level);
return 0;
}