[编程题]小米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(vector<string> matrix, int indexA, int indexB) {
        /*先要判断两个数哪个所在的深度高一些，本题中分支排列顺序没有按照对应数值大小顺序排序，所以需要利用额外的空间来完整遍历所有节点，记录节点深度*/
        int branchNum=matrix.size();
        matrix[0][0]=0;
        queue<int>level;//这里用queue或者vector都可以
        level.push(0);
        while(!level.empty())
        {
            int i=level.front();
            level.pop();
            //level.erase(level.begin());
            int j=1;
            while(j<branchNum)
            {
                  if(matrix[i][j]=='1' && matrix[j][j]=='0')
                  {
                      matrix[j][j]=matrix[i][i]+1;
                      level.push(j);
                  }
                  j++;
            }                               
        }        

        int smallerA;
        int greaterB;
        while(indexA>0 && indexB>0)
        {
            int levelA=matrix[indexA][indexA];
            int levelB=matrix[indexB][indexB];         

            if(levelA<=levelB)
            {
               smallerA=indexA;
               greaterB=indexB;
            }
            else if(levelA>levelB)
            {
               smallerA=indexB;
               greaterB=indexA;
            }

            if(smallerA==0)
            {
               return 0;
            }
            else if(smallerA==greaterB)
            {
                return smallerA;
            }
            int i;
            for(i=0;i<branchNum;i++)
            {
               if(matrix[greaterB][i]=='1' && matrix[i][i]<matrix[greaterB][greaterB])
               {
                  break;
               }
            }
            if(matrix[smallerA][i]=='1' || smallerA==i)
            {
                if(matrix[smallerA][smallerA]<matrix[i][i])
                    return smallerA;
                else
                    return i;
            }
            else
            {
              indexA=smallerA;
              indexB=i;
            }           

        }
        return 0;        

    }
};