华为2016春招机试题

华为2016春招实习生的在线机试题共3道，分值分别为100、200和300。因前两道均十分简单，这里只贴出可能被坑的第三道。

原题如下：

求二叉树的深度
描述:
给出一个字符串形式表达的二叉树，求出指定节点深度。
输入的树形结构字符串格式为：
1、以父节点、左子树、右子树表示的二叉树；每个父节点不会超过两个子节点；
2、树的每一个节点采用单个字母表示；树的层次采用数字表示，树根的层次为1，下一层为2，不会超过9层；
3、字符串以“节点名称 层次数 节点名称 层次数…”的形式出现，同一个父节点下，先出现的为左子树。
例如字符串“a1b2c2d3e3f3”生成一棵如下的树：
         a
       /   \
      b     c
     / \   / 
    d   e f     
节点a的深度为3，节点b的深度是2，节点f的深度是1
运行时间限制: 无限制
内存限制: 无限制
输入:
一行字符串，表示一个二叉树
一行字符串，一个字符一个节点，输入确保字符不会存在重复节点
输出:
指定节点的深度，如果节点不存在，返回0；整数之间用空格隔开
样例输入:
a1b2c2d3e3f3
ab
样例输出:
3 2

网上有人贴出同样题目，并写出自己的代码。但他们的代码大都存在问题。关键在于题目并没有说按照从根到叶子的顺序输入节点，也没有说是满二叉树。因此那些答案只能过一部分测试样例。因此比较好的方法应当是按照题意建树（这里采用vector二维数组来保存节点），并深搜左子树计算深度。

下面贴出自己代码（通过全部测试）：

#include <iostream>
#include <cstring>
#include <vector> 
#include <string>
#include <algorithm>
#include <cstdio>
using namespace std;

vector<char> v[11];

int findDepth(int treeLevel, int nodeIndex, int depth) {
    if (treeLevel > 9) return depth;
    int len = v[treeLevel+1].size();
    if (len == 0) return depth;
    if (treeLevel <= 9 && nodeIndex*2 < len) {
        return findDepth(treeLevel+1, nodeIndex*2, depth+1);  //  之前没有加return，结果全错了。可能因为VS2005不支持直接调用一个int型的函数。 
    } else {
        return depth;
    }
}

int main() {
    string tree, query;
    getline(cin, tree);
    getline(cin, query);
    for (int i = 0; i < 11; i++) v[i].clear();
    int len = tree.length();
    int maxDepth = 0;
    for (int i = 0; i < len; i += 2) {  // construct the tree
        int index = tree[i+1]-'0';
        v[index].push_back( tree[i] );
        if (index > maxDepth) maxDepth = index; 
    }
    int len1 = query.size();
    int depth;
    vector<int> ans;
    /* test*/
    /*for (int i = 1; i <= maxDepth; i++) {
        int l = v[i].size();
    for (int j = 0; j < l; j++)
        cout << v[i][j] << " ";
    cout << endl;
    }*/
    for (int i = 0; i < len1; i++) {
        depth = 0;
        for (int j = 1; j <= maxDepth; j++) {  //  Iterate each level
            int tempLen = v[j].size();
            for (int k = 0; k < tempLen; k++) { // Iterate each node
                if (v[j][k] == query[i]) {
                    depth = findDepth(j, k, 1);
                    break;
                }
            }
        }
        ans.push_back(depth);
    }
    int ll = ans.size();
    if (ll == 0) return 0;
    for (int i = 0; i < ll-1; i++) cout << ans[i] << " ";
    cout << ans[ll-1] << endl;
    return 0;
}