XJOI - 题目ID：8196
题目描述
时间：1s 空间：256M
题目描述：
一棵有点权的有根树如果满足以下条件，则被轩轩称为对称二叉树：
-
二叉树；
-
将这棵树所有节点的左右子树交换，新树和原树对应位置的结构相同且点权相等。
下图中节点内的数字为权值，节点外的 idid 表示节点编号。
现在给出一棵二叉树，希望你找出它的一棵子树，该子树为对称二叉树，且节点数最多。请输出这棵子树的节点数。
注意：只有树根的树也是对称二叉树。本题中约定，以节点 TT 为子树根的一棵「子树」指的是：节点 TT 和它的全部后代节点构成的二叉树。
输入格式：
第一行一个正整数 nn，表示给定的树的节点的数目，规定节点编号 1∼n1∼n，其中节点 11 是树根。
第二行 nn 个正整数，用一个空格分隔，第 ii 个正整数 vivi 代表节点 ii 的权值。
接下来 nn 行，每行两个正整数 li,rili,ri，分别表示节点 ii 的左右孩子的编号。如果不存在左 / 右孩子，则以 −1−1 表示。两个数之间用一个空格隔开。
输出格式：
输出文件共一行，包含一个整数，表示给定的树的最大对称二叉子树的节点数。
样例1
样例输入1：
2 1 3 2 -1 -1 -1
样例输出1：
1
样例解释 1
最大的对称二叉子树为以节点 2 为树根的子树，节点数为 1。
样例2
样例输入2
10 2 2 5 5 5 5 4 4 2 3 9 10 -1 -1 -1 -1 -1 -1 -1 -1 -1 2 3 4 5 6 -1 -1 7 8
样例输出 2
3
样例解释 2
最大的对称二叉子树为以节点 7 为树根的子树，节点数为 3。
数据范围与提示
共 25 个测试点。
vi≤1000vi≤1000。
测试点 1∼3,N≤101∼3,N≤10，保证根结点的左子树的所有节点都没有右孩子，根结点的右子树的所有节点都没有左孩子。
测试点 4∼8,N≤104∼8,N≤10。
测试点 9∼12,N≤1059∼12,N≤105，保证输入是一棵「满二叉树」。
测试点 13∼16,N≤10513∼16,N≤105，保证输入是一棵「完全二叉树」。
测试点 17∼20,N≤10517∼20,N≤105，保证输入的树的点权均为 1。
测试点 21∼25,N≤10621∼25,N≤106。
#include<bits/stdc++.h>
using namespace std;
long long n,a[1000005],l[1000005],r[1000005],f[1000005];
void find(long long x){
long long cnt=1;
if(l[x]!=-1){
find(l[x]);
cnt+=f[l[x]];
}
if(r[x]!=-1){
find(r[x]);
cnt+=f[r[x]];
}
f[x]=cnt;
//return cnt;
}
long long dfs(long long la,long long ra){
if(la==-1&&ra==-1) return 1;
if(la==-1||ra==-1) return 0;
if(f[la]!=f[ra]) return 0;
return min(dfs(l[la],r[ra]),dfs(r[la],l[ra]));
}
int main(){
cin>>n;
for(long long i=1;i<=n;i++){
cin>>a[i];
}
for(long long i=1;i<=n;i++){
cin>>l[i]>>r[i];
}
find(1);
long long ans=1;
for(long long i=1;i<=n;i++){
if(dfs(l[i],r[i])==1){
ans=max(ans,f[i]);
}
}
cout<<ans;
}