【NOIP2018普及组】对称二叉树

版权声明:本文为博主原创文章……懂吗?要尊重别人的劳动成果呐 https://blog.csdn.net/Tiw_Air_Op1721/article/details/84190921


@题目描述@

一棵有点权的有根树如果满足以下条件,则被轩轩称为对称二叉树:

  1. 二叉树;
  2. 将这棵树所有节点的左右子树交换,新树和原树对应位置的结构相同且点权相等。

下图中节点内的数字为权值,节点外的 id 表示节点编号。
题目描述图

现在给出一棵二叉树,希望你找出它的一棵子树,该子树为对称二叉树,且节点数最多。请输出这棵子树的节点数。
注意:只有树根的树也是对称二叉树。本题中约定,以节点 T 为子树根的一棵“子树”指的是:节点 T 和它的全部后代节点构成的二叉树。

输入
第一行一个正整数 n,表示给定的树的节点的数目,规定节点编号 1~n,其中节点1 是树根。
第二行 n 个正整数,用一个空格分隔,第 i 个正整数 vi 代表节点 i 的权值。
接下来 n 行,每行两个正整数 li , ri ,分别表示节点 i 的左右孩子的编号。如果不存在左 / 右孩子,则以 −1 表示。两个数之间用一个空格隔开。
输出
输出共一行,包含一个整数,表示给定的树的最大对称二叉子树的节点数。

输入样例 1
2
1 3
2 -1
-1 -1
输入样例 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

输出样例 1
1
输出样例 2
3

输入输出样例 1 说明
样例解释1
最大的对称二叉子树为以节点 2 为树根的子树,节点数为 1。

输入输出样例 2 说明
样例解释2
最大的对称二叉子树为以节点 7 为树根的子树,节点数为 3。

数据规模与约定
共25个测试点。vi≤1000
测试点1~3,n≤10,保证根结点的左子树的所有节点都没有右孩子,根结点的右子树的所有节点都没有左孩子。
测试点4~8,n≤10。
测试点9~12,n≤10^5,保证输入是一棵“满二叉树”。
测试点13~16,n≤10^5,保证输入是一棵“完全二叉树”。
测试点17~20,n≤10^5,保证输入的树的点权均为 1。
测试点21~25,n≤10^6。

@题解@

一开始看他们传什么马拉车算法心中一惊,以为今年普及组要上天。后来说是树哈希,要靠谱点,也差不多是普及组 T4 的水平。
结果……今年普及组真的是要上天……只不过是反向上天……

这道题做法很暴力:枚举每个结点,如果它左右子树大小相同,则暴力 Check 一下以这个结点为根的子树是否合法。

虽然看上去很暴力……但是这样搞的时间复杂度的确是O(nlog n)。
证明思路可以采用启发式合并的时间复杂度证明思路。
即:因为左右子树相同时才 check,因此每一次 check 树的大小至少增大一倍。最多 log 次树的大小就会到达 n,所以每个结点只会被 check log 次。

哎……心疼那些做 T3 结果没做出来反而 T4 这么简单却没有分的同学们……

@代码@

#include<cstdio>
inline int max(int x, int y) {
	return x > y ? x : y;
}
const int MAXN = 1000000;
int siz[MAXN + 5], le[MAXN + 5], ri[MAXN + 5], v[MAXN + 5], ans;
bool check(int r1, int r2) {
	if( v[r1] != v[r2] ) return false;
	else if( r1 == 0 && r2 == 0 ) return true;
	else return check(le[r1], ri[r2]) && check(ri[r1], le[r2]);
}
int dfs1(int rt) {
	if( !rt ) return 0;
	else return siz[rt] = dfs1(le[rt]) + dfs1(ri[rt]) + 1;
}
void dfs2(int rt) {
	if( !rt ) return ;
	if( siz[le[rt]] == siz[ri[rt]] )
		if( check(le[rt], ri[rt]) ) ans = max(ans, siz[rt]);
	dfs2(le[rt]), dfs2(ri[rt]);
}
int main() {
	int n; v[0] = -1;
	scanf("%d", &n);
	for(int i=1;i<=n;i++)
		scanf("%d", &v[i]);
	for(int i=1;i<=n;i++) {
		scanf("%d%d", &le[i], &ri[i]);
		if( le[i] == -1 ) le[i] = 0;
		if( ri[i] == -1 ) ri[i] = 0;
	}
	dfs1(1), dfs2(1);
	printf("%d\n", ans);
}

@end@

就是这样,新的一天里,也请多多关照哦(ノω<。)ノ))☆.。

猜你喜欢

转载自blog.csdn.net/Tiw_Air_Op1721/article/details/84190921
今日推荐