牛客网. 构造Maxtree

题目概述

 

解题思路

这道题的原始版本是没有告诉我们解法的（如图1所示）：

图1

这无疑给题目增加了难度。

这里先给出本题的证明：

然后就是实现的办法。这里我们用栈来实现：

• 这里有一个小技巧：因为题目里要求我们将树根节点的父节点设置为1，如果我们想要一次遍历就完成这个设置，可以选一个大于当前数组中A所有元素的数，记为INT_MAX，置于A的末端；然后原先数组中最大的元素就会指向数组的末端。也就是说，原始数组中所有元素的父节点变为：

我们现在只要将该式做一下调整，就可以得到预期的结果了：

• 另一个语法技巧是vector元素的初始化。如果想要批量化初始一个vector，可以这样去写：

• vector<xxx> A = vector<xxx>(number of A, target value of A);

   

方法性能

该方法的时空复杂度均为O(N)。

 

示例代码

class MaxTree {
public:
	vector<int> buildMaxTree(vector<int> A, int n) 
	{
		// write code here
		stack<int> max_num;
		vector<int> res_max(n, -1);
		A.push_back(INT_MAX);
		for (int i = 0, temp_max_idx = 0; i < n+1; ++i)
		{
			if (max_num.empty() || A[max_num.top()] > A[i])
			{
				temp_max_idx = i;
				max_num.push(i);
			}
			else
			{
				while (!max_num.empty())
				{
					if (A[max_num.top()] < A[i])
					{
						int temp_top = max_num.top();
						max_num.pop();
						if (max_num.empty() || (!max_num.empty() && A[max_num.top()] > A[i]))
							res_max[temp_top] = (i + 1) % (n + 1) - 1;
						else
							res_max[temp_top] = (max_num.top() + 1) % (n + 1) - 1;
					}
					else
						break;
				}
				
				if (!max_num.empty())
					temp_max_idx = max_num.top();
				else
					temp_max_idx = i;
				max_num.push(i);
			}
		}
		
		//for (int j = 0; j < n; ++j)
		//{
		//	if (res_max[j] == n)
		//		res_max[j] = -1;
		//}
		return res_max;
	}
};

---

拓展延伸

这道题的原始版本并没有给出一种确定的解法，那如果按照原版题设去做的话，可能生成多少种不同结构的二叉树呢？

假设对一个包含N个元素的数组来说，符合条件的二叉树个数为。我们写一下它的递推公式：

乍一看，这是个挺复杂的式子。。。跟我们熟悉的卡特兰数也不太一样。。。

不妨列一下这个数组的前几项：

N	0	1	2	3	4	5
	1	1	2	6	24	120

多么熟悉的一串数！！_(:з」∠)_

所以它的递推公式，其实就是阶乘的一种骚气写法^_^