小米Git

git是一种分布式代码管理工具，git通过树的形式记录文件的更改历史，比如：
base’<–base<–A<–A’
^
| — B<–B’
小米工程师常常需要寻找两个分支最近的分割点，即base.假设git 树是多叉树，请实现一个算法，计算git树上任意两点的最近分割点。
（假设git树节点数为n,用邻接矩阵的形式表示git树：字符串数组matrix包含n个字符串，每个字符串由字符’0’或’1’组成，长度为n。matrix[i][j]==’1’当且仅当git树种第i个和第j个节点有连接。节点0为git树的根节点。）

输入例子:
[01011,10100,01000,10000,10000],1,2

输出例子:
1

class Solution {
public:
    /**
     * 返回git树上两点的最近分割点
     *
     * @param matrix 接邻矩阵，表示git树，matrix[i][j] == ‘1‘ 当且仅当git树中第i个和第j个节点有连接，节点0为git树的跟节点
     * @param indexA 节点A的index
     * @param indexB 节点B的index
     * @return 整型
     */

int getSplitNode(const vector<string>& matrix, int indexA, int indexB) {
    int n = matrix.size();

    vector< vector<int> > ge(n);
    vector< int > dep(n, -1);
    vector< int > fa(n, -1);

    for (int i = 0; i < n; ++i) {
        for (int j = i + 1; j < n; ++j) {
            if ( matrix[i][j] == ‘1‘ ) {
                ge[i].push_back(j);
                ge[j].push_back(i);
            }
        }
    }

    vector<int> stk;
    stk.push_back(0);
    dep[0] = 0;

    while ( !stk.empty() )
    {
        int u = stk.back(); stk.pop_back();
        for (int k = 0; k < ge[u].size(); ++k) {
            int v = ge[u][k];
            if ( fa[v] != -1 ) continue;

            fa[v] = u;
            dep[v] = dep[u] + 1;
            stk.push_back(v);
        }
    }

    int a = indexA, b = indexB;
    while ( dep[a] > dep[b] ) {
        a = fa[a];
    }
    while ( dep[a] < dep[b] ) {
        b = fa[b];
    }
    while ( a != b ) {
        a = fa[a];
        b = fa[b];
    }
    return a;
}

};