题目 对称二叉树
题目描述
思路
检查是否符合对称条件
条件很简单——结构对称&&点权对称
要做到点权对称其实也就顺便结构对称了
于是条件可以简化为点权对称
可以考虑并行搜索
bool con(int l,int r) {
if(l == -1&&r == -1)
return 1;
if(l == -1||r == -1)
return 0;
if(w[l] == w[r])
if(check(l,r))
return 1;
return 0;
}
bool check(int x,int y) {
if(x == -1&&y == -1)
return 1;
if(x == -1||y == -1)
return 0;
if(w[x] != w[y])
return 0;
int l = Root[x].l,l1 = Root[y].l;
int r = Root[y].r,r1 = Root[x].r;
if(con(l,r)&&con(l1,r1))
return 1;
return 0;
}
信仰深搜
就三个点
int dfs(int x) {
if(x == -1) return 0;
if(check(Root[x].l,Root[x].r)) {
int ans = Find(x) + 1;
return ans;
}
int ans = max(dfs(Root[x].l),dfs(Root[x].r));
return ans;
}
找答案
加一是加根节点
int Find(int x) {
int q = 0;
int l = Root[x].l;
int r = Root[x].r;
if(l != -1) q += Find(l) + 1;
if(r != -1) q += Find(r) + 1;
return q;
}
另外
读入时要记录这样几个玩意儿
for(i = 1;i <= n;i++)
scanf("%d",&w[i]);
for(i = 1;i <= n;i++)
scanf("%d%d",&Root[i].l,&Root[i].r);
code
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define M 1000001
using namespace std;
int w[M];
struct N {
int l,r;
}Root[M];
bool con(int,int);
bool check(int,int);
//两个函数相互递归调用,并行搜索检查是否符合要求
int dfs(int);
//核心
int Find(int);
//其实就是找有多少个点
int main() {
int i,n;
scanf("%d",&n);
for(i = 1;i <= n;i++)
scanf("%d",&w[i]);
for(i = 1;i <= n;i++)
scanf("%d%d",&Root[i].l,&Root[i].r);
int ans = dfs(1);
printf("%d",ans);
return 0;
}
bool con(int l,int r) {
if(l == -1&&r == -1)
return 1;
if(l == -1||r == -1)
return 0;
if(w[l] == w[r])
if(check(l,r))
return 1;
return 0;
}
bool check(int x,int y) {
if(x == -1&&y == -1)
return 1;
if(x == -1||y == -1)
return 0;
if(w[x] != w[y])
return 0;
int l = Root[x].l,l1 = Root[y].l;
int r = Root[y].r,r1 = Root[x].r;
if(con(l,r)&&con(l1,r1))
return 1;
return 0;
}
int Find(int x) {
int q = 0;
int l = Root[x].l;
int r = Root[x].r;
if(l != -1) q += Find(l) + 1;
if(r != -1) q += Find(r) + 1;
return q;
}
int dfs(int x) {
if(x == -1) return 0;
if(check(Root[x].l,Root[x].r)) {
int ans = Find(x) + 1;
return ans;
}
int ans = max(dfs(Root[x].l),dfs(Root[x].r));
return ans;
}
总结
信仰很重要
这代码很慢但不至于卡常,还有大量可优化地方,此处不再赘述
但它非常好理解