算法学习-二叉树节点和两种实现方法，递归非递归

最新推荐文章于 2020-11-08 15:25:46 发布

原创最新推荐文章于 2020-11-08 15:25:46 发布 · 570 阅读

0 ·

CC 4.0 BY-SA版权

算法学习专栏收录该内容

46 篇文章

订阅专栏

本文介绍了一种使用递归及迭代方法计算二叉树所有节点值之和的方法。通过两种不同的实现方式，展示了如何遍历二叉树并计算节点值的总和。

struct Node
{
	Node* m_child[2];
	int m_i;
};



int getSum(struct Node* pRoot)
{
	if (!pRoot)
	{
		return 0;
	}
	return pRoot->m_i + getSum(pRoot->m_child[0]) + getSum(pRoot->m_child[1]);
}

int getSum1(struct Node* pRoot)
{
	std::stack<struct Node*> s;
	struct Node* p = pRoot;
	int nSum = 0;
	while (p != NULL || !s.empty())
	{
		while(p != NULL)
		{
			nSum += p->m_i;
			s.push(p);
			p = p->m_child[0];
		}
		if (!s.empty())
		{
			p = s.top();
			s.pop();
			p = p->m_child[1];
		}
	}
	return pRoot->m_i + getSum(pRoot->m_child[0]) + getSum(pRoot->m_child[1]);
}