uva839 Not so Mobile (重建二叉树)

题意：天平的平衡条件 Wl * Dl == Wr * Dr。 题目给出的是 二叉树天平， 平衡条件是所有天平都要平衡。 大天平的W 是小天平两边的W总和（很明显嘛~ 不行你就受力分析下啊）。

思路：好了， 这题就无法逃脱建树的命运了。 递归建树， 然后递归判断是否平衡。每次读入一行， 如果Wl == 0， 则递归建左子树， 如果Wr == 0， 则递归建右子树。 判断的时候， 如果Wl == 0， 则递归判断左子树， 并返回左子树（左天平）的w。 右边也一样。 然后 判断 Wl * Dl == Wr * Dr。

算法复杂度：建树o（N）+ 判断o（N）+delete结点o（N） == o（3N） ， 故算法复杂度为o（N）。 N为结点数。

代码：

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

#define LINE_NUM 4

struct Node
{
int Wl;
int Dl;
int Wr;
int Dr;
Node *left;
Node *right;
};

bool ok;

Node *mkTree();
void rmTree(Node *&);
int judge(Node *);		// judge and get weight

int main()
{
int cases;
scanf("%d", &cases);
for (int css = 1; css <= cases; css++){
// init
Node *root = NULL;
ok = true;

// make tree
root = mkTree();

// judge
if (css != 1) {
printf("\n");
}
judge(root);
if (ok) {
printf("YES\n");
} else {
printf("NO\n");
}

// remove node
rmTree(root);
}

return 0;
}

Node *mkTree()
{
Node *root = new Node;
scanf("%d%d", &root->Wl, &root->Dl);
scanf("%d%d", &root->Wr, &root->Dr);

if (root->Wl == 0) {
root->left = mkTree();
} else {
root->left = NULL;
}

if (root->Wr == 0) {
root->right = mkTree();
} else {
root->right = NULL;
}

return root;
}

void rmTree(Node *&root)
{
if (root == NULL) {
return;
} else {
rmTree(root->left);
rmTree(root->right);
delete root;
root = NULL;
}
}

int judge(Node *cur)
{
if (!ok) {
return 0;
}

if (cur->Wl == 0) {
cur->Wl = judge(cur->left);
}
if (cur->Wr == 0) {
cur->Wr = judge(cur->right);
}

if (cur->Wl * cur->Dl != cur->Wr * cur->Dr) {
ok = false;
}

return (cur->Wl + cur->Wr);
}