【华为编码大赛】树遍历


思路:用结构体记录下每个节点的父节点。从叶子节点依次往上遍历,直到与后面的节点最近的父节点停止,再从该节点向下遍历至后面的叶子节点,将结果存进ans数组,并统计每个数出现的次数,大于2时结束操作,输出-1,否则输出ans数组。


#include <cstdio>
#include <cstring>
const int LEN=330;
struct Node
{
    int fa;
    int n;
    int ncount;
}node[LEN];
int k;
int last[LEN];  //依次保存叶子节点
int order[LEN][LEN]; //每个叶子节点向上遍历至根节点的路径,用哈希的方式储存
int cou[LEN]; //用于统计每个节点被遍历的次数
int ans[LEN*LEN];
int res;
int f;
void ini()  //初始化函数
{
    memset(node,0,sizeof(node));
    memset(order,0,sizeof(order));
    memset(cou,0,sizeof(cou));
    res=0;
    f=1;
    for(int i=1;i<=k;i++)
        node[i].n=i;
    order[0][1]=1;
    node[1].fa=0;
}
void traversal(int t)  //从下往上依次沿父节点遍历,并用哈希方式存入数组。
{
    int a=t;
    int root=0;
    while(t!=root)
    {
        order[a][t]=1;
        t=node[t].fa;
    }
}
int find(int a,int b)   //找出两个节点最近的父节点
{
    int m=a>b?a:b;
    for(int i=m;i>=1;i--)
    {
        if(order[a][i] && order[b][i])
        {
            return i;
        }
    }
}
void add(int a,int b)  //按遍历顺序读入ans数组
{
    int tmp=find(a,b);
    for(int i=a;i>=tmp;i--)
    {
        if(order[a][i])
        {
            ans[res++]=i;
            cou[i]++;
            if(cou[i]>2)
            {
                f=0;
                return;
            }
        }
    }
    for(int i=tmp;i<=b;i++)
    {
        if(order[b][i])
        {
            ans[res++]=i;
            cou[i]++;
            if(cou[i]>2)
            {
                f=0;
                return;
            }
        }
    }
}
void addlast(int t)  //最后返回根节点经过的路径
{
    int root=0;
    t=node[t].fa;
    while(t!=root)
    {
        ans[res++]=t;
        cou[t]++;
        t=node[t].fa;
        if(cou[t]>2)
        {
            f=0;
            return;
        }
    }
}
int main()
{
    while(scanf("%d",&k)!=EOF)
    {
        ini();
        for(int i=1;i<k;i++)
        {
            int a,b;
            scanf("%d %d",&a,&b);
            node[b].fa=a;
            node[a].ncount++;   //用于判断最后一行有几个叶子节点,(是0的时候说明是叶节点)
        }
        int m=0;
        for(int i=1;i<=k;i++)
            if(node[i].ncount==0)
                m++;
        for(int i=1;i<=m;i++)
        {
            scanf("%d",&last[i]);
            traversal((last[i]));
        }
        last[0]=1;
        for(int i=0;i<m;i++)
        {
            add(last[i],last[i+1]);
            if(f==0)
            {
                printf("-1\n");
                return 0;
            }
        }
        addlast(last[m]);
        if(f==0)
        {
            printf("-1\n");
            return 0;
        }
        for(int i=0;i<res;i++)
        {
            printf("%d",ans[i]);
            if(i!=res-1)
                printf(" ");
        }
        printf("\n");
    }
    return 0;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值