【基础算法】下一棵树问题（BZOJ1188）

最新推荐文章于 2020-03-20 21:09:54 发布

LSY_LELOUCH_YUKI

最新推荐文章于 2020-03-20 21:09:54 发布

阅读量651

点赞数 2

本文链接：https://blog.csdn.net/LELOUCH_YUKI/article/details/54630698

版权

问题 J(1188): 【基础算法】下一棵树问题

时间限制: 1 Sec 内存限制: 64 MB
提交: 18 解决: 5
[ 提交][ 状态][ 我的提交]

题目描述

农场上有N（1 < = N < =1000）棵树。在上过计算机课后，lin发现所有的树实际上都是严格的二叉树，二叉树的每个非叶子节点都恰好有两个子节点，lin给每个结点一个数来表示以这个结点为根的子树的叶结点数。然后，lin按照先序遍历的结果把和结点相关的数列出作为它的特征序列。但是，他只列出了与根结点和所有的左子结点相关的数。例如，下图这棵树：用*表示的是lin列出的节点，这棵树的特征序列为：（7 4 1 2 1 1 1 ）。在用这种方法表示完所有的树后，lin发现： 1.所有的树有同样多的叶结点 2.所有的树有不同的特征序列。 3.所有可能的严格的二叉树都在农场上。所以，lin决定吧这些特征序列排序，然后希望给出一棵树的特征序列，求出紧接着的一个序列。

输入

输入：共两行，第一行为N的值，表示特征序列的长度。第二行为N个空格隔开的数，表示一棵树的特征序列。

输出

输出：单独的一行，表示按字典序排列后所给序列的下一个序列。如果所给出的是最后一个序列，则输出0. 注意：输入序列和输出序列所代表的树要有相同的叶节点数。

样例输入

(如果复制到控制台无换行，可以先粘贴到文本编辑器，再复制)

7
7 4 1 2 1 1 1

样例输出

7 4 1 2 1 2 1

分析：首先根据特征序列构造二叉树，由特征序列是先序遍历的结果,可知: 以序列的第一个数为根, 紧随其后的数必然是根的左孩子, 两数之差则是根的右孩子;向下推进一层,以左孩子为根,重复此规律, 直到根为1;然后回退到上一层, 如果该层结点的右孩子不为1,以其右孩子为根,以特征序列中的下一个数为该根结点的左孩子, 计算出根的右孩子;再次向下推进一层,以左孩子为根,重复此规律, 直到根为1;重复上述过程, 直到特征序列的所有数被用完.二叉树就被构造出来了.（实现时不需要把整棵树存下来，只需要用一个tree数组存下先序遍历的结果，那么当当前节点比访问到的特征序列大时，把它们的差（右子树）呀入栈，等到不大于时访问下一个节点，并把栈中的右子树统一访问）。满足题设要求的当前序列的下一个序列怎样求?其实就是把这棵树的最右子树的叶结点适当往左边调整,形成新的一棵二叉树的特征序列.问题是调整到什么位置?应该调整到最右子树的左边某个位置(在先序遍历中要先被访问), 这个结点的特征数要比其父结点的特征数小2以上.这样就可以把该结点的特征数加1, 表示把原来右边位置上的某个叶结点向左移动了。（实现：在构造二叉树时，同时记录每一个特征数在先序遍历中的位置，然后倒着找到一个与根相差2以上的节点，把它的特征值+1，它前面的不变，后面的全为1（可通过旋转实现））。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<stack>
using namespace std;
stack <int> q;
int n,top=1,a[1010],tree[1010],pos[1010];
int main(){
    scanf("%d",&n);
	for(int i=1;i<=n;i++)
		scanf("%d",&a[i]);
	tree[1]=a[1],pos[1]=1;
	for(int i=2;i<=n;i++){
		if(tree[top]-a[i]>0)
			q.push(tree[top]-a[i]);
	    tree[++top]=a[i],pos[i]=top;
        while(tree[top]==1&&!q.empty())
        	tree[++top]=q.top(),q.pop();
	}
	int i;
	for(i=n;tree[pos[i]-1]-tree[pos[i]]<2&&i>1;i--);
	if(i<=1){
		putchar(48);
		return 0;
	}
	printf("%d",a[1]);
	for(int j=2;j<i;j++)
		printf(" %d",a[j]);
	printf(" %d",a[i]+1);
	for(int j=i+1;j<=n;j++)
		putchar(32),putchar(49);
	return 0;
}