题解 对称二叉树

题面

 

 

解析

这题看上去好像很复杂（实际上也很复杂）。。。

但仔细想想，

若以一个点v为根的子树为对称二叉树，

则它的左子树lson和右子树rson一定是对称的，即：

lson的左子树和rson的右子树是对称的且lson的右子树和rson的左子树是对称的。

看出规律了吧，

这样只要递归求解就行了！！

终止条件就应该是：lson=rson=-1;

然后，先遍历一遍，求出以每个点为根的节点数目，再枚举就能AC了！！

 

 

上AC代码：

#include <bits/stdc++.h>
using namespace std;

struct tree{
    int ls,rs;
    int w,sonsum;
    int id;
};
tree a[1000001];
int n;
int ls=0,rs=1,ans=0;

bool dfs(int x,int y){
    if(x==-1&&y==-1) return 1;
    else if(x==-1||y==-1) return 0;
    int xl=a[x].ls,xr=a[x].rs;
    int yl=a[y].ls,yr=a[y].rs;
    if(dfs(xl,yr)&&dfs(xr,yl)){
        if(a[x].w==a[y].w) return 1;
    }
    return 0;
}

int find(int x){
    if(x==-1) return 0;
    a[x].sonsum=find(a[x].ls)+find(a[x].rs);
    return ++a[x].sonsum;
}

int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i].w);
        a[i].id=i;
    }
    for(int i=1;i<=n;i++){
        scanf("%d%d",&a[i].ls,&a[i].rs);
    }
    find(1);
    for(int i=1;i<=n;i++){
        if(dfs(a[i].ls,a[i].rs)){
            ans=max(ans,a[i].sonsum);
        }
    }
    printf("%d\n",ans);
    return 0;
}