【NOIP2018PJ正式赛】对称二叉树

最新推荐文章于 2023-08-01 19:08:16 发布

Larry1118

最新推荐文章于 2023-08-01 19:08:16 发布

阅读量285

点赞数

分类专栏： NOIP 暴力 dfs

本文链接：https://blog.csdn.net/Larry1118/article/details/85869440

版权

NOIP 同时被 3 个专栏收录

39 篇文章 0 订阅

订阅专栏

暴力

35 篇文章 0 订阅

订阅专栏

dfs

16 篇文章 0 订阅

订阅专栏

暴力dfs即可。

#include<cstdio>
#define N 1000010
using namespace std;
int n,v[N],l[N],r[N],son[N],ans=0;

inline int read()
{
	int x=0,f=0; char c=getchar();
	while (c<'0' || c>'9') f=(c=='-') ? 1:f,c=getchar();
	while (c>='0' && c<='9') x=x*10+c-48,c=getchar();
	return f ? -x:x;
}

bool pd(int x,int y)
{
	if (x==-1 && y==-1) return 1;
	if (x==-1 || y==-1) return 0;
	if (v[x]!=v[y]) return 0;
	return pd(l[x],r[y])&pd(r[x],l[y]);
}

void dfs(int x)
{
	son[x]=1;
	if (l[x]!=-1) dfs(l[x]),son[x]+=son[l[x]];
	if (r[x]!=-1) dfs(r[x]),son[x]+=son[r[x]];
	if (son[x]>ans && pd(l[x],r[x])) ans=son[x];
}

int main()
{
	freopen("tree.in","r",stdin);
	freopen("tree.out","w",stdout);
	n=read();
	for (int i=1;i<=n;i++) v[i]=read();
	for (int i=1;i<=n;i++) l[i]=read(),r[i]=read();
	dfs(1);
	printf("%d\n",ans);
	return 0;
}