3537_树查找(完全二叉树的深度)(7/4)

题目https://www.acwing.com/problem/content/3540/

给定一棵包含 n 个结点（编号 1∼n）的完全二叉树的层序遍历序列，请按照从左到右的顺序输出该树第 k层的全部结点编号。

输入格式

第一行包含整数 n。

第二行包含 n 个整数，表示该二叉树的层序遍历序列。

第三行包含整数 k。

输出格式

共一行，按照从左到右的顺序输出该树第 k 层的全部结点编号。

数与数之间用单个空格隔开。

若无该层结点，则输出 EMPTY。

数据范围

1≤n≤1000，
1≤k≤20

输入样例：

4
1 2 3 4
2

输出样例：

2 3

     Core:    深度=(lg(2)N] +1

Answer1:

#include<bits/stdc++.h>
#define ll long long
#define inf 0x3f3f3f3f
using namespace std;
const int maxx=1009;
vector<int> v[30];
int a[maxx];
int main()
{
    //freopen("input.txt","r",stdin);
    int n,k;
    scanf("%d",&n);

    for(int i=1; i<=n; i++)
    {
        scanf("%d",&a[i]);
        v[(int)(log(i*1.0)/log(2.0))+1].push_back(a[i]);
    }

    scanf("%d",&k);
    if(!v[k].size())
        printf("EMPTY\n");
    else
    {
        for(int i=0; i<v[k].size(); i++)
            printf("%d ",v[k][i]);
    }

    return 0;
}

Answer2：

 
    int L = (int)pow(2, k-1);
    int R = (int)pow(2, k)-1;//完全二叉树中第k层的左右边界
    if(L > n)
    {
        cout << "EMPTY";
        return 0;
    }
    for(int i=L; i<=min(n,R); i++)//有可能是中间层和最后一层,取个最小值
      printf("%d ",a[i]);