二叉树的递归遍历——天平（递归输入）

一、题目_,

输入一个树状数组，根据力矩相等原则判断是否平衡。采用递归方式输入（先序）：每个天平的格式为W_l, D_l ,W_r ,Dr,当W_l或W_r为0时，

表示该“砝码”实际是一个子天平，接下来会描述这个子天平。 当W_l = W_r  = 0时，会先描述左子天平，再描述右子天平。

二、解题思路

题目的输入就是采用递归的方式给的，自然要编个递归的方式输入。可以递归处理，到达根节点时，判断是否平衡，并将总重量传递给父天平。

由于引用传值，代码非常精简。紫书说这题极为重要，要确保完全理解。

三、代码实现

#include<stdio.h>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cstdbool>
using namespace std;

bool slove(int& W)
{
    int W1, D1, W2, D2;
    bool b1 = true,b2 = true;
    scanf("%d%d%d%d", &W1, &D1, &W2, &D2);
    if (!W1)    b1 = slove(W1);
    if (!W2)    b2 = slove(W2);

    //到达叶子节点，进行判断，并更新重量
    W = W1 + W2;
    return b1 && b2 && W1 * D1 == W2 * D2;
}

int main()
{
    int T, W;
    scanf("%d", &T);
    while (T--)
    {
        if (slove(W)) printf("YES
");
        else    printf("NO
");

        if (T)    printf("
");
    }
    return 0;
}