1155 Heap Paths DFS+stack

最新推荐文章于 2025-08-05 19:50:13 发布

楷楷大人

最新推荐文章于 2025-08-05 19:50:13 发布

阅读量286

点赞数

CC 4.0 BY-SA版权

文章标签：深度优先算法 c++

本文链接：https://blog.youkuaiyun.com/qq_43606316/article/details/124789045

该代码实现了一棵完全二叉树的构造，并通过深度优先搜索（DFS）进行遍历。同时，它判断了树是否为最大堆或最小堆。输入整数n和节点值，程序会输出树的类型：最大堆、最小堆或非堆。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

代码逻辑很简单
构造一颗完全二叉树再DFS

#include <cstdio>
#include <stack>
#include <algorithm>
#include <vector>
#include <map>
#include <iostream>
#include <cmath>
using namespace std;
int n;
struct Node
{
    int value;
    Node *lchild, *rchild;
    Node()
    {
        lchild = rchild = NULL;
    }
};
Node N[1010];
int type = 0;

void show(vector<Node> vs)
{
    if (!vs.empty())
    {
        for (int i = 0; i < vs.size(); i++)
        {
            printf("%d", vs[i].value);
            if (i != vs.size() - 1)
            {
                printf(" ");
            }
        }
        printf("\n");
    }
}
vector<Node> s;
void DFS(int index)
{
    if (index > n)
    {
        if (index % 2 == 0)//左孩子都没了 才输出哦
        {
            show(s);
        }
        return;
    }
    s.push_back(N[index]);
    DFS(index * 2 + 1);
    DFS(index * 2);
    s.pop_back();
}
void isMax()
{
    for (int i = 1; i <= n; i++)
    {
        if (i * 2 < n)
        {
            if (N[i * 2].value > N[i].value || N[i * 2 + 1].value > N[i].value)
            {
                type = -1;
                break;
            }
        }
        if (i == n)
        {
            type = 1;
        }
    }
}
void isMin()
{
    for (int i = 1; i <= n; i++)
    {
        if (i * 2 < n)
        {
            if (N[i * 2].value < N[i].value || N[i * 2 + 1].value < N[i].value)
            {
                type = -1;
                break;
            }
        }
        if (i == n)
        {
            type = 2;
        }
    }
}
int main()
{
    scanf("%d", &n);
    for (int i = 1; i <= n; i++)
    {
        scanf("%d", &N[i].value);
        if (i * 2 <= n)
        {
            N[i].lchild = &N[i * 2];
        }
        if (i * 2 + 1 <= n)
        {
            N[i].rchild = &N[i * 2 + 1];
        }
    }
    isMax();
    if (type == -1)
    {
        isMin();
    }
    if (type == -1)
    {
        type = 0;
    }
    DFS(1);
    if (type == 1)
    {
        printf("Max Heap\n");
    }
    else if (type == 2)
    {
        printf("Min Heap\n");
    }
    else
    {
        printf("Not Heap\n");
    }

    return 0;
}