1155 Heap Paths (30 分)

1155 Heap Paths (30 分)

题目大意

给定完全二叉树层序遍历得到的序列，输出每条叶子结点到根结点的路径（从右向左输出）。

基本思路

判断属于什么堆思路这里就不写了，请看A1147的思路。这里主要说明一下如何从右向左输出每一条路径：
定义一个数组depthc存放每个叶子结点的层次，index代表最高高度和第二高高度的临界叶子结点的编号，函数solve用来求路径（逆序），cnt存放当前是第几条路径，vector path[maxn]存放第cnt条路径的所有结点（逆序存放）

• 求出每个叶子节点的层次。
• 从叶子结点n遍历到叶子结点half+1，找到那个临界结点index。如果没有临界结点，index置为n。
• 根据叶子结点是否有层次进行solvepath，求出所有路径放进二维数组path[cnt]内。

代码

具体数据结构和详细思路请看以下代码：

#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
const int maxn=1010;
int n;
int data[maxn];
int flag1,flag2;//是否是大、小根堆
int cnt=0;//记录总的路径条数
vector<int> path[maxn/2];//记录下所有路径
void solvepath(int n,int m){
    for(int i=n;i>=m+1;i--){
        int j=i;
        while(j>=1){
            path[cnt].push_back(data[j]);
            j=j/2;
        }
        cnt++;
    }
}
int main(){
    cin>>n;
    flag1=1;
    flag2=1;
    for(int i=1;i<=n;i++) cin>>data[i];
    int half=n/2;
    //求出最高层次和最高层次减一之间临界结点的下标
    int depth[maxn];
    for(int i=n;i>=half+1;i--){
        int j=i;
        int depth1=1;
        while(j>=1){
            j/=2;
            depth1++;
        }
        depth[i]=depth1;
    }
    int index;
    if(depth[half+1]==depth[n]){
        index=n;
    }else{
        int mindepth=depth[n];
        for(int i=n;i>=half+1;i--){
            if(depth[i]<mindepth){
                index=i;
                break;
            }
        }
    }
    //求出所有路径
    if(index==n){
        solvepath(n,half);
    }else{
        solvepath(index,half);
        solvepath(n,index);
    }
    //输出所有路径
    for(int i=0;i<cnt;i++){
        for(int j=path[i].size()-1;j>=0;j--){
            if(j!=0){
                cout<<path[i][j]<<' ';
            }else{
                cout<<path[i][j]<<endl;
            }
        }
    }
    //判断是否是大根堆
    for(int i=half;i>=1;i--){
        int j=2*i;
        if(j+1<=n&&data[j+1]>data[j]) j=j+1;
        if(data[i]<data[j]){
            flag1=0;
            break;
        }
    }
    if(flag1){
        cout<<"Max Heap"<<endl;
        return 0;
    }
    //判断是否是小根堆
    for(int i=half;i>=1;i--){
        int j=2*i;
        if(j+1<=n&&data[j+1]<data[j]) j=j+1;
        if(data[i]>data[j]){
            flag2=0;
            break;
        }
    }
    if(flag2){
        cout<<"Min Heap"<<endl;
    }
    //不是大根堆也不是小根堆，那它是非根堆
    if(flag1==0&&flag2==0){
        cout<<"Not Heap"<<endl;
    }
}

---

以上的方法是我自己的理解做出来的，后来我去看了柳神的代码，用的dfs+回溯不到30行代码就解决问题了，感觉很震惊，以下是它的链接：
添加链接描述

基本思路

• 深搜打印所有路径（从右往左，先序的镜像）,vector数组v保存一路上的结点，通过push_back和pop_back()回溯，维护路径。
• 上面这步如果把2index+1和2index的部分全部交换就是我们平时常用的回溯法（从左往右，先序）
• 判断是否为堆：从第二个节点开始遍历，如果比父结点小，就不是小根堆；如果比父结点大，就不是大根堆。
• index * 2 > n能够得到出叶子结点和空结点的情况
• index<=n是为了防止空结点的情况

代码

#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
int n;
int a[1010],isMin=1,isMax=1;
vector<int> v;//采用回溯法存放一条路径，每到叶子结点输出这条路径并回溯
void dfs(int index){
    if(index*2>n){//是叶子结点或者空结点
        if(index<=n){//只能是叶子结点
            for(int i=0;i<v.size();i++){
                printf("%d%s",v[i],i==v.size()-1?"\n":" ");
            }
        }
    }else{
        v.push_back(a[2*index+1]);
        dfs(2*index+1);
        v.pop_back();
        v.push_back(a[2*index]);
        dfs(2*index);
        v.pop_back();
//         for(int i=2*index+1;i>=2*index;i--){
//             v.push_back(a[i]);
//             dfs(i);
//             v.pop_back();
//         }
    }
}
int main(){
    cin>>n;
    for(int i=1;i<=n;i++) cin>>a[i];
    v.push_back(a[1]);//根结点先压入数组
    dfs(1);//从根结点编号1开始dfs
    for(int i=2;i<=n;i++){
        if(a[i]<a[i/2]) isMin=0;
        if(a[i]>a[i/2]) isMax=0;
    }
    if(isMin) cout<<"Min Heap"<<endl;
    else if(isMax) cout<<"Max Heap"<<endl;
    else cout<<"Not Heap"<<endl;
}