第十届蓝桥杯省赛C++A组 完全二叉树的权值（DFS和递归）

试题 G：完全二叉树的权值

时间限制：1.0s     内存限制：256.0MB     本题总分：20 分

【问题描述】

       给定一棵包含 N 个节点的完全二叉树，树上每个节点都有一个权值，按从上到下、从左到右的顺序依次是 A1, A2, ··· AN，如下图所示：

       现在小明要把相同深度的节点的权值加在一起，他想知道哪个深度的节点权值之和最大？如果有多个深度的权值和同为最大，请你输出其中最小的深度。

       注：根的深度是 1。

【输入格式】

       第一行包含一个整数 N。
       第二行包含 N 个整数 A1, A2, ··· AN 。

【输出格式】

       输出一个整数代表答案。

【样例输入】

7
1 6 5 4 3 2 1

【样例输出】

2

【评测用例规模与约定】

       对于所有评测用例，1≤ N ≤100000，−100000≤ Ai ≤100000。

思路：如果学过数据结构的话，这道题就是非常容易的了，也是很容易理解的。主要就是运用了递归和dfs的思想，不断向内递归。先建立一个数组存取每一层的权值之和，遇到要求的层数时，便把它加入到结果对应的数组元素之中，最后累加的结果就是一层的权值之和。再取数组最大元素所在的层数，便是正确的答案了。

具体看代码吧哈哈哈

#include<iostream>
using namespace std;
long long int a[100000], s[100000];
int n;
void dfs(int x, int dep)//深度优先遍历，也是二叉树递归求解
{
	if (x > n)//递归出口，一定要有
		return;
	
	s[dep] = s[dep] + a[x];
	dfs(2 * x, dep + 1);
	dfs(2 * x + 1, dep + 1);
}
int main()
{
	cout << "输入节点数量" << endl;
	cin >> n;
	for (int i = 1; i <= n; i++)
	{
		cin >> a[i];

	}
	memset(s, 0, sizeof(s));
	dfs(1, 1);
	int min = 1;
	for (int i = 2; i < n; i++)
	{
		if (s[i] > s[min])//找最大层
		{
			min = i;
		}
	}
	cout << "深度为" << min << endl;
	cout << s[min] << endl;
	system("pause");
	return 0;

}