【算法】多叉树寻找A\B节点的分支点

小米笔试中的一道题，题意是，多叉树，0为根节点，给定矩阵m，
m[i][j]==1表示两点连接，给出A,B两个节点，寻找他们的最近分支节点。
如题：

 "001001";
 "001100";
 "110010";
 "010000";
 "001000";
 "100000";
               0
             /    \
            2      5
           / \
          1   4
         /
        3

给出节点(3, 4)则最近分割点为2.
思路：类似于图的遍历，根据给出连接关系，找到每个节点的父节点，存入数组，对于给出的节点（3,4），依次找出3的父节点路径，然后找出4的路径，两者开始比对，最先相同的点即为分割点。

int h(vector<string>& m, int a, int b)
{
    if(m.size() == 0)
        return 0;
    vector<int> f(m.size(), 0);//表示父节点
    vector<int> lable(m.size(), 0);//标记是否标注过此点
    lable[0] = 1;//表示根节点已标记
    deque<int> note;//要查询的节点
    note.push_back(0);//从根节点开始
    int count = 1;//标记过的数目
    while(!note.empty())
    {
        if(count == m.size())
            break;//表示所有节点的父节点都记录完毕
        int i = note.front();
        note.pop_front();
        for(int  j = 0; j < m[i].size();j++)
        {
            if(m[i][j] == '1')
            {
                if(lable[j] == 0)
                {//表示找到当前节点的子节点，同时该节点未标记过
                    count++;
                    f[j] = i;
                    lable[j] = 1;
                    note.push_back(j);//将其子节点加入队列
                }

            }
        }
    }
    set<int> ip;//记录a节点的路径
    ip.insert(a);
    int ma = f[a];
    while(ma != 0)
    {//未到根节点
        ip.insert(ma);
        ma = f[ma];
    }
    int mb = b;
    while(mb != 0)
    {
        if(ip.count(mb))
            return mb;//若找到与a相同的节点即返回
        mb = f[mb];
    }
    return 0;
}