小米笔试题

题目描述:
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 整型
     */
    void dfs(vector<string> matrix,vector<int>&path,int root,int target)
    {
        if (root == -1)
            return;
        if (root == target)
        {
            path.push_back(root);
            return;
        }
        int numofDepth = 0;
        for (int child = 0;child < matrix[root].size();child++)
        {
            if (matrix[root][child] == '1' && (root == 0 || path.back()!= child))
            {

                numofDepth++;
                path.push_back(root);
                //int child = i;
                dfs(matrix,path,child,target);
                if (path.back() == target)
                    return;
                path.pop_back();

            }
            if (numofDepth == 0)
                dfs(matrix,path,-1,target);
        }
    }   
    int getSplitNode(vector<string> matrix, int indexA, int indexB) {
        vector<int> pathA;
        vector<int> pathB;
        dfs(matrix,pathA,0,indexA);
        dfs(matrix,pathB,0,indexB);
        int i = 0,j = 0;
        for (;i < pathA.size() && j < pathB.size();i++,j++)
        {
            if (pathA[i] !=  pathB[j])
                break;
        }
        return pathA[i-1];

    }

};