问题
给定一个插入序列就可以唯一确定一棵二叉搜索树。然而,一棵给定的二叉搜索树却可以由多种不同的插入序列得到。例如分别按照序列{2, 1, 3}和{2, 3, 1}插入初始为空的二叉搜索树,都得到一样的结果。于是对于输入的各种插入序列,你需要判断它们是否能生成一样的二叉搜索树。
输入格式:
输入包含若干组测试数据。每组数据的第1行给出两个正整数N (≤10)和L,分别是每个序列插入元素的个数和需要检查的序列个数。第2行给出N个以空格分隔的正整数,作为初始插入序列。随后L行,每行给出N个插入的元素,属于L个需要检查的序列。
简单起见,我们保证每个插入序列都是1到N的一个排列。当读到N为0时,标志输入结束,这组数据不要处理。
输出格式:
对每一组需要检查的序列,如果其生成的二叉搜索树跟对应的初始序列生成的一样,输出“Yes”,否则输出“No”。
输入样例:
4 2
3 1 4 2
3 4 1 2
3 2 4 1
2 1
2 1
1 2
0
输出样例:
Yes
No
No
解答
#include <iostream>
#include <vector>
#include <string>
using namespace std;
bool compare(vector<int> &p, vector<int> &q);
int main()
{
int N, L;
int d;
vector<string> output;
while (cin >> N)
{
if (N == 0)
break;
cin >> L;
vector<int> p;
for (int i = 0; i < N; ++i)
{
cin >> d;
p.push_back(d);
}
for (int i = 0; i < L; ++i)
{
vector<int> q;
for (int j = 0; j < N; ++j)
{
cin >> d;
q.push_back(d);
}
if (compare(p, q))
output.emplace_back("Yes");
else
output.emplace_back("No");
q.clear();
}
p.clear();
}
for (const string &s: output)
cout << s << endl;
return 0;
}
bool compare(vector<int> &p, vector<int> &q)
{
if (p.empty() and q.empty())
return true;
if ((p.empty() and !q.empty()) or (!p.empty() and q.empty()))
return false;
if (p[0] != q[0])
return false;
vector<int> a;
vector<int> b;
for (unsigned i = 1, j = 1; (i < p.size()) and (j < q.size()); ++i, ++j)
{
if (p[i] < p[0])
a.push_back(p[i]);
if (q[j] < q[0])
b.push_back(q[j]);
}
vector<int> c;
vector<int> d;
for (unsigned i = 1, j = 1; (i < p.size()) and (j < q.size()); ++i, ++j)
{
if (p[i] > p[0])
c.push_back(p[i]);
if (q[j] > q[0])
d.push_back(q[j]);
}
return compare(a, b) and compare(c, d);
}