PAT_甲级_1155 Heap Paths (30point(s)) (C++)【DFS/大根堆、小根堆判断】

目录

1，题目描述

 题目大意

2，思路

3，AC代码

4，解题过程

---

1，题目描述

Sample Input 1:

8
98 72 86 60 65 12 23 50

 

Sample Output 1:

98 86 23
98 86 12
98 72 65
98 72 60 50
Max Heap

Sample Input 2:

8
8 38 25 58 52 82 70 60

 

Sample Output 2:

8 25 70
8 25 82
8 38 52
8 38 58 60
Min Heap

 

Sample Input 3:

8
10 28 15 12 34 9 8 56

 

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(*￣▽￣*)ブ

完结撒花*★,°*:.☆(￣▽￣)/$:*.°★* 。