题目意思:给定一序列的数字,每组四个,分别是W1 D1 W2 D2 表示左右节点的重量和长度,如果W1 或W2 为0说明有子节点存在,判断是否满足平衡
解题思路:1 建立二叉树,然后遍历查找 但是这个很麻烦 2 dfs 我们知道对于一颗树肯定是要先把左子树建完才能建右子树 我们可以用递归方式来代替建树直接求出左右两边的重量,判断是否相等 3 硬搞,这题数据很水,直接判断叶子节点是否全部满足平衡,如果是YES,其它都是NO(话说这样和标称ans不同,但是也可以AC哦)
代码1(dfs):
//直接dfs,我们知道对于有0它肯定一直往下建树
#include <cstdio>
#include <cstring>
#include <iostream>
using namespace std;
int t , w1 , w2 , d1 , d2 , flag;
//递归求解
void dfs(){
scanf("%d%d%d%d" , &w1 , &d1 , &w2 , &d2);
if(w1 == 0)
w1 = dfs();
if(w2 == 0)
w2 = dfs();
if(w1*d1 != w2*d2)
flag = 0;
return w1+w2;//返回重量
}
int main(){
cin>>t;
while(t--){
flag = 1;//初始化为1
dfs();
if(flag)
printf("YES\n");
if(flag == 0)
printf("NO\n");
if(t)
cout<<endl;
}
}
代码2
/*这一题数据很水,所以我用直接判断叶子节点(对应的重量值全部为正数)对应是否全部满足平衡,如果有意个不满足就输出NO,否则YES.*/
#include <cstdio>
#include <cctype>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <stack>
#include <list>
#include <algorithm>
using namespace std;
const int MAXN = 1000010;
int t , line , Index , node[MAXN][4] , flag;//node数组存储读入的数据
//处理问题的函数
void solve(){
flag = 1;
int i , j , lsum = 0, rsum = 0;
for(i = 0 ; i < Index ; i++){
for(j = 0 ; j < 4 ; j++){
if(node[i][0] == 0 || node[i][2] == 0)//这一句重点,如果是重量为0不用考虑
break;
}
if(j == 4){
lsum = node[i][0] * node[i][1];
rsum = node[i][2] * node[i][3];
if(lsum != rsum){//如果有意个不满足直接输出NO return回去
printf("NO\n");
return;
}
}
}
if(i == Index)//如果全部满足输出YES
printf("YES\n");
}
//主函数
int main(){
int i , j;
cin>>t;
for(j = 1 ; j <= t ; j++){
line = 0 ; Index = 1;//Index表示有几行,只要重量为0,Index加一
while(1){
for(i = 0 ; i < 4 ; i++)
scanf("%d" , &node[line][i]);
if(node[line][0] == 0)
Index++;
if(node[line][2] == 0)
Index++;
++line;
if(line == Index)
break;
}
solve();
if(j != t)
cout<<endl;
}
return 0;
}