二叉树遍历模板 (前中求后、后中求前)

前中求后

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=5e4+7,INF=0x3f3f3f3f;
int in[N],pre[N];
int n,k;
void dfs(int l,int r,int rt)
{
    int i;
    for(i=l;in[i]!=pre[rt];i++);//in[i]==pre[rt]时结束循环
    if(l<=i-1) dfs(l,i-1,rt+1);//左子树
    if(i+1<=r) dfs(i+1,r,rt+i-l+1);//右子树
    k==0?printf("%d",in[i]):printf(" %d",in[i]),k++;//输出(lrv)
}
int main()
{

    while(~scanf("%d",&n))
    {
        k=0;
        for(int i=1;i<=n;i++) scanf("%d",&pre[i]);
        for(int i=1;i<=n;i++) scanf("%d",&in[i]);
        dfs(1,n,1);//直接传入地址便可
        puts("");
    }
}

中后求前

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=5e4+7,INF=0x3f3f3f3f;
int in[N],aft[N];
int n,k;
void dfs(int rt,int l,int r)
{
    int i;
    for(i=l; i<=r&&in[i]!=aft[rt]; i++);
    k==0?printf("%d",in[i]):printf(" %d",in[i]),k++;//输出(lrv)
    if(l<i) dfs(rt-r+i-1,l,i-1);//访左,rt-r+i-1,就是把下标一到左子树的位置。
    if(i<r) dfs(rt-1,i+1,r);//访右
}
int main()
{
    while(~scanf("%d",&n))
    {
        k=0;
        for(int i=1; i<=n; i++) scanf("%d",&in[i]);
        for(int i=1; i<=n; i++) scanf("%d",&aft[i]);
        dfs(n,1,n);
        puts("");
    }
}

猜你喜欢

转载自blog.csdn.net/oinei/article/details/80959064