A1102 | 反转二叉树

#include <stdio.h>
#include <memory.h>
#include <math.h>
#include <string>
#include <vector>
#include <set>
#include <stack>
#include <queue>
#include <algorithm>
#include <map>

#define I scanf
#define OL puts
#define O printf
#define F(a,b,c) for(a=b;a<c;a++)
#define FF(a,b) for(a=0;a<b;a++)
#define FG(a,b) for(a=b-1;a>=0;a--)
#define LEN 100
#define MAX 0x06FFFFFF
#define V vector<int>

using namespace std;

int used[LEN];
int cnt=;

typedef struct Node{
    struct Node *l=NULL;struct Node *r=NULL;
    int d;
    Node(){}
    Node(int D){d=D;}
    Node(Node *obj){
        if(obj){
            d=obj->d;
            l=obj->l;
            r=obj->r;
        }
    }
}Node;

typedef struct inputInfo{
    int index;string l;string r;
}inputInfo;

inputInfo infos[LEN];
Node * nodes[LEN];

Node * createNodes(int index);
void inOrder(Node * node);
Node * root;
void invert(Node* node);
void levelOrder(Node*node);

string inStr="";
string levelStr="";

int main(){
//    freopen("d:/input/A1102.txt","r",stdin);
    int n;
    scanf("%d",&n);
    int i;
    FF(i,n){
        char ch1[LEN];char ch2[LEN];
        I("%s%s",ch1,ch2);
        infos[i].index=i;infos[i].l=ch1;infos[i].r=ch2;
    }
    FF(i,n){
        root=createNodes(i);
        if(cnt>=n) break;
    }
    invert(root);
    inOrder(root);
    levelOrder(root);
    puts(levelStr.substr(,levelStr.size()-).c_str());
    puts(inStr.substr(,inStr.size()-).c_str());

    return ;
}

void levelOrder(Node*node){
    queue<Node*> q;
    q.push(node);
    while(!q.empty()){
        Node* t=q.front();
        q.pop();
        char buffer[LEN];
        sprintf(buffer,"%d ",t->d);
        levelStr+=buffer;
        if(t->l) q.push(t->l);
        if(t->r) q.push(t->r);
    }
}

Node * createNodes(int index){
    if(used[index]) return NULL;
    cnt++;
    used[index]=;
    int d=infos[index].index;
    inputInfo f=infos[index];
    Node *node =new Node(d);
    if(f.l!="-"){
        int i=;
        sscanf(f.l.c_str(),"%d",&i);

        if(used[i]){
            node->l=nodes[i];
        }else{
            node->l=createNodes(i);
        }
    }
    if(f.r!="-"){
        int i=;
        sscanf(f.r.c_str(),"%d",&i);

        if(used[i]){
            node->r=nodes[i];
        }else{
            node->r=createNodes(i);
        }
    }
    nodes[index]=node;
    return node;
}

void inOrder(Node * node){
    if(node){
        inOrder(node->l);
        char buffer[LEN];
        sprintf(buffer,"%d ",node->d);
        inStr+=buffer;
        inOrder(node->r);
    }
}

void invert(Node* node){
    if(!node) return;
    if(node->l) invert(node->l);
    if(node->r) invert(node->r);
    Node * t=NULL;
    if(node->l)t=new Node(node->l);
    node->l=NULL;
    if(node->r)node->l=new Node(node->r);
    node->r=t;
}

超简单的一道题，居然写了一个小时，还写了一百多行……明天一定要研究一下大佬们是怎么写的。

---

看了蓝书之后才知道原来可以这么写。

1.静态二叉树数据结构，左右叶子结点直接用Node数组的下标记录，空结点就记录1（在结构体中直接初始化为1）

2.反转二叉树的操作实际上是后序遍历

3.查找根节点的方法：先把数据都记录到Node数组中，在录数据的时候记录出现的叶子节点的ID，让isLeaf[ID]=0，这样直接找到整棵树都不是叶子的节点，就是根节点。

#include <stdio.h>
#include <memory.h>
#include <math.h>
#include <string>
#include <vector>
#include <set>
#include <stack>
#include <queue>
#include <algorithm>
#include <map>

#define I scanf
#define OL puts
#define O printf
#define F(a,b,c) for(a=b;a<c;a++)
#define FF(a,b) for(a=0;a<b;a++)
#define FG(a,b) for(a=b-1;a>=0;a--)
#define LEN 200
#define MAX 0x06FFFFFF
#define V vector<int>

using namespace std;

typedef struct Node{
    int l=-,r=-;
}Node;
Node nodes[LEN];
int isLeaf[LEN];
int n;
int cnt;

int getRoot(){
    int i;
    FF(i,n){
        if(isLeaf[i]==)
            return i;
    }
}

void buildTree(int i,char l,char r){
    if(l!='-'){
        int i_l=l-;
        nodes[i].l=i_l;
        isLeaf[i_l]=;
    }
    if(r!='-'){
        int i_r=r-;
        nodes[i].r=i_r;
        isLeaf[i_r]=;
    }
}

void print(int id){
    O("%d",id);
    if(++cnt<n) O(" ");
}

void invert(int i){
    if(i>=){
        invert(nodes[i].l);
        invert(nodes[i].r);
        swap(nodes[i].l,nodes[i].r);
    }
}

void bfs(int root){
    cnt=;
    queue<int> q;
    q.push(root);
    while(!q.empty()){
        int t=q.front();
        q.pop();
        print(t);
        if(nodes[t].l>=) q.push(nodes[t].l);
        if(nodes[t].r>=) q.push(nodes[t].r);
    }
}

void inOrder(int i){
    if(i>=){
        inOrder(nodes[i].l);
        print(i);
        inOrder(nodes[i].r);
    }
}

int main(){
    freopen("d:/input/A1102.txt","r",stdin);
    scanf("%d",&n);
    int i=n;
    FF(i,n){
        char l,r;
        scanf("%*c%c %c",&l,&r);
        buildTree(i,l,r);
    }
    int root=getRoot();
    invert(root);
    bfs(root);OL("");
    cnt=;inOrder(root);OL("");
    return ;
}