洛谷 P5018 对称二叉树

题目传送门

解题思路：

先计算每个点的子树有多少节点，然后判断每个子树是不是对称的，更新答案。

AC代码：

#include<iostream>
#include<cstdio>

using namespace std;

struct kkk{
    int l,r,v,len;
    kkk() {l = -1;r = -1;}
}e[1000001];
int n,ans;

bool solve(int ls,int rs) {
    bool p = 1;
    if(e[ls].v != e[rs].v) return false;
    if((e[ls].l != -1 && e[rs].r == -1) || (e[ls].l == -1 && e[rs].r != -1) || (e[ls].r == -1 && e[rs].l != -1) || (e[ls].r != -1 && e[rs].l == -1)) return false;
    if(e[ls].l != -1 && e[rs].r != -1)
        p = p & solve(e[ls].l,e[rs].r);
    if(e[ls].r != -1 && e[rs].l != -1)
        p = p & solve(e[ls].r,e[rs].l);
    return p;
}

int dfs(int k) {
    e[k].len = 1;
    if(e[k].l != -1)
        e[k].len += dfs(e[k].l);
    if(e[k].r != -1)
        e[k].len += dfs(e[k].r);
    return e[k].len;
}

int main(){
    scanf("%d",&n);
    for(int i = 1;i <= n; i++)
        scanf("%d",&e[i].v);
    for(int i = 1;i <= n; i++)    
        scanf("%d%d",&e[i].l,&e[i].r);
    dfs(1);
    for(int i = 1;i <= n; i++)
        if(solve(e[i].l,e[i].r))
            ans = max(ans,e[i].len);
    printf("%d",ans);
    return 0;
}

//NOIP普及2018 T4