第五章 树 22 AcWing 1649. 堆路径

第五章 树 22 AcWing 1649. 堆路径

原题链接

AcWing 1649. 堆路径

算法标签

堆 树的遍历

思路

依题意模拟

如何判断构建的树属于大根堆， 小根堆，或不是堆

对于每条输出路径， 若路径上节点i的权值大于节点i-1的权值（即左右子节点权值<根节点权值）， 即为大根堆。 反之， 即为小根堆。否则（即既存在路径上节点i的权值大于节点i-1的权值， 同时也存在路径上节点i的权值小于节点i-1的权值）， 则不是堆。

#pragma GCC optimize(2)
#pragma GCC optimize(3)
#include<bits/stdc++.h>
#define int long long
#define x first
#define y second
#define ump unordered_map
#define pq priority_queue
#define rep(i, a, b) for(int i=a;i<b;++i)
#define Rep(i, a, b) for(int i=a;i>=b;--i)
using namespace std;
typedef pair<int, int> PII;
const int N = 100005;
int n;
int h[N];
vector<int> path;
// gt lt
bool gt, lt;
inline int rd(){
   int s=0,w=1;
   char ch=getchar();
   while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
   while(ch>='0'&&ch<='9') s=s*10+ch-'0',ch=getchar();
   return s*w;
}
void put(int x) {
    if(x<0) putchar('-'),x=-x;
    if(x>=10) put(x/10);
    putchar(x%10^48);
}
void dfs(int u){
	// 当前节点入队
    path.push_back(h[u]);
    if(u*2>n){
        printf("%lld", path[0]);
        rep(i, 1, path.size()){
            printf(" %lld", path[i]);
            if(path[i]<path[i-1]){
                lt=true;
            }else if(path[i]>path[i-1]){
                gt=true;
            }
        }
        puts("");
    }
    if(u*2+1<=n){
        dfs(u*2+1);
    }
    if(u*2<=n){
        dfs(u*2);
    }
    // 恢复现场
    path.pop_back();
}
signed main(){
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	n=rd();
	rep(i, 1, n+1){
	    h[i]=rd();
	}
	dfs(1);
	if(gt&&lt){
	    puts("Not Heap");
	}else if(gt){
	    puts("Min Heap");
	}else{
	    puts("Max Heap");
	}
	return 0;
}

参考文献

AcWing 1649. 堆路径（PAT甲级辅导课）y总视频

原创不易
转载请标明出处
如果对你有所帮助 别忘啦点赞支持哈