PAT_甲级_1155 Heap Paths (30point(s)) (C++)【DFS/大根堆、小根堆判断】
原创
©著作权归作者所有:来自51CTO博客作者再见萤火虫IT的原创作品,请联系作者获取转载授权,否则将追究法律责任
目录
1,题目描述
题目大意
2,思路
3,AC代码
4,解题过程
1,题目描述
Sample Input 1:
Sample Output 1:
98 86 23
98 86 12
98 72 65
98 72 60 50
Max Heap
Sample Input 2:
Sample Output 2:
8 25 70
8 25 82
8 38 52
8 38 58 60
Min Heap
Sample Input 3:
Sample Output 3:
10 15 8
10 15 9
10 28 34
10 28 12 56
Not Heap
题目大意
判断一棵完全二叉树为大根堆、小根堆,或者不是堆。
2,思路
(看见完全二叉树我就笑了)
一个DFS解决所有烦恼!
注意,我这里用unordered_set<int> record记录是否为堆,以及是哪种堆:每次到达叶节点时,输出ans序列,并判断相邻两个值的大小关系,大于插入1、小于插入-1(record.insert(ans[i] > ans[i+1] ? 1 : -1))。最后判断record的size,不等于1说明不是堆,否则根据*record.begin()判断是哪种堆。
3,AC代码
#include<bits/stdc++.h>
using namespace std;
int N, data[1005], last = -1;
vector<int>ans;
unordered_set<int> record;
void dfs(int root){
if(root > N) return; // !!!出口条件
ans.push_back(data[root]);
if(2 * root > N && 2 * root + 1 > N){ // !!!判断为叶节点的条件
for(int i = 0; i < ans.size(); i++){
printf("%d%c", ans[i], i == ans.size()-1 ? '\n' : ' ');
if(i < ans.size()-1)
record.insert(ans[i] > ans[i+1] ? 1 : -1); //记录是否为堆
}
}
dfs(2 * root + 1);
dfs(2 * root);
ans.pop_back(); // 注意出栈
}
int main(){
#ifdef ONLINE_JUDGE
#else
freopen("1.txt", "r", stdin);
#endif // ONLINE_JUDGE
scanf("%d", &N);
for(int i = 1; i <= N; i++)
scanf("%d", &data[i]);
dfs(1);
if(record.size() > 1) printf("Not Heap");
else if(*record.begin() > 0) printf("Max Heap");
else printf("Min Heap");
return 0;
}
4,解题过程
一发入魂o(* ̄▽ ̄*)ブ
完结撒花*★,°*:.☆( ̄▽ ̄)/$:*.°★* 。