小米笔试题

题目描述:
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];    }};