洛谷 P1305

https://www.luogu.org/problemnew/show/P1305

题目描述

输入一串二叉树，用遍历前序打出。

输入输出格式

输入格式：

 

第一行为二叉树的节点数n。(n≤26)

后面n行，每一个字母为节点，后两个字母分别为其左右儿子。

空节点用*表示

 

输出格式：

 

前序排列的二叉树

 

输入输出样例

输入样例#1： 复制

6
abc
bdi
cj*
d**
i**
j**

输出样例#1： 复制

abdicj

思路：这道题数据比较弱，第二行的第一个必定是根结点，所以可以玩得很花，老老实实用结构体建树也行。(吐槽一下这题的数据格式完全没按说明来，用getchar()或scanf就疯狂RE 惊了)

#include<iostream>
using namespace std;

struct node
{
    char ch;
    node *left;
    node *right;
    node(char c=0)
    {
        ch=c;
        left=NULL;
        right=NULL;
    }
};

node* Find(node *head,char c);
void PreOrder(node *head);

int main()
{
    int n;
    cin>>n;
    node *head;
    node *temp;
    for(int i=0;i<n;i++)
    {
        char p,l,r;
        cin>>p>>l>>r;
        if(i==0)
            head=new node(p);
        temp=Find(head,p);
        if(l!='*')
            temp->left=new node(l);
        if(r!='*')
            temp->right=new node(r);
    }
    PreOrder(head);
    return 0;
}

node* Find(node *head,char c)
{
    if(head->ch==c)
        return head;
    node *temp=NULL;
    if(head->left!=NULL)
        temp=Find(head->left,c);
    if(temp!=NULL)
        return temp;
    if(head->right!=NULL)
        temp=Find(head->right,c);
    return temp;
}

void PreOrder(node *head)
{
    cout<<head->ch;
    if(head->left!=NULL)
        PreOrder(head->left);
    if(head->right!=NULL)
        PreOrder(head->right);
}

用数组做也行：

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;

char l[105];
char r[105];

void dfs(int i);

int main()
{
    int n;
    scanf("%d",&n);
    char t1,t2,t3;
    char head;
    for(int i=0;i<n;i++)
    {
        cin>>t1>>t2>>t3;
        if(i==0)
            head=t1;    //记录头结点
        if(t2!='*')
            l[t1-'a']=t2;
        if(t3!='*')
            r[t1-'a']=t3;
    }
    dfs(head-'a');
    return 0;
}

void dfs(int i)
{
    printf("%c",i+'a');
    if(l[i]!=0)
        dfs(l[i]-'a');
    if(r[i]!=0)
        dfs(r[i]-'a');
    return ;
}