【洛谷P5018 对称二叉树】

话说这图也太大了吧

这题十分的简单，我们可以用两个指针指向左右两个对称的东西，然后比较就行了

复杂度O（n*logn）

#include<bits/stdc++.h>
using namespace std;
inline int read()
{
    int X=,w=;
    char c=getchar();
    while(c<''||c>'')
    {
        if (c=='-')
        {
            w=-;
        }
        c=getchar();
    }
    while(c>=''&&c<='')
    {
        X=(X<<)+(X<<)+c-'';
        c=getchar();
    }
    return X*w;
}
inline void out(int n)
{
    if(n<)
    {
        putchar('-');
        n=-n;
    }
    if(n>=)
    {
        out(n/);
    }
    putchar(n%+'');
}
int n,son[][],val[],size[];
//son[i][0]为i的左儿子
//son[i][1]为i的右儿子
inline void dfs(int u)
{
    size[u]=;
    if(son[u][]!=-
    )
    {
        dfs(son[u][]);
        size[u]+=size[son[u][]];
    }
    if(son[u][]!=-)
    {
        dfs(son[u][]);
        size[u]+=size[son[u][]];
    }
}
inline bool check(int u,int v)
{
    if(u==-&&v==-)
    {
        return true;
    }
    if(u!=-&&v!=-&&val[u]==val[v]&&check(son[u][],son[v][])&&check(son[u][],son[v][]))
    {
        return true;
    }
    return false;
}
int main()
{
    n=read();
    for(int i=;i<=n;i++)
    {
        val[i]=read();
    }
    for(int i=;i<=n;i++)
    {
        son[i][]=read();
        son[i][]=read();
    }
    dfs();
    int ans=;
    for(int i=;i<=n;i++)
    {
        if(check(son[i][],son[i][]))
        {
            ans=max(ans,size[i]);
        }
    }
    out(ans);
    puts("");
    return ;
}