二叉搜尋樹的2層結點統計

題目大意

主要思路

由於二叉搜尋樹是左子樹上所有結點的值均小於或等於它的根結點的值，右子樹上所有結點的值均大於它的根結點的值，所以我們每插入一個點就要從根節點dfs找到能存放該點的地方，也就是說如果這個值大於根節點就要往右找，小於等於根節點就要往左找，遞迴此過程，這樣建樹並且更新最大深度

AC程式碼

程式碼通俗易懂

#include <iostream>
#include <algorithm>
#include <cstring>
#include <cstdio>
#include <string>
#include <cmath>
#include <queue>
#include <vector>
#include <map>
#include <sstream>
using namespace std;

const int N = 1010, INF = 0x3f3f3f3f;
int n, l[N], r[N], pos[N];
int cnt[N], max_depth;

void dfs(int root, int u, int depth)
{
    if(pos[u] <= pos[root])
    {
        if(l[root] == INF)
        {
            l[root] = u;
            cnt[depth+1]++;
            max_depth = max(max_depth, depth+1);
            return ;
        }
        else
        {
            dfs(l[root], u, depth + 1);
            return ;
        }
    }
    else
    {
        if(r[root] == INF)
        {
            r[root] = u;
            cnt[depth + 1]++;
            max_depth = max(max_depth, depth + 1);
            return ;
        }
        else
        {
            dfs(r[root], u, depth + 1);
            return ;
        }
    }
    
}

int main(void)
{
    memset(l, 0x3f, sizeof(l));
    memset(r, 0x3f, sizeof(r));
    cin >> n;
    if(n == 1)
    {
        cout << 1 << endl;
        return 0;
    }
    for(int i = 0; i < n; i++)
    {
        int x;
        cin >> x;
        if(!i)
        {
            pos[i] = x;
        }
        else
        {
            pos[i] = x;
            dfs(0, i, 0);
        }
    }

    cout << cnt[max_depth] + cnt[max_depth - 1] << endl;
    return 0;
}