1156. Binary tree 二叉树找根方法

#include "iostream"

#include "memory.h"

using namespace std;

/*

找根方法：

每个节点输入次数：根节点在输入自己时输入一次，

非根节点在输入父节点时输入一次，在输入自己时又输入

一次，所以输入了两次。所以可以通过计数器，计算每个节点

被输入的次数来找出根节点。

*/

const int MAX = 1002;

struct Node{

int id;

char val;

int left, right;

Node(char c, int i = 0){

id = i;

val = c;

left = 0;

right = 0;

}

Node(){

left = 0;

right = 0;

}

};

void preOrder(int root, Node *node){

if (root != 0){

cout << node[root].val;

preOrder(node[root].left, node);

preOrder(node[root].right, node);

}

}

int main(){

int n;

while (cin >> n){

int isroot[MAX];

Node node[MAX];

memset(isroot, 0, sizeof(isroot));

int id, left, right;

char value;

for (int i = 0; i < n; i++){

cin >> id >> value >> left >> right;

node[id].id = id;

node[id].val = value;

node[id].left = left;

node[id].right = right;

isroot[id]++;

isroot[left]++;

isroot[right]++;

}

int root;

for (int i = 0; i < MAX; i++){

if (isroot[i] % 2 == 1){ //输入次数为次数的节点为根

root = i;

}

}

preOrder(root, node);

cout << endl;

}

return 0;

}