以二叉树性质遍历数组求取各路径之和 ~ 递归实现

小强~

已于 2022-09-10 23:25:53 修改

阅读量454

点赞数

分类专栏： c++ 文章标签：算法 c++ 开发语言

于 2022-09-10 18:56:15 首次发布

本文链接：https://blog.csdn.net/m0_51961114/article/details/126798519

版权

c++ 专栏收录该内容

127 篇文章 0 订阅

订阅专栏

以二叉树性质遍历数组求取各路径之和 ~ 递归实现

题目中　 以二叉树性质遍历　的意思是，从索引 x 出发，下一个位置只能是 2 * x + 1 或 2* x + 2.或者说二叉树以数组形式保存，然后找出各路径之和的最大值。
假定当前数组为 { 0, 2, 5, 2, 4 }，则该二叉树所有路径之和如下
在这里插入图片描述

如图，上述二叉树有3条路径，1路径和为4，2路径和为6，3路径和为5.
所以路径和最大是6.

程序实现如下：

#include<iostream>
#include<vector>
using namespace std;

void sumvec(vector<int>& arr, vector<int>& sum, int idx, int tmp)
{
	if (idx >= arr.size())
		return 0;
	tmp += arr[idx];
	sum.push_back(tmp);
	sumvec(arr, sum, idx * 2 + 1, tmp);
	sumvec(arr, sum, idx * 2 + 2, tmp);
}

int findNum(vector<int>& arr)
{
	vector<int> sum;// sum保存所有路径之和，路径条数为：(arr.size() / 2 + arr.size() % 2) = 2+1 = 3
	sumvec(arr, sum, 0, 0);// 运行结束后，sum = {0, 2, 4, 6, 5}其中后面3各数字对应为每条路径和
	int max = sum[0];
	for (int i = 0; i < sum.size(); i++)
	{
		if (max < sum[i])
			max = sum[i];
	}
	return max;//max = 6
}

int main()
{
	vector<int> arr = { 0, 2, 5, 2, 4 };
	cout << findNum(arr) << endl;
	return 0;
}