树的练习

#include <iostream>
#include <cstring>
#include <vector>
using namespace std;
bool isMirror;
vector<int> pre;
vector<int> post;
void dfs(int root, int tail)
{
    
    
    if (root > tail)
        return;
    int i = root + 1;
    int j = tail;
    if (!isMirror)
    {
    
    
        while (i <= tail && pre[root] > pre[i])
            i++;
        while (j > root && pre[root] <= pre[j])
            j--;
    }
    else
    {
    
    
        while (i <= tail && pre[root] <= pre[i])
            i++;
        while (j > root && pre[root] <= pre[j])
            j--;
    }
    if (i - j != 1)
    // 这个会与后面的内容结合使用 判断长度是不是合格的
        return;
    dfs(root + 1, j); // 左
    dfs(i, tail);
    //右
    post.push_back(pre[root]);
    //使用了vector
}
int main()
{
    
    
    int n, node;
    int i = 0;
    cin >> n;
    for (int i = 0; i < n; i++)
    {
    
    
        cin >> node;
        pre.push_back(node);
    }
    dfs(0, n - 1);
    if(post.size()!=n){
    
    
isMirror=true;
 post.clear();//清空
dfs(0,n-1);
    }
    if (post.size() == n)
    {
    
    
        cout << "YES" << post[i];
        for (i = 1; i < post.size(); i++)
        {
    
    
            cout << " " << post[i];
        }
    }
    else 
    cout<<"NO";
}

在dfs 函数中应该注意 递归结束的条件。当 root==tail 就是只有一个节点的时候。注意的地方就是边界条件,i<=tail && j>root 。数组也可以当作树的结构，进行遍历的时候最好还是使用数组或者是使用vector。