【PAT】1102 Invert a Binary Tree (25 分)

最新推荐文章于 2022-10-13 21:17:47 发布

fftx_00

最新推荐文章于 2022-10-13 21:17:47 发布

阅读量130

点赞数

分类专栏： PAT_advanced2 文章标签： PAT

本文链接：https://blog.csdn.net/fftx_00/article/details/122689495

版权

PAT_advanced2 专栏收录该内容

35 篇文章 0 订阅

订阅专栏

①输入的时候就左右子树互换，不用之后再递归互换了

②递归左右互换

#include <iostream>
#include <vector>
#include <queue>
#include <string.h>
using namespace std;
struct node
{
    int lchild,rchild;
};

vector<node> T;
int times[15];

int flag_levelOrder=0;
void levelOrderTraversal(int root){
    if(root==-1) return;
    queue<int> q;
    q.push(root);
    while(!q.empty()){
        int tmp=q.front();
        q.pop();
        printf("%s%d",flag_levelOrder==1?" ":"",tmp);
        flag_levelOrder=1;
        if(T[tmp].lchild!=-1) q.push(T[tmp].lchild);
        if(T[tmp].rchild!=-1) q.push(T[tmp].rchild);
    }
    printf("\n");
}

int flag_inOrder=0;
void inOrderTraversal(int root){
    if(root==-1) return;
    inOrderTraversal(T[root].lchild);
    printf("%s%d",flag_inOrder==1?" ":"",root);
    flag_inOrder=1;
    inOrderTraversal(T[root].rchild);
}

int main(){
    int n;
    scanf("%d",&n);
    T.resize(n);

    string t1,t2;
    for(int i=0;i<n;i++){
        cin>>t2>>t1;
        if(t1=="-"){
            T[i].lchild=-1;
        }else{
            T[i].lchild=stoi(t1);
            times[stoi(t1)]=1;
        }
        if(t2=="-"){
            T[i].rchild=-1;
        }else{
            T[i].rchild=stoi(t2);
            times[stoi(t2)]=1;
        }
    }

    int root=0;
    while(times[root]==1) root++;

    levelOrderTraversal(root);
    inOrderTraversal(root);

    return 0;
}

#include <iostream>
#include <vector>
#include <queue>
#include <string.h>
using namespace std;
struct node
{
    int lchild,rchild;
};

vector<node> T;
int times[15];

void invertRoot(int root){
    if(root==-1) return;
    //不管有没有子树都可以直接交换
    int t=T[root].lchild;
    T[root].lchild=T[root].rchild;
    T[root].rchild=t;
    
    invertRoot(T[root].lchild);
    invertRoot(T[root].rchild);
}

int flag_levelOrder=0;
void levelOrderTraversal(int root){
    if(root==-1) return;
    queue<int> q;
    q.push(root);
    while(!q.empty()){
        int tmp=q.front();
        q.pop();
        printf("%s%d",flag_levelOrder==1?" ":"",tmp);
        flag_levelOrder=1;
        if(T[tmp].lchild!=-1) q.push(T[tmp].lchild);
        if(T[tmp].rchild!=-1) q.push(T[tmp].rchild);
    }
    printf("\n");
}

int flag_inOrder=0;
void inOrderTraversal(int root){
    if(root==-1) return;
    inOrderTraversal(T[root].lchild);
    printf("%s%d",flag_inOrder==1?" ":"",root);
    flag_inOrder=1;
    inOrderTraversal(T[root].rchild);
}

int main(){
    int n;
    scanf("%d",&n);
    T.resize(n);

    string t1,t2;
    for(int i=0;i<n;i++){
        cin>>t1>>t2;
        if(t1=="-"){
            T[i].lchild=-1;
        }else{
            T[i].lchild=stoi(t1);
            times[stoi(t1)]=1;
        }
        if(t2=="-"){
            T[i].rchild=-1;
        }else{
            T[i].rchild=stoi(t2);
            times[stoi(t2)]=1;
        }
    }

    int root=0;
    while(times[root]==1) root++;

    invertRoot(root);

    levelOrderTraversal(root);
    inOrderTraversal(root);

    return 0;
}