标签：tput class button ret nbsp sample images lin highlight
给定一个二叉查找树，要求计算其高度，每个二叉查找树将给出先序与中序的遍历。
例如：一个二叉查找树其先序遍历为：16, 10, 4, 15, 23 ; 中序遍历为 4, 10, 15, 16, 23，则其高度为2（假定空树高度为-1，只有根节点的数高度为0）
The first line is the number of test cases. For each test case, the first line is the preorder of keys, the second line is the inorder of the keys.
第一行输入测试用例个数。
对于每个测试用例，第一行是key值的先序遍历，第二行是key值的中序遍历
对于每个测试用例，用一行输出树的高度
Copy sample input to clipboard2 4 5 6 4 5 6 6 4 8 9 10 4 6 8 9 10
2 3
#include <iostream>
#include <vector>
#include <sstream>
using namespace std;
struct TreeNode{
int data;
TreeNode* left;
TreeNode* right;
TreeNode(char data):data(data),left(NULL),right(NULL){
}
};
TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) {
if(preorder.empty()) return NULL;
TreeNode* root = new TreeNode(preorder[0]);
int pos = 0;
while(inorder[pos] != preorder[0]) pos++;
vector<int> left_pre(preorder.begin()+1, preorder.begin()+pos+1);
vector<int> right_pre(preorder.begin()+pos+1, preorder.end());
vector<int> left_in(inorder.begin(), inorder.begin()+pos);
vector<int> right_in(inorder.begin()+pos+1,inorder.end());
root->left = buildTree(left_pre, left_in);
root->right = buildTree(right_pre, right_in);
return root;
}
int h(TreeNode* root) {
if(!root) return -1;
return max(h(root->left), h(root->right)) + 1;
}
void extract(vector<int>& v, string& s) {
stringstream ss(s);
int data;
while(ss >> data) v.push_back(data);
}
int main() {
int numTest;
cin >> numTest;
cin.get();
while(numTest--) {
vector<int> preorder;
vector<int> inorder;
string pre, in;
getline(cin, pre);
getline(cin, in);
extract(preorder,pre);
extract(inorder,in);
TreeNode* root = NULL;
root = buildTree(preorder, inorder);
cout << h(root) << endl;
}
return 0;
}
标签：tput class button ret nbsp sample images lin highlight
原文地址：http://www.cnblogs.com/mmyz-sysu-panjn/p/6208731.html
