《程序员面试金典》(第六版)习题:仅为记录一下以加强印象,不为商业用途,如有侵权请联系删除。以下源码和解释参考了书中源码以及解释。
题意为给定一颗二叉搜索树,找到一个数组集合,集合中的每一个数组都是由二叉搜索树中的所有节点的值组成,通过不断的将数组中的元素按照元素在数组中的顺序插入(从当前二叉树的头结点开始,当当前值大于当前节点的值时向当前节点的右子树递归,当当前值小于当前节点的值时向当前节点的左子树递归,直到遇到空节点为止)可以形成和原二叉搜索树一样的二叉搜索树。数组集合中的每一个数组的元素均为原二叉搜索树中所有节点的值,只不过不同的数组元素值的排序不同。
算法的思想是如果将一个数组中的元素按顺序依次插入可以形成一颗二叉搜索树则数组首元素为形成的二叉搜索树的根节点。根据这一点算法利用递归思想首先将二叉搜索树的头节点拿出来,然后递归调用左子树和右子树找到可以插入形成左子树和右子树的数组集合
A
A
A和
B
B
B。这时对于数组集合
A
A
A中的每一个数组a和数组集合
B
B
B中的每一个数组b以及头节点可以形成一批结果数组(将头结点作为数组的首元素,数组的其它元素为数组a和数组b的“交织”,“交织”的含义是交织的结果中数组a和数组b中的所有元素的顺序都不变),遍历数组集合
A
A
A和
B
B
B中的每一个数组,最后得到的左右数组就是所求的结果。
“交织”操作也用到了递归,所以这里用到了两次递归。交织函数为
v
o
i
d
w
e
a
v
e
L
i
s
t
s
(
l
i
s
t
<
i
n
t
>
f
i
r
s
t
,
l
i
s
t
<
i
n
t
>
s
e
c
o
n
d
,
v
e
c
t
o
r
<
l
i
s
t
<
i
n
t
>
>
&
r
e
s
u
l
t
s
,
l
i
s
t
<
i
n
t
>
p
r
e
f
i
x
)
;
void\ weaveLists(list<int>\ first, list<int>\ second, vector<list<int>> \&results,list<int>\ prefix);
void weaveLists(list<int> first,list<int> second,vector<list<int>>&results,list<int> prefix);在“交织”函数中同时对数组
f
i
r
s
t
first
first和
s
e
c
o
n
d
second
second进行递归,对数组
f
i
r
s
t
first
first进行递归时将数组
f
i
r
s
t
first
first的首元素删除并添加到前缀
p
r
e
f
i
x
prefix
prefix中,然后进行递归调用。对数组
s
e
c
o
n
d
second
second进行递归时将数组
s
e
c
o
n
d
second
second的首元素删除并添加到前缀
p
r
e
f
i
x
prefix
prefix中,然后进行递归调用。直到遇到
f
i
r
s
t
first
first和
s
e
c
o
n
d
second
second其中一个为空时才返回,这时将数组
f
i
r
s
t
first
first和
s
e
c
o
n
d
second
second中的所有元素依次加入到数组
p
r
e
f
i
x
prefix
prefix中得到的数组为一个交织的结果数组。“交织”操作的一个实例以及测试结果分别如图1和图2所示。
算法的一个实例以及测试结果分别如图3和图4所示。
class BinaryNode
{
private:
int data;
BinaryNode* left;
BinaryNode* right;
public:
BinaryNode(int value = 0, BinaryNode* pointer1 = nullptr, BinaryNode* pointer2 = nullptr)
{
data = value;
left = pointer1;
right = pointer2;
}
BinaryNode* getLeft()
{
return left;
}
BinaryNode* getRight()
{
return right;
}
void setLeft(BinaryNode* value)
{
left=value;
}
void setRight(BinaryNode* value)
{
right=value;
}
int getData()
{
return data;
}
};
void weaveLists(list<int> first, list<int> second, vector<list<int>> &results,list<int> prefix)
{
if (first.size() == 0 || second.size() == 0)
{
list<int> result;
for (int p : prefix)
{
result.push_back(p);
}
for (int f : first)
{
result.push_back(f);
}
for (int s : second)
{
result.push_back(s);
}
results.push_back(result);
return;
}
int headFirst = first.front();
first.pop_front();
prefix.push_back(headFirst);
weaveLists(first,second,results,prefix);
prefix.pop_back();
first.push_front(headFirst);
int headSecond = second.front();
second.pop_front();
prefix.push_back(headSecond);
weaveLists(first, second, results, prefix);
prefix.pop_back();
second.push_front(headSecond);
}
vector<list<int>>* allSequences(BinaryNode* root)
{
vector<list<int>>* results = new vector<list<int>>;
if (root == nullptr)
{
return results;
}
list<int> prefix;
prefix.push_back(root->getData());
vector<list<int>>* leftSeq = allSequences(root->getLeft());
vector<list<int>>* rightSeq = allSequences(root->getRight());
if ((leftSeq->size() == 0)&& (rightSeq->size() != 0))
{
list<int> templeft;
for (list<int> right : (*rightSeq))
{
vector<list<int>> weaved;
weaveLists(templeft, right, weaved, prefix);
for (list<int> res : weaved)
{
results->push_back(res);
}
}
}
else if ((leftSeq->size() != 0) && (rightSeq->size() == 0))
{
list<int> tempRight;
for (list<int> left : (*leftSeq))
{
vector<list<int>> weaved;
weaveLists(left, tempRight, weaved, prefix);
for (list<int> res : weaved)
{
results->push_back(res);
}
}
}
else if ((leftSeq->size() == 0) && (rightSeq->size() == 0))
{
list<int> tempRight;
list<int> templeft;
vector<list<int>> weaved;
weaveLists(templeft, tempRight, weaved, prefix);
for (list<int> res : weaved)
{
results->push_back(res);
}
}
else
{
for (list<int> left:(*leftSeq))
{
for (list<int> right:(*rightSeq))
{
vector<list<int>> weaved;
weaveLists(left, right, weaved, prefix);
for (list<int> res:weaved)
{
results->push_back(res);
}
}
}
}
return results;
}
int main()
{
list<int> first;
list<int> second;
vector<list<int>> results;
list<int> prefix;
first.push_back(1);
first.push_back(2);
second.push_back(3);
second.push_back(4);
weaveLists(first, second, results, prefix);
for (list<int> array : results)
{
for (int inter : array)
{
cout << inter << " ";
}
cout << endl;
}
return 0;
}
int main()
{
BinaryNode node10(10);
BinaryNode node15(15);
BinaryNode node5(5);
BinaryNode node3(3);
BinaryNode node7(7);
node10.setLeft(&node5);
node10.setRight(&node15);
node5.setLeft(&node3);
node5.setRight(&node7);
vector<list<int>>* results=allSequences(&node10);
for (list<int> array : (*results))
{
for (int inter : array)
{
cout << inter << " ";
}
cout << endl;
}
return 0;
}