1115 Counting Nodes in a BST (30 分)

PAT目录

#include <iostream>
using namespace std;
const int N=1010;
int n;
int l[N],r[N],w[N],idx=0;
int cnt[N],max_depth=0;
void insert(int &u,int ww){
    //u前面要加&,例如开始root=0,加入25之后root=1(记录编号)，主函数当中的root应该也随着变化
    if(!u) u=++idx,w[u]=ww;
    else if(ww<=w[u]) insert(l[u],ww);//l[u]存放的是左子树的序号，若为0，l[u]也随着改变值
    else insert(r[u],ww);
}
void dfs(int root,int depth){
    if(!root) return;
    max_depth=max(max_depth,depth);
    cnt[depth]++;
    dfs(l[root],depth+1);
    dfs(r[root],depth+1);
}
int main(){
    cin>>n;
    int root=0;
    for(int i=0;i<n;i++){
        int ww;cin>>ww;
        insert(root,ww);
    }
    int depth=0;
    dfs(root,depth);
    printf("%d + %d = %d",cnt[max_depth],cnt[max_depth-1],cnt[max_depth]+cnt[max_depth-1]);
    return 0;
}