题目意思:给定一序列的数字,每组四个,分别是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; }
时间: 2024-12-21 03:41:42