PAT (Advanced Level) 1115. Counting Nodes in a BST (30)

简单题。统计一下即可。

#include<cstdio>
#include<cstring>
#include<cmath>
#include<vector>
#include<map>
#include<queue>
#include<stack>
#include<algorithm>
using namespace std;

const int maxn=+;
struct Node
{
    int left;
    int right;
    int val;
    int dep;
} s[maxn];

int n;
int a[maxn];
int max_dep,n1,n2;

void dfs(int x,int dep)
{
    max_dep=max(max_dep,dep);
    s[x].dep=dep;
    if(s[x].left!=-) dfs(s[x].left,dep+);
    if(s[x].right!=-) dfs(s[x].right,dep+);
}

void DFS(int x)
{
    if(s[x].dep==max_dep) n1++;
    else if(s[x].dep==max_dep-) n2++;
    if(s[x].left!=-) DFS(s[x].left);
    if(s[x].right!=-) DFS(s[x].right);
}

int main()
{
    scanf("%d",&n);

    if(n==) printf("0 + 0 = 0\n");
    else
    {
        for(int i=; i<=n; i++) scanf("%d",&a[i]);

        for(int i=; i<=n; i++) s[i].left=s[i].right=-;
        int id=;
        s[id++].val=a[];

        for(int i=; i<=n; i++)
        {
            int now=;
            while()
            {
                if(a[i]<=s[now].val)
                {
                    if(s[now].left!=-) now=s[now].left;
                    else
                    {
                        s[now].left=id;
                        s[id++].val=a[i];
                        break;
                    }
                }

                else
                {
                    if(s[now].right!=-) now=s[now].right;
                    else
                    {
                        s[now].right=id;
                        s[id++].val=a[i];
                        break;
                    }
                }
            }
        }

        max_dep=n1=n2=;
        dfs(,);
        DFS();
        printf("%d + %d = %d\n",n1,n2,n1+n2);
    }
    return ;
}