5430: 数据结构实验：左叶子之和

5430: 数据结构实验：左叶子之和

时间限制(普通/Java):1000MS/3000MS     内存限制:65536KByte
总提交: 117            测试通过:43

描述

求一棵二叉树所有左叶子节点元素值之和。

二叉树节点定义如下：

struct TreeNode {

    int val;

    struct TreeNode *left;

    struct TreeNode *right;

};

题目部分代码已经完成，您只需要补充并提交以下函数：

int sumOfLeftLeaves(struct TreeNode* root);

输入

输入为若干个整数（不超过512）表示一棵二叉树顺序表示时的结点元素值，其中0表示二叉树对应结点为空。输入以-1结束。 

输出

输出所有左叶子节点元素值之和。

样例输入

1 2 3 4 5 0 0 0 0 6 7 -1

样例输出

10

题目来源

TOJ

#include <stdio.h>

struct TreeNode
{
	int val;
	struct TreeNode *left, *right;
};

struct TreeNode a[520];

int sum = 0;

struct TreeNode* Creat(struct TreeNode* root)
{
	int n, i, j, k = 0;
	while (scanf("%d", &n), n != -1)
		a[k++].val = n;
	for (i = 0, j = 1; i < k; i++, j++)
	{
		if ((i + j >= k) || (a[i + j].val == 0)) a[i].left = NULL;
		else a[i].left = &a[i + j];
		if ((i + i + 1 >= k) || (a[i + j + 1].val == 0)) a[i].right = NULL;
		else a[i].right = &a[i + j + 1];
	}
	return &a[0];
}

void dfs(struct TreeNode* root)
{
	if (!root) return;
	if (root->left)
	{
		if (root->left->left == NULL&&root->left->right == NULL)
			sum += root->left->val;
		dfs(root->left);
	}
	if (root->right) dfs(root->right);
}

int sumOfLeftLeaves(struct TreeNode* root)
{
	dfs(root);
	return sum;
}

int main()
{
	struct TreeNode *root = NULL;
	root=Creat(root);
	printf("%d\n", sumOfLeftLeaves(root));
	return 0;
}