题目
当本菜鸡补题的时候,发现比赛的时候题意又又又又读错了,一直被题目中的满二叉树和完全二叉树所困扰,一直推不出来样例的答案是怎么来的。
其实这些站点就是满二叉树的叶子结点,明确这一点之后,用一次dfs就可以得到答案了,注意遍历的结点的条件。
#include <iostream>
const int maxn=1e5+10;
int m, n, k, a, b[maxn], step=0, num=0;
void dfs(int root){
step+=1;
if(b[root]==1){
num+=1;
return;
}
if(b[root]==0)
return;
if(b[root]>=2){
if(2*root<=m && num<=k-1)
dfs(2*root);
if(2*root+1<=m && num<=k-1)
dfs(2*root+1);
}
}
int main()
{
scanf("%d%d", &n, &k);
m=2*n-1;
for(int i=0; i<k; i++){
scanf("%d", &a);
int temp=a+n-1;
while(temp){
b[temp]+=1;
temp/=2;
}
}
dfs(1);
printf("%d\n", step);
return 0;
}