839 - Not so Mobile

/*
解题思路:
创建二叉树
遍历检查

这道题其实不需要建立树的结构,因为遍历检查可以在创建时就能完成
需要用到数组做存储结构
*/
#include <iostream>
#include <string>
#include <cstdio>
#include <cstdlib>
using namespace std;

struct Node
{
	int w1,d1,w2,d2;
	Node *lchild,*rchild;
};
bool isEqual;

Node *createTree()
{
	int w1,d1,w2,d2;
	cin>>w1>>d1>>w2>>d2;
	Node *root=(Node *)malloc(sizeof(Node));
	root->w1=w1;
	root->d1=d1;
	root->w2=w2;
	root->d2=d2;
	if(w1==0)
		root->lchild=createTree();
	else
		root->lchild=NULL;
	if(w2==0)
		root->rchild=createTree();
	else
		root->rchild=NULL;
	return root;
}

int check(Node *root)
{
	if(root->lchild!=NULL)
		root->w1=check(root->lchild);
	if(root->rchild!=NULL)
		root->w2=check(root->rchild);
	if(isEqual==true)																//这里有个技巧******
		isEqual=(root->w1*root->d1==root->w2*root->d2);
	/*		另一种表示形式
	if(root->w1*root->d1!=root->w2*root->d2)
		isEqual=false;
	*/
	return root->w1+root->w2;
}


int main()
{
	//freopen("data.in","r",stdin);
	int T;
	cin>>T;
	while(T--)
	{
		Node *root=createTree();
		isEqual=true;
		check(root);
		cout<<(isEqual?"YES":"NO")<<endl;
		//The outputs of two consecutive cases will be separated by a blank line.
		if(T!=0)//仔细看题,WA原因
			cout<<endl;
	}
	return 0;
}


你可能感兴趣的:(839 - Not so Mobile)