目录
思路
代码
注意点
题目链接:
P8681 [蓝桥杯 2019 省 AB] 完全二叉树的权值
(可跳)
这道题刚看到的时候想着主要就是算出每层2的次方个节点的权值和。
我的思路经过了很多次缝缝补补。创建一个sum数组,下标表示深度,每个元素代表每层的和,变量cnt来记录2的次方(初始值为1),变量x(初始值为1)通过+=得到每一层的起始位置,定义一个数组(其实可以不定义)记录输入的数据,当前输入的数据下标从1开始,当这个下标值等于我们的x的时候,也就是说我们该开始新的一层了的时候,将新算出来的一层的值与上一层算出的进行比较,处理好“如果有多个深度的权值和同为最大,请你输出其中最小的深度。”这个要求,进行答案的更新。注意完全二叉树最后一层不一定是满的,因此在输入数据的循环结束之后,还要对最后一层进行特判。
哎🥀改了好多次缝缝补补还是不能全部通过😂,放过我自己了🥀。看题解了。然后发现了很好的处理方式。
思路
我们根据完全二叉树的性质:每层的数据个数是2的d-1次方,用d表示层数(初始值为1)。那么我们记录下当前输入的数据个数,利用pow函数,计算2的d-1次方,当这两边相等的时候,说明该进入下一层了,再进入下一层之前,我们要先处理好这一层的权值和,做好最大值的更新,以及更新完之后注意将num sum这些值也重新赋值为0,避免影响下一层的计算。
代码
#include<iostream>
#include<algorithm>
#include<cmath>
using namespace std;
int n;
long long sum;
int main()
{
cin>>n;
int x;
int num=0,d=1;//num表示这一层当前读取的数字个数 ,d表示当前层数
long long maxx=-1e18;//maxx表示最大权值和 注意输入可以是负数
int mxid=0;//mxid表示其层数
for(int i=1;i<=n;i++)
{
cin>>x;
sum+=x;
num++;
if(num==pow(2,d-1) || i==n)
{
if(sum>maxx){
maxx=sum;
mxid=d;
}
//当开始进行新的一层运算时
sum=num=0;
d++;
}
}
cout<<mxid;
return 0;
}
注意点
①pow函数头文件#include<cmath>,两个参数:底数和指数
②注意尽量把有可能过大的数据开 long long
③注意这道题输入的数据有可能是负数,因此我们初始的表示最大值的maxx值应该设的足够小
long long maxx=-1e18;//maxx表示最大权值和 注意输入可以是负数
④只用一个if语句就同时解决了完全二叉树最后一层如果不满的情况
只比较当前sum>记录过的maxx,就直接解决了“如果有多个深度的权值和同为最大,请你输出其中最小的深度。”这个要求。感觉还挺妙的。
if(num==pow(2,d-1) || i==n)//i==n处理了最后一层
{
if(sum>maxx){//直接默认了相等关系的时候不修改mxid
maxx=sum;
mxid=d;
}
}
⑤感觉一些函数如果知道的话真是挺方便的。(粘的)
#include<cmath> // 包含了一些常用的数学函数
#include<algorithm> // 包含了一些常用的算法函数
// 数学函数
std::sqrt(x) // 计算x的平方根,x是一个数值类型的参数
std::pow(x, y) // 计算x的y次方,x和y都是数值类型的参数
std::abs(x) // 计算x的绝对值,x是一个数值类型的参数
std::max(x, y) // 返回x和y中的最大值,x和y都是数值类型的参数
std::min(x, y) // 返回x和y中的最小值,x和y都是数值类型的参数
std::sin(x) // 计算x的正弦值,x是一个数值类型的参数
std::cos(x) // 计算x的余弦值,x是一个数值类型的参数
std::tan(x) // 计算x的正切值,x是一个数值类型的参数
std::log(x) // 计算x的自然对数,x是一个数值类型的参数
std::exp(x) // 计算e的x次方,x是一个数值类型的参数
// algorithm头文件中的函数
std::sort(begin, end) // 对序列进行排序,begin和end是指向序列开始和结束的迭代器
std::reverse(begin, end) // 反转序列,begin和end是指向序列开始和结束的迭代器
std::find(begin, end, value) // 在序列中查找值,begin和end是指向序列开始和结束的迭代器,value是要查找的值
std::count(begin, end, value) // 计算序列中某个值出现的次数,begin和end是指向序列开始和结束的迭代器,value是要计数的值
std::replace(begin, end, old_value, new_value) // 将序列中的某个值替换为新的值,begin和end是指向序列开始和结束的迭代器,old_value是要被替换的值,new_value是新的值
写到这里。
有问题欢迎指出,一起加油!!