1735: 树查找 完全二叉树

#include<bits/stdc++.h>
using namespace std;
int main()
{
    while(1){
        int n;
        cin>>n;
        if(n==0) return 0;
        vector<int> ve(n+1);
        for(int i=1;i<=n;i++) cin>>ve[i];
        int d;
        cin>>d;
        int sum=pow(2,d-1);
        if(n>=sum){
            int f=0;
            int idx=pow(2,d-1);   //第k层的第一个结点
            int ide=pow(2,d)-1;  //第k层的最后一个结点
            //第k层可能是最后一层，而最后一层可能没满
            for(int i=idx;i<=ide;i++){
                if(ve[i]!=0&&f==0){
                    cout<<ve[i];
                    f=1;
                }else if(ve[i]!=0) cout<<" "<<ve[i];
            }
            cout<<endl;
            
        }else cout<<"EMPTY\n";
    }
    return 0;
}

解题思路:

注意事项:

参考代码: