uva10562 Undraw the Trees （不定叉树重建）_undraw the trees uva

本文介绍了一种根据给定的图树构建括号树的算法。通过递归方式建立树结构，并使用二维char数组和vector来保存孩子节点，最终输出括号树。文章详细解析了算法实现过程及代码。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题意：根据给定的图树，输出对应的括号树。树的结点可以是任意可打印的字符，不包括'|' 、'-'、 '#' '、 ' '\0'、 '\n'。对于\n 其实我是看别人题解的。那句英语折磨死我了。

思路：关键在于递归建树。用二维char数组一次性读掉整张图。因为是不定叉，所以用vector保存孩子结点。具体如下：

特殊处理第一行（可能为空树），找到根结点的坐标开始递归。对于每一个有效的结点字符，都要构建一个结点，然后返回它自己。如果它的下一行是'|' 说明它有孩子，则在返回之前要往下递归建树。找到分割线'-----'的线头 for下去，如果分割线在下一行一有结点字符就再建树，每返回的一个结点指针 childs.push_back(孩子结点指针); (vector<Node *> childs;)

算法复杂度： o（N）， N是结点数。

代码：

#include <cstdio>
#include <vector>
#include <cstring>
#include <cctype>
using namespace std;

#define MAX_LINE 220
#define MAX_CHAR 220

struct Node
{
	char ch;
	vector<Node *> childs;
};

char pfTree[MAX_LINE][MAX_CHAR];
Node *root;

Node *mkTree(int, int);
void rmTree(Node *&);
void printOurTree(Node *);
bool isNode(char);

int main()
{
	int cases;
	scanf("%d%*c", &cases);
	while (cases--) {
		// init
		memset(pfTree, 0, sizeof(pfTree));
		root = NULL;

		// enter
		for (int i = 0; i < MAX_LINE; i++) {
			gets(pfTree[i]);
			if (pfTree[i][0] == '#') {
				break;
			}
		}

		// make tree
		if (pfTree[0][0] == '#') {
			printf("()\n");
		} else {
			int y;
			for (int i = 0; i < MAX_CHAR; i++) {
				if (isNode(pfTree[0][i])) {
					y = i;
					break;
				}
			}
			root = mkTree(0, y);

			// printf
			printf("(");
			printOurTree(root);
			printf(")\n");

			// remove tree
			rmTree(root);
		}
	}

	return 0;
}

Node *mkTree(int row, int col)
{
	Node *root = new Node;
	root->ch = pfTree[row][col];

	// child ?
	int verticalRow = row + 1;
	if (pfTree[verticalRow][col] == '|') {
		int head;
		int lineRow = row + 2;
		for (head = col; head >= 0; head--) {
			if (pfTree[lineRow][head] != '-') {
				break;
			}
		}
			head++;

		int chilRow = row + 3;
		for (int i = head; pfTree[lineRow][i] == '-'; i++) {
			if (isNode(pfTree[chilRow][i])) {
				root->childs.push_back(mkTree(chilRow, i));
			}
		}
	}

	return root;
}

void rmTree(Node *&root)
{
	int size = root->childs.size();
	for (int i = 0; i < size; i++) {
		rmTree(root->childs[i]);
	}

	delete root;
	root = NULL;
}

void printOurTree(Node *root)
{
	printf("%c", root->ch);

	int size = root->childs.size();
	if (size != 0) {
		printf("(");
		for (int i = 0; i < size; i++) {
			printOurTree(root->childs[i]);
		}
		printf(")");
	} else {
		printf("()");
	}
}

bool isNode(char ch)
{
	if (ch == '-') {
		return false;
	} else if (ch == '|') {
		return false;
	} else if (ch == ' ') {
		return false;
	} else if (ch == '\n') {
		return false;
	} else if (ch == '\0') {
		return false;
	} else if (ch == '#') {
		return false;
	} else {
		return true;
	}
}