记录编号 555079 评测结果 AAAAAAAAAAAAAAAAAAAAAAAAA
题目名称 [NOIP 2018PJ]对称二叉树 最终得分 100
用户昵称 Gravatar夜莺 是否通过 通过
代码语言 C++ 运行时间 1.680 s
提交时间 2020-09-26 11:07:26 内存使用 17.22 MiB
显示代码纯文本
#include<cstdio>
using namespace std;
const int MAXN=1e6+5;
int l[MAXN],r[MAXN],v[MAXN],son[MAXN];
int n,ans;
bool can(int ll,int rr){
	if(ll==-1&&rr==-1)return 1;
	if(ll==-1||rr==-1)return 0;
	if(v[ll]!=v[rr])return 0;
	return can(l[ll],r[rr])&can(r[ll],l[rr]);
}
void dfs(int k){
	son[k]=1;
	if(l[k]!=-1){dfs(l[k]);son[k]+=son[l[k]];}
	if(r[k]!=-1){dfs(r[k]);son[k]+=son[r[k]];}
	if(son[k]>ans&&can(l[k],r[k]))ans=son[k];
}
int main(){
	freopen("noip2018pj_tree.in","r",stdin);
	freopen("noip2018pj_tree.out","w",stdout);
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
		scanf("%d",&v[i]);
	for(int i=1;i<=n;i++)
		scanf("%d%d",&l[i],&r[i]);
	dfs(1);
	printf("%d",ans);
	return 0;
}