1115 Counting Nodes in a BST二叉排序树

#include<bits/stdc++.h>
using namespace std;
#define rep(x,y,z) for(int (x)=(y);(x)<(z);i++)
#define ll long long
const int maxn=1e4+10;
int n,tot=0,x,dat[maxn],L[maxn],R[maxn];
void Insert(int rt,int x){
    if(dat[rt]<x){
        if(R[rt]==-1){
            R[rt]=tot,dat[tot]=x;
            L[tot]=R[tot]=-1;
            tot++;
        }	
        else Insert(R[rt],x);
    }else{
        if(L[rt]==-1){
            L[rt]=tot,dat[tot]=x;
         	L[tot]=R[tot]=-1;
            tot++;
        }
        else Insert(L[rt],x);
    }
}
vector<int> g[maxn];
int dp=-1;
void dfs(int x,int dep){
    if(x==-1) return ;
    dp=max(dp,dep);
    g[dep].push_back(x);
    dfs(L[x],dep+1);
    dfs(R[x],dep+1);
}
int main(){
    cin>>n;cin>>x;
    L[tot]=R[tot]=-1;
    dat[tot++]=x;
    rep(i,1,n){
        cin>>x;
        Insert(0,x);
    }
    dfs(0,0);
    int l1=0,l2=0;
    if(g[dp].size()) l1=g[dp].size();
    if(dp) l2=g[dp-1].size();
    cout<<l1<<" + "<<l2<<" = "<<l1+l2<<endl;
    return 0;
}