cdoj 483 Data Structure Problem DFS

题目链接:

http://acm.uestc.edu.cn/#/problem/show/483

题意:

给你n个数，然后这n个数构成的二叉树，是平衡二叉树还是堆

题解:

dfs
根据题目中给的
堆 (a[x]>=a[x*2] && a[x]>=a[x*2+1]) || (a[x]<=a[x*2] && a[x]<=a[x*2+1]) 最大堆 最小堆
BST (a[x]大于a[x*2]&& a[x]小于a[x*2+1]) || (a[x]小于a[x*2]&&a[x]大于a[x*2+1]) 平衡二叉树 还不会。。是这个条件吗

代码:

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define MS(a) memset(a,0,sizeof(a))
#define MP make_pair
#define PB push_back
const int INF = 0x3f3f3f3f;
const ll INFLL = 0x3f3f3f3f3f3f3f3fLL;
inline ll read(){
    ll x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
//////////////////////////////////////////////////////////////////////////
const int maxn = 2e5+10;

int a[maxn];
int flag1,flag2,flag3,flag4;

void dfs1(int x){
    if(flag1 == 0)
        return ;

    if(a[x*2]!=0){
        if(a[x] >= a[x*2])
            dfs1(x*2);
        else
            flag1 = 0;
    }
    if(a[x*2+1]!=0){
        if(a[x] >= a[x*2+1])
            dfs1(x*2+1);
        else
            flag1 = 0;
    }
}

void dfs2(int x){
    if(flag2 == 0)
        return ;

    if(a[x*2]!=0){
        if(a[x] <= a[x*2])
            dfs2(x*2);
        else
            flag2 = 0;
    }
    if(a[x*2+1]!=0){
        if(a[x] <= a[x*2+1])
            dfs2(x*2+1);
        else
            flag2 = 0;
    }
}

void dfs3(int x){
    if(flag3 == 0)
        return ;

    if(a[x*2]!=0){
        if(a[x]>a[x*2])
            dfs3(x*2);
        else
            flag3 = 0;
    }
    if(a[x*2+1]!=0){
        if(a[x]<a[x*2+1])
            dfs3(x*2+1);
        else
            flag3 = 0;
    }
}

void dfs4(int x){
    if(flag4 == 0)
        return ;

    if(a[x*2]!=0){
        if(a[x]<a[x*2])
            dfs4(x*2);
        else
            flag4 = 0;
    }
    if(a[x*2+1]!=0){
        if(a[x]>a[x*2+1])
            dfs4(x*2+1);
        else
            flag4 = 0;
    }
}

int main(){
    int T=read();
    for(int cas=1; cas<=T; cas++){
        MS(a);
        int n=read();
        for(int i=1; i<=n; i++)
            a[i] = read();

        flag1=1,flag2=1; // 最大堆 最小堆
        flag3=1,flag4=1; // 平衡二叉树
        dfs1(1);
        dfs2(1);
        dfs3(1);
        dfs4(1);

        if((flag1||flag2) && (flag3||flag4))
            cout << "Case #" << cas << ": Both\n";
        else if((flag1||flag2) && !(flag3||flag4))
            cout << "Case #" << cas << ": Heap\n";
        else if(!(flag1||flag2) && (flag3||flag4))
            cout << "Case #" << cas << ": BST\n";
        else 
            cout << "Case #" << cas << ": Neither\n";
    }

    return 0;
}