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;
}