题目链接
自己写了一个DFS判断的代码,提交上去显示多种错误,我是真的菜。
看了别人的代码,发现自己写递归搜索的时候,总是出现各种问题,很难受。
这个版本是柳神的代码,略做修改,方便以后自己刷第二遍的时候查看。
#include<bits/stdc++.h>
using namespace std;
struct node{
int data;
node* left;
node* right;
};
void build(node* &root, int v){
if(root==NULL){
root= new node;
root->data = v;
root->left=root->right=NULL;
}else if(abs(v) < abs(root->data)){
build(root->left,v);
}else build(root->right,v);
}
bool judge1(node *root) {
if (root == NULL) return true;
if (root->data < 0) {
if (root->left != NULL && root->left->data < 0) return false;
if (root->right != NULL && root->right->data < 0) return false;
}
return judge1(root->left) && judge1(root->right);
}
int getNum(node *root) {
if (root == NULL) return 0;
int l = getNum(root->left);
int r = getNum(root->right);
return root->data > 0 ? max(l, r) + 1 : max(l, r);
}
bool judge2(node *root) {
if (root == NULL) return true;
int l = getNum(root->left);
int r = getNum(root->right);
if(l != r) return false;
return judge2(root->left) && judge2(root->right);
}
int main(){
int t,n,num;
cin>>t;
while(t--){
cin>>n;
node * root = NULL;
for(int i=0;i<n;i++){
cin>>num;
build(root, num);
}
if(root->data < 0 || judge1(root)==false ||judge2(root)==false){
cout<<"No"<<endl;
}else cout<<"Yes"<<endl;
}
return 0;
}