小米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 len=matrix.size();
vector<int> parent(len,0);
vector<int> visited(len,0);
queue<int> q;
q.push(0);
while(!q.empty())
{
int i=q.front();
q.pop();
visited[i]=1;
for(int j=1;j!=len;++j)
if(visited[j]==0&&matrix[i][j]=='1')
{
q.push(j);
parent[j]=i;
}
}
vector<int> Arow;
int k=indexA;
while(k)
{
Arow.push_back(k);
k=parent[k];
}
Arow.push_back(k);
vector<int> Brow;
k=indexB;
while(k)
{
Brow.push_back(k);
k=parent[k];
}
Brow.push_back(k);

k=0;
auto itB=Brow.rbegin();
int Bsz=Brow.size();
auto iter=Arow.rbegin();
int Asz=Arow.size();
while(k!=Bsz&&k!=Asz &&*(itB+k)==*(iter+k))
++k;
return *(itB+k-1);
    }
};